Applied Research Engineer, Frontier AI Environments

Embedding; New York (NY); On-site; USD 140,000 - 190,000; Full time

Job summary

Chakra Labs in New York seeks a seasoned ML engineer to shape post-training workflows, environment design, and data-quality improvements for frontier AI research. You will design and run model-improvement loops across supervised fine-tuning, preference data, and RL approaches, and collaborate with labs to translate research goals into concrete environments and experiments.

The role emphasizes building robust training ecosystems, evaluation harnesses, and scalable infrastructure to push agent

Qualifications

• Masters or PhD-level knowledge of ML fundamentals.
• Experience with LLM fine-tuning, reward modeling, or reinforcement learning for language models.
• Experience designing agent environments and evaluation harnesses.
• Proficiency in Python and PyTorch; familiarity with FastAPI and ML infra.
• Strong engineering judgment for research-to-production transitions.
• 3–5 years of relevant experience.

Skills

ML fundamentals

Post-training experience

Environment-building

Python & PyTorch

Engineering judgment

3-5 years experience

Education

Masters or PhD in ML/CS

Tools

PyTorch

FastAPI

Job description

Chakra Labs in New York seeks a seasoned ML engineer to shape post-training workflows, environment design, and data-quality improvements for frontier AI research. You will design and run model-improvement loops across supervised fine-tuning, preference data, and RL approaches, and collaborate with labs to translate research goals into concrete environments and experiments.

The role emphasizes building robust training ecosystems, evaluation harnesses, and scalable infrastructure to push agent