Member of Technical Staff, Reinforcement Learning

Member of Technical Staff, Reinforcement Learning Bay Area

Research

In office

Full-time

Inception creates the world’s fastest, most efficient AI models. Our Mercury model is the world’s fastest reasoning LLM and first commercially available diffusion LLM, delivering 5x greater speed and efficiency than today’s LLMs, with best-in-class quality.

We are the AI researchers and engineers behind such breakthrough AI technologies as diffusion models, flash attention, and DPO.

The Role We seek experienced scientists and engineers with deep expertise in post-training large language models through reinforcement learning. You will design and implement RL training pipelines for our diffusion LLMs, develop reward modeling strategies, and build the algorithms that align model behavior with human intent at scale.

Key Responsibilities

• Design, develop, and optimize RL training pipelines (PPO, DPO, RLHF, and novel approaches) for diffusion-based LLMs.
• Build and iterate on reward models, reward shaping strategies, and evaluation of reward quality.
• Implement innovative approaches for fine-tuning and scaling generative AI models.
• Work on data preprocessing pipelines, model evaluation, and alignment to enterprise use cases.
• Research and implement techniques for controlled text generation and constraint satisfaction.
• Improve training stability, efficiency, and reproducibility of RL workloads.

Qualifications

• BS/MS/PhD in Computer Science or a related field (or equivalent experience).
• At least 2 years of experience working on ML projects in PyTorch (or equivalent), preferably in a research lab or engineering role.
• Excellent familiarity with transformers and core LLM concepts (autoregressive pretraining, instruction tuning, in-context learning, KV caching).
• Hands-on experience with reinforcement learning from human feedback (RLHF), PPO, DPO, or related post-training methods.
• Familiarity with training and inference in diffusion models.
• Experience training deep learning models at scale in distributed computing environments.

Preferred Skills

• Extensive experience training transformer-based language models from scratch.
• Experience designing and implementing reward models or preference learning systems.
• Knowledge of advanced training techniques (mixed precision, gradient accumulation, etc.).
• Background in optimization theory and neural network architecture design.
• Experience with LLM serving frameworks like vLLM, SGLang, or TensorRT.

Why Join Inception

• Work with World-Class Talent: Collaborate with the inventors of diffusion models and leading AI researchers
• Shape Foundational Technology: Your decisions will influence how the next generation of AI products are built and used
• Immediate Impact: Join at the ground floor where your contributions directly shape product direction and company trajectory

Perks & Benefits

• Competitive salary and equity in a rapidly growing startup
• Flexible vacation and paid time off (PTO)
• Health, dental, and vision insurance
• Catered meals (breakfast, lunch, & dinner)
• Commuter subsidies
• A collaborative and inclusive culture

About Us Inception creates the world’s fastest, most efficient AI models. Today’s autoregressive LLMs generate tokens sequentially, which makes them painfully slow and expensive. Inception’s diffusion-based LLMs (dLLMs) generate answers in parallel. They are 5x faster and more efficient, while delivering best-in-class quality.

Inception was co-founded by Stanford professor Stefano Ermon, who co-invented such breakthrough AI technologies as diffusion models, flash attention, and DPO, UCLA professor Aditya Grover, who co-invented node2vec, decision transformers, and d1 reasoning, and Cornell professor and Afresh co-founder Volodymyr Kuleshov, who co-invented MDLM and Block Diffusion.

We pioneered the application of diffusion to language, with world’s first (and only) commercially available dLLM, Mercury. We are currently deploying our large-scale diffusion LLMs at Fortune 500 companies. Diffusion is the technology behind today’s image and video AI, and we’re making it the standard for LLMs as well.

Our team includes engineers from Google DeepMind, Meta AI, Microsoft AI, and OpenAI. Based in Palo Alto, CA, we are backed by A-list venture capitalists, including Menlo Ventures, Mayfield, M12 (Microsoft’s venture fund), Snowflake Ventures, Databricks, and Innovation Endeavors, and by tech luminaries such as Andrew Ng, Andrej Karpathy, and Eric Schmidt.

If you are talented, innovative, and ambitious, come help us invent the future of AI. We are an equal opportunity employer and encourage candidates of all backgrounds to apply.

Ready to apply? Powered by

First name *

Last name *

Email *

LinkedIn URL

Resume * Click to upload or drag and drop here

Req ID: R28