About the job
About Us
At Chakra Labs, we are dedicated to transforming human taste into a sophisticated intelligence framework. Our focus is on creating high-fidelity trajectories and environments that push the boundaries of AI research, collaborating with leading laboratories in the field.
Why Join Us?
Engage with cutting-edge technologies spanning data, AI, and infrastructure.
Receive mentorship from a highly qualified and experienced team.
Enjoy a competitive salary package, comprehensive benefits, and a daily meal stipend of $30, alongside an attractive equity offering.
Considerations Before Applying
This is our passion. While we value fun, we are dedicated to hard work. If you prefer a standard 9-5 role, this may not be the right environment for you.
Uncertainty is part of the journey. The landscapes of data and AI are ever-evolving. While we have a defined mission, those uncomfortable with ambiguity may find this role challenging.
Your Profile
Advanced Knowledge: You hold a Master's or PhD in machine learning, with a solid grasp of linear algebra and stochastic gradient descent, capable of deriving linear regression manually.
Post-Training & Environments: You’re adept at executing SFT on distributed GPU clusters utilizing FSDP. You possess a keen interest in enhancing existing models with high-quality data and can conceptualize new environments and online reinforcement learning (GRPO, PPO).
Proficient in Python: Familiarity with FastAPI and PyTorch is essential, including REST APIs and backend development, along with the ability to abstract high-level concepts in Torch using lower-level primitives.
Experience: We seek approximately 3-5 years of relevant experience, though we value skills and alignment with our expectations over strict qualifications.
