AI/ML Engineer at Kaizen Labs · LumiJobs

Responsibilities

Most companies building with AI connect someone else's models to their software. Kaizen's AI-native platform turns requirements into working applications, and today much of that intelligence comes from general-purpose models that carry capabilities we don't need, lack knowledge we do, and leave us and our customers paying frontier-model token costs for intelligence we don't control. This role changes that. You'll fine-tune, adapt and distill open-weight models into specialized intelligence Kaizen owns, and prove where it beats the general-purpose alternative. The goal isn't the biggest model or the most papers: it's public-sector software that works better, costs less to run, and is harder to replicate.
Fine-tune, adapt and distill open-weight models for requirements interpretation, code generation, repository understanding, structured output and software validation.
Own the experiment loop: curate datasets, build training pipelines, and test architectures and techniques until a promising idea becomes a working model.
Build evaluations on real software engineering tasks that set baselines, expose failure modes, and separate real gains from benchmark noise.
Work with platform engineers to find where current models fall short, then build the capabilities that close those gaps and ship them into production.
Experiment with supervised fine-tuning, synthetic data, preference optimization and reinforcement learning to improve model behavior on specialized tasks.
Deliver applications built around each customer's needs without frontier-model token spend, so public budgets go to residents, not inference.
Lay the groundwork (datasets, evals, research practices) for Kaizen to eventually train its own models from scratch.
What You'll Bring
Hands-on experience training, fine-tuning or adapting language models. You've worked with model weights, not just model APIs or prompts.
Strong software engineering fundamentals. You build reliable training pipelines, debug complex systems, and write code other engineers build on.
Strong Python and PyTorch, with a working understanding of transformer architectures, tokenization and language model training.

AI/ML Engineer

Responsibilities

Qualification

Required

Benefits

Company

Kaizen Labs

More roles at Kaizen Labs