Staff Machine Learning Engineer - World Models City
CloudKitchens5 days ago
Base Salary
$273k - $321k/yr
Responsibilities
- Define and build technical architecture for world models and foundation models for physical AI.
- Develop large-scale multimodal models that represent, predict, and model the evolution of complex physical environments.
- Research self-supervised, generative, predictive, and representation-learning approaches.
- Develop training strategies using large-scale real-world and simulated datasets.
- Make architectural decisions across data, model design, pretraining, fine-tuning, evaluation, inference, and deployment.
- Establish evaluation methodologies for physical-world understanding and prediction.
- Translate research into systems capable of operating on real machines and collaborate across perception, robotics, autonomy, simulation, and ML infrastructure.
- Provide technical leadership through research direction, architecture reviews, mentorship, experimentation, hands-on engineering, and talent assessment.
Requirements
- Deep expertise in machine learning and experience developing large-scale deep learning or foundation model systems.
- Strong understanding of modern model architectures and representation learning.
- Experience in areas such as multimodal learning, video models, generative models, self-supervised learning, predictive models, spatial intelligence, or embodied AI.
- Experience training models on large-scale datasets and understanding the relationships among data, architecture, compute, and performance.
- Strong understanding of the full ML lifecycle, including data strategy, architecture, training, evaluation, optimization, and inference.
- Experience translating research ideas into functioning machine learning systems.
- Strong software engineering fundamentals with hands-on ability in Python and modern ML frameworks.
- Ability to work in ambiguous research areas, make consequential technical decisions, influence research or engineering direction, and communicate complex ideas across disciplines.
Benefits
- Medical, dental, vision, disability, and life insurance.
- Flexible Spending Account and Health Savings Account options.
- 401(k), equity, sick time, unlimited flexible time off, and paid holidays.
- Paid parental leave and pre-tax commuter benefit plan.
- Team lunch in the SoMa office every Tuesday and Thursday.
- This is a full-time exempt role based in San Francisco and requires onsite work five days per week.
Tech Stack
Categories
AI ResearchML Engineering
About CloudKitchens
CloudKitchens builds delivery-first commercial kitchen facilities and restaurant software for food and beverage operators. It leases turnkey kitchen space and provides tools for order management, delivery logistics, and multi-brand operations, enabling restaurants and entrepreneurs to expand with lower upfront costs. The company is privately held and headquartered in Los Angeles, serving markets across the U.S. with a mix of owned sites and enterprise solutions.