Member of Technical Staff - Code Generation

Moonlake AI

$120K — $180K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • PhD in Computer Science, Machine Learning, Robotics, Graphics, or related field
  • Experience in leading AI research groups or frontier research labs
  • Strong publication record or impactful contributions in open-source projects
  • Proficiency in deep learning frameworks (e.g., PyTorch, JAX)
  • Strong coding skills in Python, C++, or CUDA
  • Experience with large-scale training or distributed systems
  • Expertise in fine-tuning or post-training large models, especially in code generation or multimodal contexts

Responsibilities

  • Train models to generate structured interactive environments
  • Enhance multimodal reasoning across vision, language, and simulation
  • Build agent systems for generating game logic and world structures
  • Design reasoning systems for interactive content creation
  • Develop real-time diffusion renderers for stylized gaming visuals
  • Optimize high-performance training and inference pipelines
  • Create environments for training embodied agents and explore reinforcement learning

Benefits

  • Collaborative work environment with cutting-edge AI technologies
  • Opportunity to shape the future of interactive digital worlds
  • Access to substantial funding and support from top-tier investors
  • Involvement in a team oriented towards rapid research-to-product transitions
  • On-site team culture fostering close collaboration and innovation
Full Job Description
Member of Technical Staff

Location: San Mateo, CA
Type: Full-time • On-site
Overview

Moonlake is building the next generation of AI infrastructure for interactive digital worlds. Our mission is to enable anyone to create, simulate, and interact with rich environments using natural language and multimodal inputs, turning simple ideas into worlds with structure, logic, and agents that can perceive and act.

Our team has raised $28M in seed funding from NVIDIA Ventures, Threshold Ventures, AIX ventures and notable angels including Naval Ravikant and Jeff Dean to build the foundational layer for the future of AI - powering everything from creative tools and games to robotics training, simulations, and digital twins. Our goal is to make building and experimenting with these environments as accessible and scalable as publishing video on the internet.

Our research sits at the intersection of:
  • reasoning models
  • diffusion models
  • simulation environments
  • embodied agents
  • generative game engines

We are looking for exceptional research engineers and applied researchers to help push the frontier of interactive AI.

What You'll Do

Depending on your strengths and interests, your work may include:

World Modeling
  • Train models that generate structured interactive environments
  • Improve multimodal reasoning across vision, language, and simulation
Code Generation Agents
  • Build agent systems that generate game logic and world structures
  • Design tool-using reasoning systems for interactive content generation
  • Work on post-training and fine-tuning of code generation models to improve reasoning, tool use, and environment construction
Diffusion Rendering
  • Develop real-time diffusion renderers for stylized game visuals
  • Improve performance and controllability of video diffusion models
Systems & Infrastructure
  • Build high-performance training and inference pipelines
  • Optimize large-scale training and distributed systems
Simulation & Embodied Intelligence
  • Build environments for training embodied agents
  • Explore reinforcement learning and multimodal reasoning
What We're Looking For

Strong Signals
  • PhD in Computer Science, Machine Learning, Robotics, Graphics, or related field
  • Experience working at frontier research labs or leading AI research groups
  • Strong publication record or impactful open-source contributions
Technical Skills
  • Deep learning frameworks (PyTorch, JAX, etc.)
  • Experience with large-scale training or distributed systems
  • Strong coding ability in Python/C++/CUDA
  • Experience fine-tuning or post-training large models (especially code generation or multimodal models)
Ideal Candidates
  • Researchers who want to move quickly from research to product
  • Engineers who enjoy building new systems from scratch
  • People excited about the intersection of AI + simulation + interactive worlds


We are committed to being an on-site, in-person team currently based in San Mateo

Similar Jobs

More Jobs at Moonlake AI

More Information Technology Jobs

Find similar Member of Technical Staff - Code Generation jobs: