Member of Technical Staff, AI Research - Post-Training
About Us Chakra Labs' mission is to encode human taste into intelligence. We source differentiated, proprietary data from end consumers — giving AI teams access to mission critical datasets for inference and post-training 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. Some of our people: • Nirmal Krishnan - CEO • Formerly: Microsoft, Hedge Fund, Head of Engineering at Artemis. • Alex Fung - CTO • Formerly: Snap for 5+ years (built Snap Map, led Snap’s crypto efforts), Founding Engineer at TipLink • 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. Adding in a distributed component in terms of sourcing labor - we are constantly learning and 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 Master / 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. Apply tot his job