Machine Learning Engineer - Inference

Responsibilities

• 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.

Requirements

• 3+ years of experience writing high-performance, well-tested, production-quality code.
• Proficiency with Python and PyTorch.
• Demonstrated experience in building high performance libraries and tooling.
• Excellent understanding of low-level operating systems concepts including multi-threading, memory management, networking, storage, performance, and scale.
• Preferred: Knowledge of existing AI inference systems such as TGI, vLLM, TensorRT-LLM, Optimum.
• Preferred: Knowledge of AI inference techniques such as speculative decoding.
• Preferred: Knowledge of CUDA/Triton programming.
• Nice to have: Knowledge of Rust, Cython and compilers.

Compensation 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.