Machine Learning, Platform Engineer

Job Application for Machine Learning, Platform Engineer at Together AI Machine Learning, Platform Engineer San Francisco

About the Role

Our team focuses on enabling custom models and dedicated inference on Together. We are responsible for building a container platform, optimizing autoscaling, minimizing cold starts, achieving the best end-to-end model performance, and providing a best-in-class developer experience with great tooling. We often focus on video or audio generation across the stack: CUDA kernels, pytorch optimization, inference engines, container orchestration, queueing theory, etc. An ideal candidate will be great at profiling/optimization but know the word kubernetes, or be intimately familiar with multi-cluster scheduling and have some sense of ML bottlenecks.

Responsibilities

New hires may work on multi-cluster orchestration, portfolio optimization, predictive autoscaling, control panes, model bring-up, model optimization, APIs for managing deployments, inference worker SDKs, and CLI tools.

Analyze and improve the robustness and scalability of existing distributed systems, APIs, databases, and infrastructure

Partner with product teams to understand functional requirements and deliver solutions that meet business needs

Write clear, well-tested, and maintainable software and IaC for both new and existing systems

Conduct design and code reviews, create developer documentation, and develop testing strategies for robustness and fault tolerance

Requirements

5+ years of demonstrated experience in building large scale, fault tolerant, distributed systems.

Experience running serverless inference platforms, doing model bring-up on short notice, being on call, or running a cloud provider is a very big plus

Good taste and ability to thoughtfully discuss how what you’ve built has failed over time

Experience designing, analyzing and improving efficiency, scalability, and stability of various system resources

Excellent understanding of low level operating systems concepts including concurrency, networking and storage, performance and scale

Expert-level programmer in one or more of Python, Golang, Rust, C++, or Haskell

Proficiency in writing and maintaining Infrastructure as Code (IaC) using tools like Terraform

Experience with Kubernetes internals or other container orchestration systems

Sound judgement for when to use and when to not use LLMs for code

Bachelor’s or Master’s degree in Computer Science, Computer Engineering, or a related technical field, or equivalent practical experience

Writing-heavy roles or companies are a plus

About Together AI

Together AI is a research-driven artificial intelligence company. We believe open and transparent AI systems will drive innovation and create the best outcomes for society, and together we are on a mission to significantly lower the cost of modern AI systems by co-designing software, hardware, algorithms, and models. We have contributed to leading open-source research, models, and datasets to advance the frontier of AI, and our team has been behind technological advancement such as FlashAttention, Hyena, FlexGen, and RedPajama. We invite you to join a passionate group of researchers and engineers in our journey in building the next generation AI infrastructure.

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 - $250,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.

Equal Opportunity

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

Please see our privacy policy at https://www.together.ai/privacy

Create a Job Alert Interested in building your career at Together AI? Get future opportunities sent straight to your email. Create alert

Apply for this job

*

indicates a required field

Quick Apply with MyGreenhouse

First Name *

Last Name *

Preferred First Name

Email *

Phone Country

Phone

Resume/CV Attach Attach

Dropbox Google Drive Enter manually Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Cover Letter Attach Attach

Dropbox Google Drive Enter manually Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

LinkedIn Profile *

Please link to some code you're proud of writing. *

Are you willing to work four days per week in our SF or Amsterdam office? * Select...

Are you legally authorized to work in the country where the job is located? * Select...

Will you now or in the future require company sponsorship to retain or extend your work authorization in the country where the job is located? * Select...