Senior Machine Learning EngineerLocation: San Francisco
We are looking for an experienced Senior Machine Learning Engineer who is eager to advance the frontier of AI, help us design, build, and scale end-to-end novel foundation models, and leverage their hands-on experience implementing a wide range of pre-training and post-training models, including large foundation models (beyond just LLM fine-tuning).
This role is focused on:
- Designing, implementing, and scaling state-of-the-art models
- Productionizing research codes, models and technologically complex systems
- Shaping benchmark design and model evaluation frameworks
- Building agentic AI capabilities and long-term technical bets
About the roleThe role will involve:
- Collaborating with researchers and scientists to implement, evaluate and scale proof-of-concept models.
- Owning, implementing and integrating the latest state-of-the-art methods and external open-source codes.
- Develop AI systems capable of accurately understanding the universe and generating new knowledge.
- Training multi-modal models supporting different sensor and other modalities like text
Requirements- Bachelor's degree in computer science, engineering, a related field, or equivalent experience.
- 5+ years of relevant work experience.
- Prior experience building distributed training pipelines for multi-node systems using PyTorch and Ray.
- Experience training large diffusion or transformer models. Preferably on video or time series data.
- Proficiency with Python, Ray Trainer, PyTorch, and Anyscale framework.
- Familiarity with cloud platforms such as AWS, GCP, or Azure.
Nice to have- Past training of video or time-series models
- Startup experience, comfortable with a small dynamic team.
- Location wise, strong preference for in-person in Waterloo or San Francisco however remote work is possible for exceptional candidates.
