AI Researcher, Core ML
About the Role
As an AI Researcher, you will be pushing the frontier of foundation model research and make them a reality in products. You will be working on developing novel architectures, system optimizations, optimization algorithms, and data-centric optimizations, that go beyond state-of-the-arts. As a team, we have been pushing on all these fronts (e.g., Hyena, FlashAttention, FlexGen, and RedPajama). You will also work closely together with the machine learning systems, NLP/CV, and engineering teams for inspiration of research problems and to jointly work on solutions to practical challenges. You will also interact with customers to help them in their journey of training, using, and improving their AI applications using open models. Your research skills will be vital in staying up-to-date with the latest advancements in machine learning, ensuring that we stay at the cutting edge of open model innovations.
Responsibilities
- Develop novel architectures, system optimizations, optimization algorithms, and data-centric optimizations, that significantly improve over state-of-the-art
- Take advantage of the computational infrastructure of Together to create the best open models in their class
- Understand and improve the full lifecycle of building open models; release and publish your insights (blogs, academic papers etc.)
- Collaborate with cross-functional teams to deploy your models and make them available to a wider community and customer base
- Stay up-to-date with the latest advancements in machine learning
Qualifications
- Strong background in Machine Learning
- Experience in building state-of-the-art models at large scale
- Experience in developing algorithms in areas such as optimization, model architecture, and data-centric optimizations
- Passion in contributing to the open model ecosystem and pushing the frontier of open models
- Excellent problem-solving and analytical skills
- Bachelor's, Master's, or Ph.D. degree in Computer Science, Electrical Engineering, or a related field