Together AI

Machine Learning Engineer - Inference

San Francisco · Full-time

Sponsorship not specified$160k-$230kDetected 75 days ago

PythonRustAlgorithmsCode ReviewMachine LearningPyTorchLLMsResearch

About the role

Together AI is seeking a Machine Learning Engineer to join our Inference Engine team, focusing on optimizing and enhancing the performance of our AI inference systems.
This role involves working with state-of-the-art large language models models and ensuring they run efficiently and effectively at scale.
Join us in shaping the future at Together AI!

Design and build the production systems that power the Together AI inference engine, enabling reliability and performance at scale.
Develop and optimize runtime inference services for large-scale AI applications.
Collaborate with researchers, engineers, product managers, and designers to bring new features and research capabilities to the world.
Conduct design and code reviews to ensure high standards of quality.
Create services, tools, and developer documentation to support the inference engine.
Implement robust and fault-tolerant systems for data ingestion and processing.
This position offers the chance to collaborate closely with AI researchers and engineers to create cutting-edge AI solutions.

3+ years of experience writing high-performance, well-tested, production-quality code.
Proficiency with Python and PyTorch.

Knowledge of Rust, Cython and compilers.
Together AI is a research-driven artificial intelligence company.
We have contributed to leading open-source research, models, and datasets to advance the frontier of AI.
Our team has been behind technological advancements such as FlashAttention, Hyena, FlexGen, and RedPajama.
About Together AI
Individual compensation will be determined by experience, skills, and job-related knowledge.
Equal Opportunity
Knowledge of existing AI inference systems such as TGI, vLLM, TensorRT-LLM, Optimum

We offer competitive compensation, startup equity, health insurance, and other competitive benefits.
The US base salary range for this full-time position is $160,000 - $230,000 + equity + benefits.
Our salary ranges are determined by location, level, and role.
Individual compensation will be determined by experience, skills, and job-related knowledge.

Together AI is an Equal Opportunity Employer and is proud to offer equal employment opportunities to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more.

opportunities to everyone regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, veteran status, and more

This listing is sourced directly from Together AI's careers page and normalized into a canonical job model.