Machine Learning Engineer, Visual Generative Models (Post-Training & Evaluation) San Francisco Office · Hybrid · Full Time

Nunchux Ai; San Francisco (CA); Hybrid; USD 180,000 - 250,000; Full time

Benefits offered by this job

Health insurance

401(k) plan

Equity

Job summary

Nunchux AI, Inc. in San Francisco, CA is seeking a Machine Learning Engineer to build post-training pipelines that improve efficiency and quality of multimodal visual generation models.

You will develop post-training recipes, data pipelines, evaluation systems, and benchmarks, and bring cutting-edge research into production using Python and PyTorch. This is a hybrid role with four days in the office and one day remote.

Qualifications

• Hands-on experience with image or video generation models, or multimodal visual systems.
• Depth in post-training methods such as distillation, LoRA, or evaluation of visual generative models.
• Strong Python and PyTorch skills, with experience building post-training or evaluation code.
• Experience running post-training workloads across multiple GPUs with DeepSpeed or FSDP.
• Ability to design clean experiments, interpret results, and make data-driven recommendations.

Responsibilities

• Develop post-training recipes for image and video generation models.
• Build data pipelines to curate and version training and benchmark data.
• Create benchmarks and automated judges; run human preference studies.
• Benchmark and release models by measuring against baselines and identifying quality regressions.
• Integrate research methods into production pipelines and keep up with evolving post-training work.

Skills

Visual generative-model experience

Post-training or evaluation depth

ML engineering strength

Training systems

Experimental judgment

Tools

PyTorch

DeepSpeed

FSDP

Job description

About Nunchux AI

Nunchux AI builds infrastructure that makes multimodal generative AI faster and cheaper to serve, and easier to build on. Founded by MIT PhDs Muyang Li, Yujun Lin, and Zhekai Zhang with CMU Professor Jun-Yan Zhu, Nunchux brings together deep research expertise and production systems experience. Our work is built on nearly a decade of research from MIT and CMU, including nunchaku project, whose models have surpassed 4 million downloads. We have top VC backing, and we build for enterprises and for millions of developers.

The Role

Nunchux makes visual generative models fast and efficient enough for production. As a Machine Learning Engineer on post-training and evaluation, you will build post-training pipelines that improve model efficiency and quality, measure the trade-offs, and turn the best recipes into reliable workflows for the models we ship.

What You’ll Do

• Develop post-training recipes: Establish and validate post-training recipes for image and video generation models.

• Build data pipelines: Curate and version the training and benchmark data used for post-training and model evaluation.

• Build evaluation systems: Create benchmarks and automated judges, and run human preference studies, to measure generation quality, fidelity, and efficiency.

• Benchmark and release models: Measure models against relevant baselines, catch quality regressions, and give clear evidence for release decisions.

• Bring research into production: Keep up with post-training and evaluation research, then integrate the useful methods into the team's pipelines.

What You Bring

• Visual generative-model experience: Hands-on experience with image or video generation models, or with other multimodal visual systems. You understand the artifacts, failure modes, and quality trade-offs that matter in generated visual content.

• Post-training or evaluation depth: Depth in one of two areas: post-training methods such as distillation or LoRA, or the evaluation of visual generative models.

• ML engineering strength: Strong Python and PyTorch skills, with experience building post-training or evaluation code that others can run and maintain.

• Training systems: Comfortable running and adapting post-training workloads across multiple GPUs with FSDP, DeepSpeed, or similar tools.

• Experimental judgment: Able to design clean experiments, interpret the results, and make practical recommendations from the data.

Bonus Points

• Experience distilling or fine-tuning large-scale diffusion or video-generation models.
• Experience building automated judges or reward models for visual content, based on multimodal LLMs or vision-language models.
• Experience with large-scale evaluation datasets, annotation pipelines, or preference data.
• Experience building agentic visual systems.
• Contributions to major open-source ML projects.

Why Join

• Core technical work: Build the systems that let Nunchux make models faster without losing the quality customers care about.
• From research to product: Your work will inform model releases, support customization, and run in production rather than stay in a notebook.
• Technical collaboration: Work closely with a CMU PhD on our post-training work, and with the Nunchux research and inference teams.
• Proven traction: Build on open-source work with more than 4 million model downloads, and on growing industry partnerships.
• The team: Work with world-class researchers from MIT, Berkeley, and CMU, and with industry veterans from NVIDIA, AMD, Snowflake, and Adobe.
• Compensation: $180,000 to $250,000 USD base salary, plus equity and comprehensive benefits that include health insurance and a 401(k). Actual compensation will depend on relevant experience, skills, and qualifications.

Location: San Francisco, CA. 4 days in office, 1 day remote.

Start date: As soon as available

Visa: We sponsor H-1B and other work visas for exceptional candidates.

Learn more: nunchux.ai

Nunchux AI is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees.