AI Researcher
Core
Build large multi-modal generative world models, data infrastructure, and ML systems to enable robots to learn from experience and operate in real-world environments.
Role type
Senior IC AI Researcher (Embodied AI / World Models)
Builds
Large-scale generative world models, data processing pipelines, distributed training/inference systems, and evaluation frameworks for humanoid robots.
Domain
Robotics, Embodied AI, Large-Scale Machine Learning
Deliverable
production ML models | infrastructure
Required skills
Python, PyTorch, large-scale codebase development, data tooling and visualization, model architecture design, distributed training, inference optimization
Preferred skills
TorchTitan, DeepSpeed, FSDP/ZeRO, multi-modal generative models, world models, diffusion models, quantization, CUDA/Triton kernel development, TensorRT
Technologies
PyTorch, TorchTitan, DeepSpeed, FSDP, ZeRO, TensorRT, CUDA, Triton
Responsibilities
Design and operate data engines for training on visual and robot data; own distributed training and inference systems to maximize GPU utilization; build evaluation infrastructure connecting pre-training metrics to real-world robot performance; ship research results to production on robot hardware.
Seniority
Senior, hands-on IC