Senior Machine Learning Engineer, Cosmos Engineering

What you'll do

• Build and deliver Metropolis AI solutions using Cosmos world foundation models for video analytics and physical AI.
• Identify model performance gaps and develop solutions through synthetic data generation, tuning, and architectural improvements.
• Partner with the Cosmos team to prioritize and develop new platform features aligned with the Metropolis roadmap.
• Lead the open-sourcing of research artifacts and solutions to contribute to the broader AI community.
• Monitor and integrate the latest advances in foundation models, generative architectures, and training methodologies.
• Coordinate with Product, Engineering, and Data teams to align goals and unblock project execution.

What we're looking for

• MSc or PhD in Computer Science, Electrical Engineering, or a related field, or equivalent experience.
• 8+ years of proven experience in applied machine learning or AI research.
• Deep expertise in deep learning fundamentals, including hands-on experience with diffusion models and generative architectures.
• Experience in pre-training or refining large language models (LLMs), vision-language models (VLMs), or world foundation models (WFMs).
• Experience working with large-scale foundation models, including training workflows, fine-tuning techniques, and evaluation approaches.
• Experience working with simulation environments like Isaac Sim or similar platforms.
• End-to-end understanding of the ML development and deployment life cycle.
• Hands-on experience with model compression, quantization, and real-time inference optimization (preferred).