Member of Technical Staff, Post-Training
Chakra LabsAI Training company
Brooklyn, United StatesLead
Castle Island Ventures
Collab+Currency
Hash3
Race Capital
Solana Incubator
Data & AI
About the role
TL;DR
Design and implement model improvement workflows for AI research.
- •Chakra Labs builds high-fidelity trajectories and environments for AI research.
- •You'll design and run model improvement workflows, create environments, and work on reward and evaluation systems.
- •Key Responsibilities Design and run model improvement workflows.
- •Build environments and tasks for AI agents.
- •Create and improve high-fidelity trajectories for model training.
- •Develop reward functions and evaluation systems.
- •Run experiments on distributed GPU clusters.
- •Collaborate with frontier AI labs on research problems.
- •Requirements Master's/PhD in machine learning.
- •Experience with LLM fine-tuning, preference data, and reinforcement learning.
- •Python and PyTorch fluency.
- •Strong engineering judgment. 3-5 years of experience.
Required skills
PythonPyTorchFastAPIMLflowKubeflowAWSGoogle CloudCI/CDDockerKubernetesLLMs
Domain expertise
ai
Tech stack
PythonPyTorchFastAPIMLflowKubeflowAWSGoogle CloudCI/CDDockerKubernetes
