AIML - Machine Learning Engineer, Foundation Models

What you'll do

• Design and implement scalable, high-performance machine learning infrastructure for multi-modal foundation models.
• Optimize models for performance, efficiency, and on-device intelligence on Apple silicon.
• Productionize state-of-the-art AI algorithms across text, image, and speech domains.
• Implement machine learning systems that meet stringent privacy and security requirements.
• Develop mid-training approaches to bridge general capabilities with task-specific performance.
• Solve reward modeling problems to resist reward hacking and handle sparse rewards in agentic settings.
• Align models for creative tasks and precise, action-taking workflows.
• Manage a small team of engineers as required.

What we're looking for

• MS or PhD in Computer Science, Machine Learning, or a related technical field.
• Expert-level programming skills in Python.
• Proficiency in machine learning frameworks such as Jax, PyTorch, or TensorFlow.
• Strong background in distributed training, model optimization, and machine learning infrastructure.
• Experience with large-scale model training and deployment.
• Familiarity with Kubernetes, Docker, and cloud platforms like AWS, GCP, or Azure.
• Experience with foundation models and large language models (preferred).
• Background in multi-modal AI systems (preferred).