LLM Inference Frameworks and Optimization Engineer
Together AISan Francisco, California, United States · Posted Mar 25, 2025Description
About the Role
At Together.ai, we are building state-of-the-art infrastructure to enable efficient and scalable inference for large language models (LLMs). Our mission is to optimize inference frameworks, algorithms, and infrastructure, pushing the boundaries of performance, scalability, and cost-efficiency.
We are seeking an Inference Frameworks and Optimization Engineer to design, develop, and optimize distributed inference engines that support multimodal and language models at scale. This role will focus on low-latency, high-throughput inference, GPU/accelerator optimizations, and software-hardware co-design, ensuring efficient large-scale deployment of LLMs and vision models.
This role offers a unique opportunity to shape the future of LLM inference infrastructure, ensuring scalable, high-performance AI deployment across a diverse range of applications. If you're passionate about pushing the boundaries of AI inference, we’d love to hear from you!
Responsibilities
Inference Framework Development and Optimization
- Design and develop fault-tolerant, high-concurrency distributed inference engine for text, image, and multimodal generation models.
- Implement and optimize distributed inference strategies, including Mixture of Experts (MoE) parallelism, tensor parallelism, pipeline parallelism for high-performance serving.
- Apply CUDA graph optimizations, TensorRT/TRT-LLM graph optimizations, and PyTorch-based compilation (torch.compile), and speculative decoding to enhance efficiency and scalability.
Software-Hardware Co-Design and AI Infrastructure
- Collaborate with hardware teams on performance bottleneck analysis, co-optimize inference performance for GPUs, TPUs, or custom accelerators.
- Work closely with AI researchers and infrastructure engineers to develop efficient model execution plans and optimize E2E model serving pipelines.
Requirements
Must-Have:
- Experience:
- 3+ years of experience in deep learning inference frameworks, distributed systems, or high-performance computing.
- Technical Skills:
- Familiar with at least one LLM inference frameworks (e.g., TensorRT-LLM, vLLM, SGLang, TGI(Text Generation Inference)).
- Background knowledge and experience in at least one of the following: GPU programming (CUDA/Triton/TensorRT), compiler, model quantization, and GPU cluster scheduling.
- Deep understanding of KV cache systems like Mooncake, PagedAttention, or custom in-house variants.
- Programming:
- Proficient in Python and C++/CUDA for high-performance deep learning inference.
- Optimization Techniques:
- Deep understanding of Transformer architectures and LLM/VLM/Diffusion model optimization.
- Knowledge of inference optimization, such as workload scheduling, CUDA graph, compiled, efficient kernels
- Soft Skills:
- Strong analytical problem-solving skills with a performance-driven mindset.
- Excellent collaboration and communication skills across teams.
Nice-to-Have:
- Experience in developing software systems for large-scale data center networks with RDMA/RoCE
- Familiar with distributed filesystem(e.g., 3FS, HDFS, Ceph)
- Familiar with open source distributed scheduling/orchestration frameworks, such as Kubernetes (K8S)
- Contributions to open-source deep learning inference projects.
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 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 - $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.
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
Similar jobs
Est. 205,000 USD
About the Role Together AI is building the Inference Platform that brings the most advanced generative AI models to the world. Our platform powers multi-tenant serverless workloads and dedicated endpoints, enabling devel…
Est. 195,000 USD
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…
Est. 240,000 USD
About the Role The Turbo team sits at the intersection of efficient inference (algorithms, architectures, engines) and post‑training / RL systems. We build and operate the systems behind Together’s API, including high‑pe…
Est. 240,000 USD
About the Role This is a research engineering role with direct production impact. You won’t be publishing ideas in isolation—you will translate new RL algorithms, scheduling methods, and inference optimizations into prod…
Est. 245,000 USD
About the role The Model Shaping team at Together AI works on products and research focused on tailoring open foundation models to downstream applications. We build services that enable machine learning developers to cho…
Est. 255,000 USD
About the Role Together AI is building the AI Native Cloud, an end-to-end platform for the fullgenerative AI lifecycle, combining the fastest LLM inference engine with state-of-the-artAI cloud infrastructure. The Togethe…
Est. 120,000 GBP
About the Role As a Solutions Architect (Inference) at Together AI, you will work with customers and prospects to create business value through Generative AI applications. Solutions Architects at Together are trusted adv…
Est. 260,000 USD
About the Role We're looking for a Software Engineer to build the systems that treat infrastructure as software. This role owns the software state machines that provision hardware, bring it into service, and manage its f…
Est. 195,000 USD
About the role As a Technical Support Engineer at a pioneering AI company, you'll be the first line of defense to support customers as they build out training, fine tuning, and inference solutions with Together AI. You'l…
Est. 220,000 USD
About the Role As a Solutions Architect at Together AI, you will work with customers and prospects to create business value through Generative AI applications. Solutions Architects at Together are trusted advisors to our…
Est. 195,000 USD
About the Role Together AI is building the AI Acceleration Cloud, an end-to-end platform for the full generative AI lifecycle, combining the fastest LLM inference engine with state-of-the-art AI cloud infrastructure. As…
Est. 120,000 GBP
About the Role We're looking for a Software Engineer to build the systems that treat infrastructure as software. This role owns the software state machines that provision hardware, bring it into service, and manage its f…
Est. 90,000 EUR
About the Role We're looking for a Software Engineer to build the systems that treat infrastructure as software. This role owns the software state machines that provision hardware, bring it into service, and manage its f…
Est. 245,000 USD
About the Role The Model Shaping team at Together AI works on products and research for tailoring open foundation models to downstream applications. We build services that allow machine learning developers to choose the…
Est. 275,000 USD
About the Role In this role, you will operate, scale, and optimize multi-petabyte storage systems purpose-built for the world’s largest AI training and inference workloads. You’ll manage and scale high-performance parall…
About the role As a Technical Support Engineer at a pioneering AI company, you'll be the first line of defense to support customers as they build out training, fine tuning, and inference solutions with Together AI. You'l…
Est. 120,000 EUR
About the Role Together AI is building the AI Acceleration Cloud, an end-to-end platform for the full generative AI lifecycle, combining the fastest LLM inference engine with state-of-the-art AI cloud infrastructure. As…
Est. 230,000 USD
Build the infrastructure powering the next generation of AI. At Together AI, you’ll build and operate one of the world’s largest GPU fleets used for frontier model training and inference. This isn’t a traditional infrast…
About the role As a Forward Deployed Engineer (FDE) focused on Inference & Post-Training, you will be a hands-on technical partner to our most strategic customers — production AI teams looking to leverage high qualit…
Est. 155,000 USD
About The Role Together AI is rapidly scaling its compute infrastructure across multiple sites and deployment types. The Associate, Infrastructure Strategy & Operations will be the analytical backbone of the Infrastr…
Est. 195,000 USD
About the Role As a Systems Research Engineer specialized in GPU Programming, you will play a crucial role in developing and optimizing GPU-accelerated kernels and algorithms for ML/AI applications. Working closely with…
Est. 240,000 USD
About the Role Together AI is building the AI Acceleration Cloud, an end-to-end platform for the full generative AI lifecycle, combining the fastest LLM inference engine with state-of-the-art AI cloud infrastructure. The…
Est. 120,000 EUR
About the Role Together AI is building the AI Acceleration Cloud, an end-to-end platform for the full generative AI lifecycle, combining the fastest LLM inference engine with state-of-the-art AI cloud infrastructure. As…
About the Role We're looking for a Software Engineer to build the systems that treat infrastructure as software. This role owns the software state machines that provision hardware, bring it into service, and manage its f…
Est. 195,000 USD
About the Role Together AI's product is what developers touch every day — Playground, Model Garden, Fine-Tuning, File Management, Batch Processing, Model Evals. These surfaces are how users experience the platform, and t…
Est. 235,000 USD
About the Role Together AI is looking for a Senior Network Engineer to design, deploy, and operate the global network infrastructure supporting our production services and high-performance AI compute environments. This i…
Est. 215,000 USD
About the Role Together AI is looking for a Commercial Counsel to support both the agreements that secure our infrastructure and the deals that drive our growth. You'll work alongside the Infrastructure, Sales, Finance,…
Est. 202,500 USD
As a Software Engineer on the ML Infrastructure team, you will design and build platforms for scalable, reliable, and efficient serving of LLMs. Our platform powers cutting-edge research and production systems, supportin…
Est. 245,000 USD
About the Role The Model Shaping team at Together AI works on products and research for tailoring open foundation models to downstream applications. We build services that allow machine learning developers to choose the…
Est. 225,000 USD
About The Role Together AI is growing its compute footprint, and making sure new capacity meets our technical standards is an important priority for the company. Every new cluster has to clear a technical bar before it c…