AWS Trainium Jobs - Remote & On-site

AWS Trainium jobs hiring now at vetted startups and product companies on Alion, updated daily. Average salary around $360,811/year. Compare remote, hybrid and on-site positions with relocation and visa sponsorship. Browse and apply today.

AI/ML
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Executive
Networking
Product
Design
Analytics
Support
Enterprise Apps
Quantum

Anthropic

anthropic.com
Anthropic is an American artificial intelligence safety and research company founded in 2021 by former OpenAI researchers, among them the siblings Dario and Daniela Amodei. It develops the Claude family of large language models and ships them through a consumer assistant, an enterprise developer platform and the Claude Code agentic coding tool, alongside open standards such as the Model Context Protocol. Incorporated as a public benefit corporation and headquartered in San Francisco, the company concentrates on interpretability, alignment and reliability research and counts Google and Amazon among its largest investors.
anthropic.com • HQ: San Francisco, United States • Code Intelligence • AI Agents • Machine Learning • LLM & Generative AI • Artificial Intelligence • 1001-5000 employees • Est. 2021
HQ: San Francisco, United States • Code Intelligence • AI Agents • Machine Learning • LLM & Generative AI • Artificial Intelligence • 1001-5000 employees • Est. 2021
Top 25% payVerified live · 1 hour ago 2 months ago

Staff+ Software Engineer, Inference Velocity

$405k per year • Staff+ • 8+ years expIn office (San Francisco, United States) • Visa sponsorship • Bachelor's Degree • Staff
Claude
CUDA Toolkit
CUDA
Anthropic
AWS Trainium
TPU
Multimodal AI
Triton
DevOps
AWS
CI/CD
HPC
Kubernetes
Platform Engineering
Apply
Top 25% payVerified live · 1 hour ago 1 month ago

Staff+ Software Engineer, Capacity Engineering

$320k per year • Staff+ • In office (San Francisco, United States) • Visa sponsorship • Bachelor's Degree • Staff
Python
SQL
Databases
Google BigQuery
AI/ML
Anthropic
Multimodal AI
AWS Trainium
TPU
DevOps
Azure
GCP
Grafana
Kubernetes
Prometheus
Apply
Top 25% payVerified live · 1 hour ago 6 months ago

Technical Program Manager, Infrastructure

$290k per year • Lead • 5+ years expIn office (San Francisco, United States) • Visa sponsorship • Bachelor's Degree • Staff
Claude
Multimodal AI
Anthropic
AWS Trainium
TPU
DevOps
AWS
Azure
CI/CD
GCP
Kubernetes
Apply
Report

OpenAI

openai.com
OpenAI is an American artificial intelligence research and deployment company founded in 2015 with the mission of ensuring that artificial general intelligence benefits all of humanity. It develops the GPT family of large language models and turns them into consumer and developer products, including the ChatGPT assistant, the Sora video model, the Codex coding agent and a commercial API used by millions of developers. Structured as a public benefit corporation controlled by a non-profit foundation, the company is backed by Microsoft and SoftBank and operates from San Francisco.
openai.com • HQ: San Francisco, United States • Multimodal AI • AI Agents • Artificial Intelligence • Machine Learning • LLM & Generative AI • 1001-5000 employees • Est. 2015
HQ: San Francisco, United States • Multimodal AI • AI Agents • Artificial Intelligence • Machine Learning • LLM & Generative AI • 1001-5000 employees • Est. 2015
Top 25% payVerified live · 16 min ago 12 days ago

Software Engineer, Trainium

$295k – $380k per year • Middle • 3+ years expRemote/Hybrid (San Francisco, United States) • Full-Time • Middle
JAX
PyTorch
Reinforcement Learning
AWS Trainium
OpenAI
TPU
DevOps
AWS
Apply
Report

World Labs

worldlabs.ai
World Labs is a spatial intelligence company, building frontier models that can perceive, generate, and interact with the 3D world.
worldlabs.ai • San Francisco • Data Science • Computer Vision • Geospatial Analytics
San Francisco • Data Science • Computer Vision • Geospatial Analytics
Open 121 dayTop 25% pay 3 months ago

Performance Engineer (Inference, Training & GPU)

$200k – $300k per year • Equity • In office (San Francisco, United States)
C++
Python
Rust
C++
PyTorch C++
AI/ML
CUDA
CUDA Toolkit
JAX
PyTorch
Quantization
Triton
AWS Trainium
NCCL
NVLink
TPU
Apply
Report

Figure AI

figure.ai
Figure AI is an American robotics company founded in 2022 that develops general-purpose humanoid robots for commercial and household work. Its Figure 02 and Figure 03 machines combine custom actuators, battery systems and hands with the in-house Helix vision-language-action model, which lets one neural network drive both perception and motor control from natural-language instructions. Headquartered in San Jose, California, the company runs the BotQ manufacturing facility for high-volume production and has piloted its robots on logistics and automotive assembly work with commercial partners.
figure.ai • HQ: San Jose, United States • Manufacturing • Humanoid Robots • Artificial Intelligence • Robotics • Robotics AI • 501-1000 employees • Est. 2022
HQ: San Jose, United States • Manufacturing • Humanoid Robots • Artificial Intelligence • Robotics • Robotics AI • 501-1000 employees • Est. 2022
Top 25% payVerified live · 9 hours ago 18 days ago

Helix AI Engineer, Training Performance

$200k – $400k per year • Middle • 3+ years expIn office (San Jose, United States) • Bachelor's Degree • Full-Time • Middle
C++
Python
C++
PyTorch C++
AI/ML
CUDA Toolkit
DeepSpeed
JAX
PyTorch
vLLM
AWS Trainium
FSDP
InfiniBand
Megatron-LM
NCCL
NVLink
TPU
AI Agents
Apply
Report

Yottalabs

yottalabs.com
yottalabs.com • Hong Kong • Singapore
Hong Kong • Singapore
Verified live · 8 hours ago 29 days ago

Research Engineer Intern - AI Systems

Intern • Hong KongSingapore • Remote (United States, Canada, Hong Kong, Singapore) • Bachelor's Degree • Internship • Intern
C++
Python
C++
PyTorch C++
AI/ML
CUDA
CUDA Toolkit
LLM
PyTorch
Quantization
SGLang
Triton
vLLM
AWS Trainium
ROCm
Mixture of Experts
DevOps
AWS
GitHub
Apply
Verified live · 8 hours ago 2 months ago

Research Engineer - AI Systems

$136k – $298k per year (Estimated)Hong KongSingapore • Remote (United States, Canada, Hong Kong, Singapore) • Full-Time
C++
Python
C++
PyTorch C++
AI/ML
CUDA Toolkit
LLM
PyTorch
Quantization
SGLang
TensorRT
TensorRT-LLM
vLLM
AWS Trainium
ROCm
Mixture of Experts
RLHF
DevOps
AWS
Apply
Open 406 daysVerified live · 8 hours ago 1 year ago

GPU Cloud Platform Engineer

$139k – $249k per year (Estimated) • Senior • 5+ years expHong KongSingapore • Remote (United States, Canada, Hong Kong, Singapore) • Bachelor's Degree • Full-Time • Senior
Go
Python
AI/ML
Ray
AWS Trainium
CUDA Toolkit
Quantization
SGLang
vLLM
CUDA
Triton
Hugging Face
NCCL
NVLink
DevOps
AWS
Azure
Docker
GCP
Grafana
Helm
kubectl
Kubernetes
Prometheus
Rancher
SLURM
gRPC
GitHub
Apply
Report

Lila Sciences

lila.ai
Lila Sciences is an artificial intelligence enterprise developing an autonomous platform designed to accelerate discovery across the life, chemical, and materials sciences. Headquartered in Cambridge, Massachusetts, the company pairs generative AI models with automated robotic laboratories to formulate hypotheses, execute physical experiments, and analyze results. Its integrated system aims to streamline the scientific method, facilitating the rapid creation of novel therapeutics, advanced materials, and industrial compounds for global commercial partners.
lila.ai • HQ: Cambridge, United States • Biotechnology • Software • Operating Systems • 51-200 employees
HQ: Cambridge, United States • Biotechnology • Software • Operating Systems • 51-200 employees
Verified live · 1 day ago 1 month ago

Staff/Principal DevOps Engineer, AI Inference

$106k – $252k per year (Estimated) • Lead • In office (Cambridge, United States) • Full-Time • Principal
Python
Rust
AI/ML
CUDA
CUDA Toolkit
LLM
Triton
Triton Inference Server
vLLM
AWS Trainium
NCCL
TGI
AWQ
GPTQ
Quantization
DevOps
Amazon EC2
Amazon EKS
AWS
CI/CD
Helm
Kubernetes
Platform Engineering
SLI/SLO/SLA
Terraform
Amazon S3
IAM
Chaos Engineering
Incident Management
Cybersecurity
Least Privilege
Analytics
A/B Testing
Apply
Report

Aivar Innovations

aivar.tech
Aivar Innovations is a tech company delivering IT services with a focus on AI, ML, and cloud solutions to help businesses drive digital transformation by developing AI-powered platforms and voice AI solutions.
aivar.tech • Coimbatore • Bengaluru • Cloud Computing • Information Technology • Artificial Intelligence • Est. 2024
Coimbatore • Bengaluru • Cloud Computing • Information Technology • Artificial Intelligence • Est. 2024
$25k – $63k per year (Estimated) • Senior • 5+ years expIn office (Bengaluru, India) • Senior
Llama
AWS Trainium
DevOps
AIOps
Amazon EC2
Amazon EKS
ArgoCD
AWS
CI/CD
GCP
Grafana
Helm
Istio
Karpenter
Kubernetes
Kustomize
Prometheus
Terraform
IAM
Cybersecurity
Calico
Falco
Trivy
Apply
Report

Luma AI

lumalabs.ai
Luma AI is an artificial intelligence company that specializes in building advanced 3D visual technology and generative AI tools. Founded in 2021, the company is best known for its realistic text-to-3D models, NeRF (Neural Radiance Fields) capture app, and its high-quality video generation model, Dream Machine. By enabling creators and developers to easily capture, generate, and edit photorealistic 3D assets and video, it aims to make next-generation digital media creation accessible to everyone.
lumalabs.ai • HQ: Palo Alto, United States • Artificial Intelligence • Generative Image • Generative Video • Est. 2021
HQ: Palo Alto, United States • Artificial Intelligence • Generative Image • Generative Video • Est. 2021
Verified live · 5 hours ago 1 month ago

Tech Lead Manager, Inference

$220k – $432k per year (Estimated) • Lead • Remote/Hybrid (Redwood City, United States) • Full-Time • Staff
Python
C
C++
Rust
C++
PyTorch C++
C
FFmpeg
AI/ML
LLM
PyTorch
Quantization
SGLang
TensorRT
TensorRT-LLM
vLLM
CUDA
CUDA Toolkit
Multimodal AI
Ray
AWS Trainium
InfiniBand
NVLink
TPU
DevOps
Kubernetes
Apply
Report

Blockchain Capital

blockchaincapital.com
Blockchain Capital is one of the earliest dedicated digital asset venture firms, founded in 2013 in San Francisco. It has invested across many funds in exchanges, infrastructure, decentralised finance and consumer applications. The firm was also a pioneer of tokenised fund structures, issuing a security token to represent limited partner interests.
blockchaincapital.com • HQ: San Francisco, United States • Blockchain & Crypto • Est. 2013
HQ: San Francisco, United States • Blockchain & Crypto • Est. 2013
Open 115 daysVerified live · 1 hour ago 3 months ago

Senior AI Compute Infrastructure Engineer

$90k – $170k per year (Estimated) • Senior • 5+ years expPanama • Remote (United Kingdom, Argentina, Brazil, Bulgaria, Canada) • Full-Time • Senior
Python
C++
Rust
AI/ML
KServe
Ray
Ray Serve
TensorRT
Triton
Triton Inference Server
vLLM
TorchServe
CUDA
CUDA Toolkit
DeepSpeed
AWS Trainium
FSDP
Megatron-LM
NCCL
TPU
DevOps
Kubernetes
AWS
Apply
Report

Zyphra

zyphra.com
COMPANY RESEARCH CLOUD Zyphra Cloud Login Two sides. Two sides.
zyphra.com • HQ: San Francisco, United States • AI Infrastructure • Artificial Intelligence • LLM & Generative AI
HQ: San Francisco, United States • AI Infrastructure • Artificial Intelligence • LLM & Generative AI
Open 167 daysVerified live · 1 day ago 5 months ago

Research Engineer - AI Performance & Kernel Optimization

$156k – $342k per year (Estimated)In office (San Francisco, United States) • Relocation • Full-Time
CUDA Toolkit
CUDA
Triton
AWS Trainium
TPU
Mixture of Experts
DevOps
AWS
HPC
Apply
Report

Parspec

parspec.io
Parspec is a technology company that leverages AI to help sales agents and distributors by simplifying the process of discovering and sourcing the best available construction products and materials.
parspec.io • Bengaluru • San Mateo • Commerce • Marketplaces • Artificial Intelligence • Est. 2020
Bengaluru • San Mateo • Commerce • Marketplaces • Artificial Intelligence • Est. 2020
Open 174 daysVerified live · 11 hours ago 5 months ago

AI Ops Engineer

$30k – $75k per year (Estimated) • Senior • 5+ years expIn office (Bengaluru, India) • Bachelor's Degree • Full-Time • Senior
Python
Python
Asyncio
FastAPI
Databases
Apache Kafka
pgvector
Pinecone
Weaviate
PostgreSQL
Qdrant
AI/ML
AWS Bedrock
Kubeflow
LiteLLM
LLM
MLFlow
Portkey
Ray
vLLM
AWQ
CUDA
CUDA Toolkit
Embeddings
GPTQ
Hallucination
Langfuse
LoRA
Prompt Engineering
PyTorch
Quantization
SGLang
TensorRT
TensorRT-LLM
Triton
Triton Inference Server
PEFT
Amazon SageMaker
AWS Trainium
LLM Guardrails
LLMOps
NCCL
NVLink
TGI
Multimodal AI
AI Agents
DevOps
AIOps
Amazon EC2
Amazon EKS
ArgoCD
AWS
AWS Lambda
CI/CD
CloudFormation
Docker
GitHub Actions
Grafana
Kubernetes
OpenTelemetry
Prometheus
Terraform
Vector
Karpenter
Platform Engineering
Amazon EventBridge
Amazon S3
API Gateway
IAM
AWS Step Functions
GitHub
Cybersecurity
Least Privilege
Apply
Report
AWS Trainium Jobs - Remote & On-site
Frequently asked questions
How many AWS Trainium jobs are available on Alion?
There are currently 24 active listings from 11 hiring companies on Alion, refreshed daily.
What salary do AWS Trainium jobs pay?
The average salary is around $360,811 per year based on the current listings.
Are remote options available for AWS Trainium jobs?
Yes, many of these roles offer remote or hybrid work, and you can filter listings by remote, relocation and visa sponsorship.
How do I apply for AWS Trainium jobs?
Open any listing on Alion and apply directly with the employer through the provided link - no lengthy sign-up required.