Hybrid ML Engineer: Post-Training for Visual Gen

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

Benefits offered by this job

Equity

Health insurance

401(k)

Job summary

Nunchux AI in San Francisco is looking for a Machine Learning Engineer focused on post-training and evaluation to improve model efficiency and quality. You will build post-training recipes and data pipelines, run benchmarks, and turn research findings into production-ready workflows.

You will work with cutting-edge multimodal models, collaborating across research and inference teams to ship reliable systems. This role offers competitive compensation and equity.

Qualifications

• Visual generative-model experience with image or video generation.
• Depth in post-training methods or evaluation of visual generative models.
• Strong Python and PyTorch skills for producing post-training/evaluation code.
• Experience running post-training workloads across multiple GPUs with tools like FSDP/DeepSpeed.
• 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 to measure generation quality, fidelity, and efficiency.
• Benchmark and release models, tracking quality and regressions with clear evidence.
• Bring research into production by integrating useful methods into pipelines.

Skills

Visual generative-model experience

Post-training depth

ML engineering strength

Training systems

Experimental judgment

Job description

Nunchux AI in San Francisco is looking for a Machine Learning Engineer focused on post-training and evaluation to improve model efficiency and quality. You will build post-training recipes and data pipelines, run benchmarks, and turn research findings into production-ready workflows.

You will work with cutting-edge multimodal models, collaborating across research and inference teams to ship reliable systems. This role offers competitive compensation and equity.