AI Researcher, Core ML

Together AI•San Francisco, CA

42d•$160,000 - $230,000

About The Position

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.

Requirements

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

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