Member of Technical Staff, Post-Training

ABOUT US Chakra Labs' mission is to encode human taste into intelligence. We build high-fidelity trajectories and environments for frontier AI research, working with several of the top labs.   WHY YOU SHOULD WORK WITH US - You will get to touch the latest and greatest technologies across the data, AI, and infrastructure stack. - You will get mentored by an experienced and qualified team. - You will get paid a very competitive cash package with top benefits and a $30 daily meal stipend. We also provide a compelling equity package. WHY YOU SHOULDN’T WORK WITH US - This is our life’s work. We have fun, but we work hard. If you want to clock a 9-5, this probably isn’t a great fit. - There’s no clear playbook for success. Data and AI are constantly shifting. While we have a clear mission and vision, if you’re uncomfortable with ambiguity this role will be challenging for you. ABOUT YOU - You have a Masters / PhD level knowledge of machine learning. You understand the fundamentals of linear algebra, stochastic gradient descent, and can derive linear regression by hand. - Post-training & Environments. You are comfortable running SFT across distributed GPU clusters using FSDP. You have an appetite to improve existing models using high-quality data and are capable of reasoning about new environments and online RL (GRPO, PPO). - Python (FastAPI, PyTorch, etc.): REST APIs, back-end, vector-math. You can derive some high level abstractions in Torch using lower level primitives. - Experience: we don’t believe in hard and fast rules here. Roughly 3-5 years of experience is what we’re looking for, but more or less experience works as long as the above expectations resonate with you.