Member of Technical Staff, AI Research - Post-Training
Automate your job search with Sonara.
Submit 10x as many applications with less effort than one manual application.1
Reclaim your time by letting our AI handle the grunt work of job searching.
We continuously scan millions of openings to find your top matches.
Job Description
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.
Automate your job search with Sonara.
Submit 10x as many applications with less effort than one manual application.