Kaizen Labs

AI/ML Engineer

Kaizen Labs
Apply
3 days ago

Base Salary

$240k - $280k/yr

Responsibilities

  • Fine-tune, adapt, and distill open-weight models for software-engineering-related tasks.
  • Own the model experimentation loop, including dataset curation, training pipelines, architecture experiments, and technique evaluation.
  • Build evaluations on real software engineering tasks to establish baselines, expose failure modes, and measure genuine improvements.
  • Collaborate with platform engineers to identify model limitations and ship capabilities into production.
  • Experiment with supervised fine-tuning, synthetic data, preference optimization, and reinforcement learning.
  • Deliver customer-focused applications that reduce reliance on expensive frontier-model inference.
  • Establish datasets, evaluations, and research practices for future in-house model training.

Requirements

  • Demonstrated hands-on experience training, fine-tuning, or adapting language models using model weights rather than only model APIs or prompts.
  • Strong software engineering fundamentals, including building reliable training pipelines, debugging complex systems, and writing reusable code.
  • Strong Python and PyTorch skills, with working knowledge of transformer architectures, tokenization, and language-model training.
  • A demonstrated model-engineering project, such as a fine-tuned model, original training approach, or meaningful open-source ML contribution.
  • Preferred qualifications include experience with code generation, program synthesis, repository understanding, synthetic data, distillation, quantization, preference optimization, reinforcement learning, model development, applied ML research, or reproducing significant research results.

Benefits

  • Medical, dental, and vision coverage is fully paid for employees and dependents, with Oxford/United, Guardian PPO, and Beam plans.
  • Includes fully paid life insurance, FSA and Dependent Care FSA, One Medical membership, fertility and family-building support, and a 401(k) with a 2% company match.
  • Provides 16 weeks of paid parental leave for birthing parents and 10 weeks for non-birthing parents.
  • Offers unlimited PTO with a two-week minimum, federal holidays, a company-wide winter break, and company offsites.
  • Hybrid New York City work arrangement with in-office work Monday through Thursday; candidates must live in or commute to New York City.
  • Provides home-office or desk setup stipends, commuter benefits, expensed office lunches, late-work dinner and Uber rides, and a company laptop.
  • Includes gym or fitness reimbursement, utility, home cleaning, professional development, recreation, and pet-care stipends.

Tech Stack

Categories

Contact me