Production ML Engineer: Diffusion LLMs

Inception; San Francisco (CA); On-site; USD 200,000 - 350,000; Full time

Benefits offered by this job

Equity

Health insurance

Dental insurance

Vision insurance

Flexible PTO

Catered meals

Career growth

Job summary

Inception is seeking experienced engineers and researchers to bridge research and real-world application of diffusion LLMs. You will design, train, and deploy models at scale, collaborate with customers to translate their needs into technical solutions, and ensure reliability in production environments.

You will work with cutting-edge tech, from diffusion models to MLOps, and join a fast-growing team in Palo Alto, CA, with competitive compensation, equity, flexible PTO, and comprehensive health

Qualifications

• BS/MS/PhD in Computer Science, Machine Learning, or a related field (or equivalent experience).
• At least 5 years of experience working on ML projects in PyTorch (or equivalent).
• Excellent familiarity with transformers and core LLM concepts (autoregressive pretraining, instruction tuning, in-context learning, LoRA, KV caching).
• Experience training LLMs, including fine-tuning.
• Familiarity with large-scale systems and high-performance computing, including GPU/TPU utilization.
• Experience with version control (Git) and containerization (Docker).
• Excellent communication skills with the ability to explain technical concepts to non-technical stakeholders.

Responsibilities

• Design, develop, and optimize our models for production use cases.
• Partner with customers to understand their requirements and translate them into technical solutions.
• Implement innovative approaches for post-training generative AI models, including agentic workflows.
• Work on data preprocessing pipelines, model evaluation, and alignment to enterprise use cases.
• Contribute to the deployment and maintenance of models in production environments.
• Collaborate with product teams to design and implement customer-facing ML features.

Skills

PyTorch

Transformers

LLM concepts

Model training

Production systems

Communication skills

GPU/TPU usage

Education

BS/MS/PhD in Computer Science/ML or related field

Tools

Docker

Git

vLLM

TensorRT

Kubernetes

Job description

Inception is seeking experienced engineers and researchers to bridge research and real-world application of diffusion LLMs. You will design, train, and deploy models at scale, collaborate with customers to translate their needs into technical solutions, and ensure reliability in production environments.

You will work with cutting-edge tech, from diffusion models to MLOps, and join a fast-growing team in Palo Alto, CA, with competitive compensation, equity, flexible PTO, and comprehensive health