Senior Machine Learning Scientist

📋 Description

• Work on LLM, MLLM, CV in cloud, devices, and robotics.
• Own key tech areas across LLM/MLLM/CV product portfolio.
• Lead junior scientists and translate research into Axon features.
• Develop and deploy cutting-edge LLM/MLLM and CV models.
• Optimize models for performance, memory, and energy on constrained devices.
• Collaborate with hardware teams to leverage accelerators.

🎯 Requirements

• PhD with 8+ years in ML/AI focusing on LLM/MLLM/CV.
• Proven publications in LLM/MLLM/CV and GenAI.
• Strong Python, C/C++, PyTorch, TensorFlow, Keras, ROS.
• Experience shaping data, modeling, training, evaluation pipelines.
• Deliver high-quality models at scale using state-of-the-art research.
• Excellent problem solving and cross-functional communication.

🎁 Benefits

• Competitive salary and 401k with employer match
• Discretionary paid time off
• Paid parental leave
• Medical, Dental, Vision plans
• Fitness Programs
• Emotional & Mental Wellness support