CrawlJobs Logo

LLM Inference Frameworks and Optimization Engineer

United States, San Francisco 160000.00 - 230000.00 USD / Year · Job Posted February 18, 2026
Apply Position
Job Link Share

Job Description

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.

Job Responsibility

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

  • 3+ years of experience in deep learning inference frameworks, distributed systems, or high-performance computing
  • 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
  • Proficient in Python and C++/CUDA for high-performance deep learning inference
  • Deep understanding of Transformer architectures and LLM/VLM/Diffusion model optimization
  • Knowledge of inference optimization, such as workload scheduling, CUDA graph, compiled, efficient kernels
  • 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

What we offer

  • competitive compensation
  • startup equity
  • health insurance
  • other competitive benefits

Looking for more opportunities?

Search for other job offers that match your skills and interests.

Similar Jobs for

LLM Inference Frameworks and Optimization Engineer

8 matching positions

Senior LLM Inference Performance Engineer

WHAT YOU DO AT AMD CHANGES EVERYTHING At AMD, our mission is to build great prod...
Location
Location
Finland , Helsinki
Salary
Salary:
Not provided
amd.com Logo
AMD
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Seasoned in deploying LLMs and other AI model types in production using frameworks like vLLM, SGLang, or similar tooling
  • Deep knowledge about LLM serving and performance metric evaluation
  • Comfortable with Python software development and bash scripting
  • Experience with Docker, Kubernetes and Helm
  • Desire and ability to continuously learn in a fast-changing environment
  • Initiative, pragmatic problem solving, and great collaboration skills
  • Bachelor's or master's degree in computer science, computer engineering, electrical engineering, or an equivalent field
Job Responsibility
Job Responsibility
  • LLM and AI Performance: Measure, analyze, and optimize LLM and AI service performance across metrics like latency and throughput for various training and inference use cases
  • Design and implement methodologies for measuring model performance, and automating optimization strategies to identify optimal configurations
  • Stay on top of current advances in AI, models, APIs, and open-source ecosystems, and translate them into scalable solutions
  • LLM and AI Tooling: Design and develop tooling to measure and analyze the performance of AI model deployments and the effect of different configurations and infrastructure, standalone and Kubernetes clusters
  • Develop and maintain tooling for interacting with different ecosystem functions to improve developer and user experience
  • Develop and maintain internal tooling to support LLM and AI performance tuning at scale
  • Fulltime
Read More
Arrow Right

Software Engineer II and Senior Software Engineer - Performance

The Artificial Intelligence Performance team at Microsoft develops AI software t...
Location
Location
United States , Mountain View
Salary
Salary:
100600.00 - 199000.00 USD / Year
https://www.microsoft.com/ Logo
Microsoft Corporation
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's Degree in Computer Science or related technical field AND 2+ years technical engineering experience with coding in languages including, but not limited to, C, C++, C#, or Python OR equivalent experience
  • Ability to meet Microsoft, customer and/or government security screening requirements are required for this role
  • Microsoft Cloud Background Check: This position will be required to pass the Microsoft Cloud background check upon hire/transfer and every two years thereafter
Job Responsibility
Job Responsibility
  • Identify and drive improvements to end-to-end inference performance of OpenAI and other state-of-the-art LLMs
  • Measure, benchmark performance on Nvidia/AMD GPUs and first party Microsoft silicon
  • Optimize and monitor performance of LLMs and build SW tooling to enable insights into performance opportunities ranging from the model level to the systems and silicon level to improve customer experience and reduce the footprint of the computing fleet
  • Enable fast time to market of LLMs/models and their deployments at scale by building SW tools that afford velocity in porting models on new Nvidia and AMD GPUs
  • Design, implement, and test functions or components for our AI/DNN/LLM frameworks and tools
  • Speeding up/reducing complexity of key components/pipelines to improve performance and/or efficiency of our systems
  • Communicate and collaborate with our partners both internal and external
  • Embody Microsoft's Culture and Values
  • Fulltime
Read More
Arrow Right

Member of Technical Staff, LLM Inference - MAI Superintelligence Team

Our Inference team is responsible for building and maintaining the tools and sys...
Location
Location
United States , Mountain View
Salary
Salary:
139900.00 - 274800.00 USD / Year
https://www.microsoft.com/ Logo
Microsoft Corporation
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's Degree in Computer Science or related technical field AND 6+ years technical engineering experience with coding in languages including, but not limited to, C, C++, C#, Java, JavaScript, or Python
  • OR equivalent experience
  • Experience with generative AI
  • Experience with distributed computing
  • Python and Python ecosystem (eg. uv, pybind/nanobind, FastAPI) expertise
  • Experience with large scale production inference
  • Experience with GPU kernel programming
  • Experience benchmarking, profiling, and optimizing PyTorch generative AI models
  • Experience with open source inference frameworks like vLLM and SGLang
  • Working experience and conversant with the material in the JAX scaling book
Job Responsibility
Job Responsibility
  • Work alongside researchers and engineers to implement frontier AI research ideas
  • Introduce new systems, tools, and techniques to improve model inference performance
  • Build tools to help debug performance bottlenecks, numeric instabilities, and distributed systems issues
  • Build tools and establish processes to enhance the team’s collective productivity
  • Find ways to overcome roadblocks and deliver your work to users quickly and iteratively
  • Enjoy working in a fast-paced, design-driven product development cycle
  • Embody our Culture and Values
  • Fulltime
Read More
Arrow Right

AI Inference Engineer

We are looking for an AI Inference engineer to join our growing team. Our curren...
Location
Location
United States , San Francisco; Palo Alto; New York City
Salary
Salary:
210000.00 - 385000.00 USD / Year
perplexity.ai Logo
Perplexity
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Experience with ML systems and deep learning frameworks (e.g. PyTorch, TensorFlow, ONNX)
  • Familiarity with common LLM architectures and inference optimization techniques (e.g. continuous batching, quantization, etc.)
  • Understanding of GPU architectures or experience with GPU kernel programming using CUDA
Job Responsibility
Job Responsibility
  • Develop APIs for AI inference that will be used by both internal and external customers
  • Benchmark and address bottlenecks throughout our inference stack
  • Improve the reliability and observability of our systems and respond to system outages
  • Explore novel research and implement LLM inference optimizations
What we offer
What we offer
  • equity
  • health
  • dental
  • vision
  • retirement
  • fitness
  • commuter and dependent care accounts
  • Fulltime
Read More
Arrow Right

AI Inference Engineer

We are looking for an AI Inference engineer to join our growing team. Our curren...
Location
Location
United Kingdom , London
Salary
Salary:
Not provided
perplexity.ai Logo
Perplexity
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Experience with ML systems and deep learning frameworks (e.g. PyTorch, TensorFlow, ONNX)
  • Familiarity with common LLM architectures and inference optimization techniques (e.g. continuous batching, quantization, etc.)
  • Understanding of GPU architectures or experience with GPU kernel programming using CUDA
Job Responsibility
Job Responsibility
  • Develop APIs for AI inference that will be used by both internal and external customers
  • Benchmark and address bottlenecks throughout our inference stack
  • Improve the reliability and observability of our systems and respond to system outages
  • Explore novel research and implement LLM inference optimizations
What we offer
What we offer
  • Equity may be part of the total compensation package
  • Fulltime
Read More
Arrow Right

Full Stack LLM Engineer

Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs. ...
Location
Location
India , Bengaluru
Salary
Salary:
Not provided
cerebras.net Logo
Cerebras Systems
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor’s, Master’s, or PhD in Computer Science, Engineering, or a related field with 8 to 12 years’ experience
  • Comfort navigating the full AI toolchain: Python modelling code, compiler IRs, performance profiling, etc.
  • Strong debugging skills across performance, numerical accuracy, and runtime integration
  • Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and familiarity with model internals (e.g., attention, MoE, diffusion)
  • Proficiency in C/C++ programming and experience with low-level optimization
  • Strong background in optimization techniques, particularly those involving NP-hard problems
Job Responsibility
Job Responsibility
  • Contribute to the end-to-end bring up of frameworks for RL, inference serving, ML models on Cerebras CSX systems
  • Work across the stack: model architecture translation, graph lowering, compiler optimizations, runtime integration, and performance tuning
  • Debug performance and correctness issues spanning model code, compiler IRs, runtime behavior, and hardware utilization
  • Propose and prototype improvements across tools, APIs, or automation flows to accelerate future bring ups
What we offer
What we offer
  • Competitive salary and benefits package
  • Opportunities for professional growth and career advancement
  • A dynamic and innovative work environment
  • The chance to work on cutting-edge technologies and make a significant impact on the future of AI
Read More
Arrow Right

Ai Engineer – Genai And Agentic Systems

We are seeking a creative and motivated engineer to join our AI/ML development t...
Location
Location
Greece , Athens
Salary
Salary:
Not provided
gtisoft.com Logo
Gamma Technologies
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's or Master's in Computer Science, Data Science, Machine Learning, or a related technical field
  • 3+ years of industry experience in NLP, LLMs, or GenAI solution development
  • Proven experience delivering at least one end-to-end production deployment of an LLM or agentic AI system
  • Strong expertise in GenAI orchestration strategies, multi-agent workflows, and high-performance model integration
  • Hands-on experience with Python (primary), with working knowledge of C++ preferred for optimized inference
  • Understanding of RAG pipelines, vector databases, and evaluation frameworks
  • Strong analytical skills and the ability to deliver well-architected solutions independently and within teams
  • Authorization to work in Athens Greece, on a full-time basis
  • visa sponsorship is not available for this role
Job Responsibility
Job Responsibility
  • Architect and implement agentic AI systems using frameworks such as MCP, Claude Agent SDK, LangGraph, or Amazon Strands
  • Build robust orchestration and automation workflows integrating multi-agent capabilities
  • Develop and maintain pipelines for training, benchmarking, validating, and deploying LLMs and autonomous agents into production environments
  • Conduct prompt engineering and fine-tuning of LLMs for domain-specific tasks
  • Evaluate and iterate model performance based on metrics and user feedback
  • Integrate AI capabilities with GT-SUITE and related engineering tools via APIs, SDKs, and microservices
  • Ensure software quality through clean, maintainable, and optimized code, including support for efficient inference
  • Stay current with emerging technologies in GenAI, LLM orchestration, and autonomous agents
What we offer
What we offer
  • Dynamic and highly talented team of experts
  • An inclusive and supportive work environment that celebrates diverse perspectives and fosters belonging for all
  • The opportunity to bring in your own ideas, implement them, and make them accessible to a large customer base
  • The opportunity to build a professional network in various branches of industry
  • An attractive salary and additional company benefits
  • Fulltime
Read More
Arrow Right

Senior Software Engineer - AI Frameworks

The AI Frameworks team at Microsoft accelerates and optimizes large language mod...
Location
Location
United States , Redmond
Salary
Salary:
119800.00 - 234700.00 USD / Year
https://www.microsoft.com/ Logo
Microsoft Corporation
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's Degree in Computer Science or related technical field AND 4+ years technical engineering experience with coding in languages including, but not limited to, C, C++, or Python OR equivalent experience.
  • Master's Degree in Computer Science or related technical field AND 6+ years technical engineering experience with coding in languages including, but not limited to, C, C++, or Python OR Bachelor's Degree in Computer Science or related technical field AND 8+ years technical engineering experience with coding in languages including, but not limited to, C, C++, or Python OR equivalent experience.
  • Experience with PyTorch internals, custom operators, hardware backend, or torch.compile/Dynamo-based optimization flows.
  • Experience with AI inference stacks such as vLLM, SGLang, or similar large-scale model serving systems.
  • Experience with NPU or GPU kernel development and optimization (e.g., CUDA, Triton, or accelerator-specific toolchains).
  • Familiarity with common LLM concepts such as attention mechanisms, KV caching, quantization (PTQ/QAT), and distributed parallelism strategies (TP, PP, DP).
Job Responsibility
Job Responsibility
  • Architect and implement efficient tensor computation primitives and software abstractions for custom AI accelerators
  • Develop and extend PyTorch features for model onboarding, optimization, and execution on custom AI accelerators
  • Contribute to and improve AI inference stacks such as vLLM and SGLang, including scheduling, KV cache management, and serving pipelines
  • Design, develop, profile, and optimize high-performance kernels for NPUs (MAIA) and GPUs to accelerate LLM inference and training workloads
  • Collaborate across disciplines to define requirements and deliver practical solutions to new technical challenges
  • Fulltime
Read More
Arrow Right