Machine Learning Engineer, Simulation Scenario Generation
Develops ML models and workflows for generating simulation scenarios in autonomous vehicle testing, integrating LLMs/VLMs, collecting data for fine-tuning, and collaborating with teams to enhance safety validation. Requires MS/PhD, 2+ years ML experience, and Python/PyTorch proficiency.
Responsibilities
- Integrate and validate LLMs/VLMs and implement other models for complex scenario generation workflows, leveraging techniques like advanced prompting, agentic tool use, and more.
- Contribute to tooling for AI-based scenario understanding and validation.
- Collect data and design metrics to drive business intelligence, product iteration, and model fine-tuning.
- Collaborate directly with internal customers and partner teams to provide generative AI solutions for their test creation workflows.
- Directly contribute to the safety and reliability of autonomous software.
Qualifications
- MS or PhD in Computer Science, Machine Learning, or related field
- 2+ years of industry experience in Machine Learning
- Solid understanding of LLM or NLP concepts
- Proficiency in Python and ML libraries (PyTorch, NumPy) demonstrated through professional or research projects
Bonus Qualifications
- Practical experience in dataset creation for fine-tuning, system integration of ML models into production, or optimization techniques for low-latency inference systems
- Familiarity with autonomous vehicles, robotics, and/or complex simulation environments
- Hands-on experience in areas like program synthesis, diffusion models, and/or formal methods/V&V
- Relevant publications in conferences (e.g., CVPR, ICCV, RSS, and/or ICRA)
Senior Software Engineer, AI Runtime
Senior Software Engineer building and scaling Databricks' managed GPU training platform (AI Runtime) for large-scale distributed AI model training. Requires 5+ years in distributed systems and hands-on experience with GPU training frameworks.
Sr. Machine Learning Engineer, Computer Vision
Build and prototype diffusion-based text-to-image generative models (Pinterest Canvas) using large-scale visual-text datasets. Requires 5+ years industry computer vision experience and an M.S. or Ph.D.
Senior AI/ML Engineer
Senior AI/ML Engineer building transformer and deep learning models on financial and behavioral data to power personalized growth and marketing experiences at Chime. Requires strong production ML experience with PyTorch, AWS, and large-scale data infrastructure.