Member of Technical Staff, ML Performance

About the Company

Odyssey is an AI lab pioneering general world models: causal, multimodal systems that learn to predict and interact with the world over long horizons. Our founders previously pioneered self-driving cars, and our team includes researchers from DeepMind, Tesla, Waymo, Meta, and Apple.

Responsibilities

• Optimize models for real-time use by hundreds of thousands of users
• Design and implement distributed training strategies to reduce resource consumption on large GPU clusters
• Partner with ML researchers to ensure model architectures are performant from conception
• Develop tools to identify performance bottlenecks and stability issues in training and serving environments
• Pioneer innovative frameworks and system designs to enhance performance metrics
• Exercise significant autonomy in technical decisions using latest-generation GPUs

Requirements

• 8+ years of software engineering experience
• Significant experience in ML performance
• Deep insight into modern machine learning architectures
• Expertise in distributed training and inference optimization
• Proficiency with PyTorch, TF, or JAX
• Proficiency with Triton and NVIDIA GPU ecosystems
• Proven track record of owning projects end to end
• Highly metric-based approach to problem solving