ML Researcher, AIML

What you'll do

• Pre-train foundation models and develop mid-training approaches to bridge general capabilities with task-specific performance.
• Optimize foundation models specifically for Apple silicon hardware.
• Develop reward modeling techniques to resist reward hacking and handle sparse or delayed rewards.
• Align models for open-ended creative tasks and precise, action-taking workflows.
• Build end-to-end voice interactions, natural language processing, and knowledge-integrated learning systems.
• Implement reinforcement learning and on-policy distillation techniques for large language models.
• Execute post-training and mid-training processes for large language models.

What we're looking for

• PhD or equivalent practical experience in Computer Science or a related technical field.
• Demonstrated expertise in deep learning with a publication record in relevant conferences or a track record in product application.
• Proficient programming skills in Python and one of the deep learning toolkits such as JAX, PyTorch, or Tensorflow.
• Ability to work in a collaborative environment.
• Experience coding Large Language Models (preferred).
• Experience in building end-to-end voice models (preferred).
• Experience in speech recognition and generation (preferred).
• Experience in reinforcement learning, on-policy distillation, and post-training or mid-training large language models (preferred).