Inference Optimization

Inference Optimization Engineer Jobs

2 jobs found

Today

Machine learning engineer II, AWS Inference Optimization

Amazon Development Center U.S., Inc. Santa Clara, CA

Santa Clara, CA

Today

$165k - $224k/yr Full time Artificial Intelligence AWS Aws Bedrock

Apply Now

Job Description

Machine learning engineer on an AWS inference optimization team focused on improving large language model inference performance.

Responsibilities

Develop model servers for large language OSS models (examples include Qwen, Gemma, gptoss, and others)
Build inference stack components, including CUDA kernel development in vLLM and SGLang
Create and iterate on services for GenAI optimization and benchmarking
Create and modify a mid-size application or service
Apply inference optimization techniques to improve performance, including:
- Speculative decoding
- Disaggregated Inference
- CUDA kernel tuning
Own price-performance for Amazon LLM inference on SageMaker, Hyperpod, and Bedrock
Deliver production-ready solutions for customer-facing products, using existing academic/industrial research and practical problem solving
Optimize inference performance using AWS inference stacks including vLLM and SGLang
Tune GenAI inference performance on Amazon SageMaker, Amazon Bedrock, and Hyperpod

Requirements

3+ years of non-internship professional software development experience
2+ years of non-internship experience designing or architecting new and existing systems (design patterns, reliability, scaling)
1+ years of software development engineer or related occupational experience
1+ years building large-scale, multi-tiered, multi-threaded, embedded, or distributed software applications, tools, systems, and services using C#, C++, Java, or Perl
Bachelor’s degree (Computer Science, Engineering, Mathematics, or related field) or foreign equivalent
Knowledge of Machine Learning and LLM fundamentals, including:
- Transformer architecture
- Training and inference lifecycles
- Optimization techniques

Technologies

vLLM, SGLang
Amazon SageMaker, Amazon Bedrock, Hyperpod
CUDA
C#, C++, Java, Perl
Transformer architecture

Preferred Qualifications

3+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations
Bachelor’s degree in computer science or equivalent

Benefits

Sign-on payments
Restricted stock units (RSUs)
Health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance, option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line)
Flexible Spending Accounts
Adoption and Surrogacy Reimbursement coverage
401(k) matching
Paid time off
Parental leave
Workplace accommodation or adjustment during the application and hiring process (including support for interview or onboarding)

Additional Information for Los Angeles County Applicants

Work safely and cooperatively with employees, supervisors, and staff
Adhere to standards of excellence despite stressful conditions
Communicate effectively and respectfully to ensure exceptional customer service
Follow all federal, state, and local laws and company policies
Criminal history may have a direct, adverse, and negative relationship with some material job duties, including the abilities to follow policies, exercise sound judgment, manage stress, work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the company’s reputation
Under the Los Angeles County Fair Chance Ordinance, qualified applicants with arrest and conviction records will be considered

Location: Santa Clara, CA (onsite)

Compensation: USD 165,200 - 223,600 per year

View full job page →

3mo ago

Sr. Lead AI Engineer (Inference Optimization, FM hosting, AI Platform)

Capital One McLean, VA

McLean, VA

3mo ago Closed

$230k - $262k/yr Full time Ai Ml Ai Platform Artificial Intelligence

This position is no longer accepting applications (closed Aug 20, 2026).

Job Description

Capital One is seeking a Sr. Lead AI Engineer focused on inference optimization, foundation model hosting, and AI platform development, onsite in McLean, VA, with a salary range of USD 229,900 to 262,400 per year.

Responsibilities

Collaborate with a cross functional team of engineers, research scientists, technical program managers, and product managers to deliver AI powered products that transform how associates work and how customers engage with Capital One.
Design, build, test, deploy, and support AI software components including foundation model training, large language model inference, similarity search, guardrails, model evaluation, experimentation, governance, and observability.
Leverage a broad stack of Open Source and SaaS AI technologies such as AWS Ultraclusters, Huggingface, VectorDBs, Nemo Guardrails, PyTorch, and more.
Develop and introduce state of the art LLM optimization techniques to improve performance, scalability, cost, latency, and throughput of large scale production AI systems.
Contribute to the technical vision and the long term roadmap of foundational AI systems at Capital One.

Requirements

Bachelor's degree in Computer Science, AI, Electrical Engineering, Computer Engineering, or related fields plus at least 6 years of experience developing AI and ML algorithms or technologies
Master's degree in Computer Science, AI, Electrical Engineering, Computer Engineering, or related fields plus at least 4 years of experience developing AI and ML algorithms or technologies
At least 6 years of experience programming with Python, Go, Scala, or Java
7 years of experience deploying scalable and responsible AI solutions on cloud platforms (e.g. AWS, Google Cloud, Azure, or equivalent private cloud)
Experience designing, developing, integrating, delivering, and supporting complex AI systems
Demonstrated ability to lead and mentor an engineering team and influence cross functional stakeholders
Experience developing AI and ML algorithms or technologies (e.g. LLM Inference, Similarity Search and VectorDBs, Guardrails, Memory) using Python, C++, C#, Java, or Golang
Experience developing and applying state of the art techniques for optimizing training and inference software to improve hardware utilization, latency, throughput, and cost
Passion for staying abreast of the latest AI research and AI systems, and judiciously apply novel techniques in production
Excellent communication and presentation skills, with the ability to articulate complex AI concepts to peers

Technologies

AWS Ultraclusters
Huggingface
VectorDBs
Nemo Guardrails
PyTorch
Python
Go
Scala
Java
C++
C#
Golang

Benefits

Capital One offers a comprehensive, competitive, and inclusive set of health, financial and other benefits that support your total well being.
This role is also eligible to earn performance based incentive compensation, which may include cash bonus(es) and/or long term incentives (LTI).

Overview

At Capital One, we are creating responsible and reliable AI systems, changing banking for good. For years, Capital One has been an industry leader in using machine learning to create real-time, personalized customer experiences. Our investments in technology infrastructure and world-class talent — along with our deep experience in machine learning — position us to be at the forefront of enterprises leveraging AI. From informing customers about unusual charges to answering their questions in real

Team Description

The Intelligent Foundations and Experiences (IFX) team is at the center of bringing our vision for AI at Capital One to life. We work hand-in-hand with our partners across the company to advance the state of the art in science and AI engineering, and we build and deploy proprietary solutions that are central to our business and deliver value to millions of customers. Our AI models and platforms empower teams across Capital One to enhance their products with the transformative power of AI, in res

The Ideal Candidate

You love to build systems, take pride in the quality of your work, and also share our passion to do the right thing. You want to work on problems that will help change banking for good.
Passion for staying abreast of the latest research, and an ability to intuitively understand scientific publications and judiciously apply novel techniques in production.
You adapt quickly and thrive on bringing clarity to big, undefined problems. You love asking questions and digging deep to uncover the root of problems and can articulate your findings concisely with clarity. You have the courage to share new ideas even when they are unproven.
You are deeply Technical. You possess a strong foundation in engineering and mathematics, and your expertise in hardware, software, and AI enable you to see and exploit optimization opportunities that others miss.
You are a resilient trail blazer who can forge new paths to achieve business goals when the route is unknown.

View full job page →

Inference Optimization Engineer Jobs

Machine learning engineer II, AWS Inference Optimization

Sr. Lead AI Engineer (Inference Optimization, FM hosting, AI Platform)

Get Inference Optimization Engineer Jobs alerts