JAX Jobs - Remote & On-site

JAX jobs at vetted startups and product companies focused on high-performance ML and research, updated daily. These roles often pay competitively and support remote or hybrid work arrangements. 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 · 2 hours ago 5 months ago

Research Engineer, Machine Learning (RL Velocity)

$501k per year • In office (London, United Kingdom) • Visa sponsorship • Bachelor's Degree
JAX
Multimodal AI
PyTorch
Anthropic
Post-training
Pre-training
Apply
Open 343 daysTop 25% pay 12 months ago

Performance Engineer, GPU

$280k per year • In office (San Francisco, United States) • Visa sponsorship • Bachelor's Degree
Claude
CUDA Toolkit
Flash Attention
JAX
Multimodal AI
PyTorch
Quantization
CUDA
Triton
Anthropic
NCCL
NVLink
Apply
$500k per year • In office (San Francisco, United States) • Visa sponsorship • Bachelor's Degree
C++
Python
Rust
C++
PyTorch C++
TensorFlow C++
AI/ML
Claude
JAX
LLM
Multimodal AI
PyTorch
Reinforcement Learning
TensorFlow
Anthropic
Red Teaming
AI Agents
Function Calling
DevOps
Kubernetes
Apply
Report

Flagship Pioneering

flagshippioneering.com
Flagship Pioneering is a life sciences venture creation and platform investment firm headquartered in Cambridge, Massachusetts. Founded in 2000 by CEO Noubar Afeyan, the firm operates on a novel institutional innovation model where it acts as a principal founder, funder, and operator of first-in-category biotechnology and sustainability companies.
flagshippioneering.com • HQ: Cambridge, United States • Biotechnology • Financial Services • Venture Capital • 501-1000 employees • Est. 2000
HQ: Cambridge, United States • Biotechnology • Financial Services • Venture Capital • 501-1000 employees • Est. 2000
Top 25% payVerified live · 1 hour ago 3 months ago

Principal Scientist, Machine Learning - Biomolecules

$208k – $286k per year • Lead • 5+ years expIn office (Cambridge, United States) • Master's Degree • Principal
Python
Python
FastAPI
Databases
PostgreSQL
Amazon Redshift
FAISS
pgvector
Snowflake
AI/ML
LLM
TensorFlow
AI Agents
DVC
Gradio
JAX
MLFlow
Prefect
PyTorch
Streamlit
Amazon SageMaker
DevOps
Kubernetes
Amazon EKS
AWS
CI/CD
Docker
GitHub Actions
Terraform
Vector
Amazon ECS
Amazon S3
GitHub
Cybersecurity
GDPR
Apply
Top 25% payVerified live · 1 hour ago 27 days ago

Principal Scientist, Machine Learning

$208k – $286k per year • Lead • 5+ years expIn office (Cambridge, United States) • Master's Degree • Principal
Python
Python
Pydantic
AI/ML
LLM
TensorFlow
AI Agents
AutoGen
CrewAI
JAX
LangChain
LangGraph
Pydantic AI
PyTorch
Vertex AI
Anthropic
OpenAI
DevOps
FinOps
Cybersecurity
GDPR
Apply
Report

Twilio

twilio.com
Twilio is a leading cloud communications platform that enables software developers to programmatically build, scale, and integrate real-time communication capabilities into applications via application programming interfaces (APIs). Headquartered in San Francisco, California, the company provides infrastructure for automated SMS messaging, voice calls, WhatsApp integration, video, and two-factor authentication without requiring traditional telecom hardware.
twilio.com • HQ: San Francisco, United States • Telecommunications • Software • Online Communication • 1001-5000 employees • Est. 2008
HQ: San Francisco, United States • Telecommunications • Software • Online Communication • 1001-5000 employees • Est. 2008
Verified live · 2 hours ago 3 months ago

Machine Learning Engineer

$34k – $100k per year (Estimated) • Junior • 2+ years exp • Remote (Spain) • Bachelor's Degree • Junior
Python
AI/ML
AutoGen
CrewAI
Fine-tuning
JAX
LangGraph
LLM
NLP
NLTK
Prompt Engineering
PyTorch
TensorFlow
Transformers
LangChain
Hugging Face
LLMOps
AI Agents
Mobile
Twilio
DevOps
AWS
Azure
GCP
Apply
Verified live · 2 hours ago 3 months ago

Machine Learning Engineer

$52k – $155k per year (Estimated) • Junior • 2+ years exp • Remote (Ireland) • Bachelor's Degree • Junior
Python
AI/ML
AutoGen
CrewAI
Fine-tuning
JAX
LangGraph
LLM
NLP
NLTK
Prompt Engineering
PyTorch
TensorFlow
Transformers
LangChain
Hugging Face
LLMOps
AI Agents
Mobile
Twilio
DevOps
AWS
Azure
GCP
Apply
Report

Tower Research Capital

tower-research.com
Tower Research Capital is a quantitative trading firm operating automated strategies globally. It builds its own trading infrastructure, exchange connectivity and research systems. The firm supports multiple independent trading teams under one platform.
tower-research.com • HQ: New York, United States • Artificial Intelligence • Financial AI • Est. 1998
HQ: New York, United States • Artificial Intelligence • Financial AI • Est. 1998
Top 25% payVerified live · 1 day ago 21 day ago

Machine Learning Performance Engineer (Inference)

$200k – $300k per year • Junior • 2+ years expRemote/Hybrid (New York, United States) • Part-Time • Junior
C++
Python
C++
PyTorch C++
AI/ML
JAX
Knowledge Distillation
ONNX
PyTorch
Quantization
TensorRT
Triton
DevOps
HPC
Apply
Verified live · 1 day ago 3 months ago

Machine Learning Engineer

Junior • 2+ years expRemote/Hybrid (Shanghai, China) • Part-Time • Junior
Python
AI/ML
JAX
PyTorch
TensorFlow
DevOps
Docker
Kubernetes
HPC
Apply
Verified live · 1 day ago 3 months ago

Machine Learning Engineer

$69k – $203k per year (Estimated) • Junior • 2+ years expRemote/Hybrid (Singapore) • Part-Time • Junior
Python
AI/ML
JAX
PyTorch
TensorFlow
DevOps
Docker
Kubernetes
HPC
Apply
Report

Huntington

huntington.com
About Us We're looking out for the life you're building At Huntington, we believe money should support your life—not complicate it. That's why we offer tools, guidance and care that help you feel more confident, more capable and more in control of your financial future.
huntington.com • Columbus • Houston • Pittsburgh • Dallas • Memphis • Higher Education • Government • Financial Services • 1001-5000 employees
Columbus • Houston • Pittsburgh • Dallas • Memphis • Higher Education • Government • Financial Services • 1001-5000 employees
Verified live · 1 day ago 11 days ago

IS Technical Specialist (Backend Java Dev)

$70k – $140k per year • Senior • 5+ years expAustinChicagoDallasHoustonDetroit • Remote (United States) • Bachelor's Degree • Full-Time • Senior
Java
SQL
Java
Spring Boot
Maven
Databases
Apache Kafka
Db2
Google BigQuery
Google Bigtable
MySQL
Oracle
PostgreSQL
AI/ML
JAX
DevOps
CI/CD
GCP
Rest API
Git
Google Cloud Run
Jenkins
Kubernetes
Terraform
QA
Swagger
Apply
Report

Domyn

domyn.com
Domyn is an artificial intelligence company headquartered in Milan, Italy, and founded in 2016 under the name iGenius. The company builds large language models and decision intelligence tools aimed at regulated industries, including sovereign models trained for specific languages and jurisdictions. It works with banks, insurers, and public sector bodies in Europe that need AI deployed inside their own infrastructure rather than on a public model provider.
domyn.com • HQ: Milan, Italy • Financial Consulting • Financial Services • Business Intelligence • Data & Analytics • Decision Intelligence • LLM & Generative AI • Artificial Intelligence • Est. 2016
HQ: Milan, Italy • Financial Consulting • Financial Services • Business Intelligence • Data & Analytics • Decision Intelligence • LLM & Generative AI • Artificial Intelligence • Est. 2016
Verified live · 1 day ago 2 months ago

Junior AI Research Engineer

$35k – $58k per year • Equity • Junior • 2+ years expIn office • Master's Degree • Full-Time • Junior
C++
Python
Rust
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
PyTorch
TensorFlow
Post-training
AI Agents
CUDA Toolkit
Knowledge Distillation
Quantization
TensorBoard
Weights & Biases
DevOps
HPC
Apply
Open 95 daysVerified live · 1 day ago 4 months ago

AI Research Engineer

$58k – $93k per year • Equity • Senior • 5+ years expIn office • PhD • Full-Time • Senior
C++
Python
Rust
C++
PyTorch C++
TensorFlow C++
AI/ML
CUDA Toolkit
DeepSpeed
JAX
PyTorch
Ray
TensorBoard
TensorFlow
Weights & Biases
Post-training
AI Agents
Knowledge Distillation
Quantization
DevOps
HPC
Apply
Report

Xero

xero.com
Xero (ASX: XRO) is a global cloud-based accounting software platform headquartered in Wellington, New Zealand. Founded in 2006 by Rod Drury and Hamish Edwards, the company provides small business owners, accountants, and bookkeepers with automated financial management software. Operating on a SaaS business model, Xero offers real-time bank reconciliation, automated invoicing, expense tracking, payroll processing, fixed asset management, and financial reporting.
xero.com • HQ: Wellington, New Zealand • Software • Financial Services • FinTech • 51-200 employees • Est. 2006
HQ: Wellington, New Zealand • Software • Financial Services • FinTech • 51-200 employees • Est. 2006
Verified live · 4 hours ago 12 days ago

Engineering Manager - Machine Learning (AI Products)

Lead • Remote/Hybrid (Auckland, New Zealand, Sydney, Melbourne, Australia) • Full-Time • Staff
JAX
Apply
Verified live · 4 hours ago 2 months ago

Staff Machine Learning Engineer

$158k – $274k per year (Estimated) • Equity • Staff+ • In office (Vancouver, Canada) • Full-Time • Staff
Python
AI/ML
JAX
Pydantic AI
AI Agents
DevOps
AWS
Kubernetes
Management
Slack
Apply
Report

People Inc.

people.inc
People Inc. is a digital and print media publisher headquartered in New York City, tracing its corporate history to Meredith Corporation, founded in 1902. The company owns brands including People, Better Homes & Gardens, Allrecipes, Food & Wine, Travel + Leisure, Verywell, and Investopedia, reaching a large share of the United States online audience. It took its current name in 2025 after operating as Dotdash Meredith, remains a subsidiary of IAC, and keeps offices in New York, Des Moines, and Birmingham.
people.inc • HQ: New York, United States • Content Creation • Advertising • Sales & Marketing • Publishing • Digital Media • Media & Entertainment • 501-1000 employees • Est. 1902
HQ: New York, United States • Content Creation • Advertising • Sales & Marketing • Publishing • Digital Media • Media & Entertainment • 501-1000 employees • Est. 1902
Top 25% payVerified live · 1 day ago 12 days ago

Senior Data Scientist

$175k – $190k per year • Senior • Remote/Hybrid (New York, United States) • Master's Degree • Full-Time • Senior
Python
SQL
Databases
Google BigQuery
Milvus
Pinecone
AI/ML
Computer Vision
Embeddings
JAX
LLM
NLP
NumPy
PyTorch
Scikit-learn
TensorFlow
Vertex AI
Amazon SageMaker
Feature Store
DevOps
GCP
Vector
Analytics
A/B Testing
Apply
Report

KLA Corporation

kla.com
KLA Corporation is a global technology leader that specializes in process control and yield management solutions for the semiconductor and electronics industries. Founded in 1975 and headquartered in Milpitas, California, the company manufactures advanced wafer inspection tools, metrology systems, and computational analytics software to help chipmakers detect defects and optimize manufacturing efficiency. Listed on the NASDAQ, it serves major semiconductor foundries, packaging providers, and electronics manufacturers worldwide.
kla.com • Ann Arbor • Hsinchu • Milpitas • Erfurt • Boise • Manufacturing • Hardware • Semiconductor Manufacturing • Semiconductors • 1001-5000 employees
Ann Arbor • Hsinchu • Milpitas • Erfurt • Boise • Manufacturing • Hardware • Semiconductor Manufacturing • Semiconductors • 1001-5000 employees
Top 25% payVerified live · 1 hour ago 12 days ago

Algorithm Engineer - Deep Learning

$136k – $232k per year • Equity • Middle • 3+ years expIn office (Milpitas, United States) • Master's Degree • Full-Time • Middle
C++
Java
Python
Rust
C++
PyTorch C++
TensorFlow C++
AI/ML
Computer Vision
JAX
Knowledge Distillation
Multimodal AI
NLP
PyTorch
TensorFlow
VLM
Cryptography
GPG
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 · 1 hour 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
Top 25% payVerified live · 1 hour ago 2 months ago

Research Engineer, Privacy

$380k – $445k per year • Remote/Hybrid (San Francisco, United States) • Relocation • Full-Time
Federated Learning
JAX
PyTorch
OpenAI
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 122 daysTop 25% pay 4 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

Mistral AI

mistral.ai
Mistral AI is a French artificial intelligence company founded in Paris in 2023 by former researchers from Google DeepMind and Meta. It builds efficient open-weight and commercial large language models, including the Mistral, Mixtral, Magistral and Codestral families, and distributes them through its own developer platform as well as the major cloud marketplaces. Positioned as Europe's leading frontier model developer, the company also ships the Le Chat assistant, on-premise deployment options for regulated industries and a sovereign compute offering for customers who must keep data inside the region.
mistral.ai • HQ: Paris, France • Code Intelligence • Machine Learning • AI Agents • Artificial Intelligence • LLM & Generative AI • 1001-5000 employees • Est. 2023
HQ: Paris, France • Code Intelligence • Machine Learning • AI Agents • Artificial Intelligence • LLM & Generative AI • 1001-5000 employees • Est. 2023
Verified live · 23 min ago 13 days ago

AI Scientist - Physics Models

$77k – $184k per year (Estimated)In office (Paris, France, Amsterdam, Netherlands, Linz, Austria, Munich, Berlin, Germany) • Relocation • Master's Degree • Full-Time • English: C1
Python
AI/ML
JAX
DevOps
HPC
Apply
Verified live · 23 min ago 2 months ago

Applied Scientist, EMEA

$61k – $105k per year (Estimated)Remote/Hybrid (Paris, France, Amsterdam, Netherlands, London, United Kingdom, Lausanne, Switzerland) • Relocation • Master's Degree • Full-Time • English: C1
Python
AI/ML
JAX
Mistral SDK
PyTorch
Post-training
Pre-training
Diffusion Models
Multimodal AI
Time Series Forecasting
Apply
Verified live · 23 min ago 2 months ago

AI Scientist

$77k – $183k per year (Estimated)In office (Paris, France, London, United Kingdom) • Relocation • Full-Time
Go
Java
Python
Rust
AI/ML
Fine-tuning
JAX
Multimodal AI
PyTorch
Ray
DevOps
Kubernetes
Apply
Report

Baseten

baseten.co
Baseten is an AI infrastructure platform designed to help developers and machine learning teams deploy, serve, and scale open-source and custom AI models. The platform provides performant, low-latency inference infrastructure alongside developer tools like Truss, an open-source model packaging framework. Headquartered in San Francisco, California, Baseten enables companies to run state-of-the-art models in production seamlessly without managing underlying cloud infrastructure.
baseten.co • HQ: San Francisco, United States • Hardware • Machine Learning • AI Infrastructure • Artificial Intelligence
HQ: San Francisco, United States • Hardware • Machine Learning • AI Infrastructure • Artificial Intelligence
Top 25% payVerified live · 12 hours ago 13 days ago

Forward Deployed Engineer (Training)

$200k – $400k per year • Junior • 1+ year expRemote/Hybrid (San Francisco, New York, United States) • Full-Time • Junior
Cursor
JAX
LLM
PyTorch
Ray
SGLang
TensorRT
TensorRT-LLM
vLLM
InfiniBand
Post-training
SFT
DevOps
Kubernetes
SLURM
Chips/EDA
PoC Library
Apply
Report

Washington Post

washingtonpost.com
The Washington Post is a major American daily newspaper and digital media company headquartered in Washington, D.C. Founded in 1877, it is one of the premier national publications in the United States, widely renowned for its political reporting, investigative journalism, and international coverage - notably winning 76 Pulitzer Prizes over its operational history.
washingtonpost.com • HQ: Washington, United States • Publishing • Media & Entertainment • News Agencies • 501-1000 employees
HQ: Washington, United States • Publishing • Media & Entertainment • News Agencies • 501-1000 employees
Verified live · 3 hours ago 13 days ago

Applied Machine Learning Scientist 2

$109k – $181k per year • Middle • 2+ years expIn office (Washington, United States) • Bachelor's Degree • Full-Time • Middle
Python
Databases
Google BigQuery
AI/ML
JAX
LLM
NLP
PyTorch
TensorFlow
Recommender Systems
Prompt Engineering
Spark
DevOps
AWS
GCP
Analytics
A/B Testing
Apply
Top 25% payVerified live · 3 hours ago 13 days ago

Senior Applied Machine Learning Scientist

$132k – $219k per year • Senior • 4+ years expIn office (Washington, United States) • Bachelor's Degree • Full-Time • Senior
Python
Databases
Google BigQuery
AI/ML
JAX
NLP
PyTorch
Semantic Search
TensorFlow
Recommender Systems
Semantic Search
AI Agents
LLM
Reinforcement Learning
Spark
VLM
RAG
DevOps
AWS
GCP
Analytics
A/B Testing
Apply
Report

Constellationenergy

constellationenergy.com
Discover how Constellation's strategy and purpose makes us the nation's largest producer of clean, reliable energy for homes, businesses and industries.
constellationenergy.com • HQ: Baltimore, United States • Fossil Fuels • Nuclear Energy • Energy & Utilities • Est. 2022
HQ: Baltimore, United States • Fossil Fuels • Nuclear Energy • Energy & Utilities • Est. 2022
Top 25% pay 13 days ago

Research Scientist

$180k – $250k per year • In office (San Francisco, United States) • Relocation • Full-Time
Python
AI/ML
Computer Vision
JAX
Multimodal AI
NLP
PyTorch
Time Series Forecasting
Apply
Report

Intuitive

intuitive.com
Intuitive is a global medical technology company headquartered in Sunnyvale, California, and a pioneer in robotic-assisted surgery. It develops, manufactures, and markets minimally invasive care systems, including the da Vinci surgical system and the Ion endoluminal system. The company helps physicians and hospitals perform complex procedures with enhanced precision, visualization, and control.
intuitive.com • HQ: Sunnyvale, United States • Health Care • Robotics • Medical Devices • Medical & Healthcare Robotics • 1001-5000 employees
HQ: Sunnyvale, United States • Health Care • Robotics • Medical Devices • Medical & Healthcare Robotics • 1001-5000 employees
Verified live · 2 hours ago 13 days ago

Senior Software Engineer – Simulation & ML Platform

$159k – $308k per year (Estimated) • Senior • 5+ years expIn office (San Francisco, United States) • Bachelor's Degree • Full-Time • Middle • English
C++
Java
Python
TypeScript
C++
PyTorch C++
TensorFlow C++
Python
Celery
Databases
Apache Kafka
RabbitMQ
Redis
AI/ML
Computer Vision
Synthetic Data
Time Series Forecasting
JAX
MLFlow
PyTorch
Ray
TensorFlow
Weights & Biases
DevOps
CI/CD
Docker
Kubernetes
SLURM
Game Dev
NVIDIA PhysX
Robotics
Isaac Lab
Isaac Sim
MuJoCo
Sim-to-Real
Gazebo
ROS
Apply
Verified live · 2 hours ago 3 months ago

Senior AI/ML Research Engineer (Computer Vision)

$156k – $303k per year (Estimated) • Senior • In office (Sunnyvale, United States) • Master's Degree • Full-Time • Middle • English
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
Computer Vision
JAX
Multimodal AI
PyTorch
Self-Supervised Learning
TensorFlow
Human-in-the-Loop
TensorRT
Edge AI
Robotics
Sim-to-Real
Apply
Verified live · 2 hours ago 3 months ago

Senior AI/ML Research Engineer – Model development

$158k – $305k per year (Estimated) • Senior • 5+ years expIn office (Sunnyvale, United States) • Master's Degree • Full-Time • Middle • English
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
Fine-tuning
JAX
Multimodal AI
PyTorch
Reinforcement Learning
Self-Supervised Learning
TensorFlow
Human-in-the-Loop
Robotics
Reinforcement Learning
Isaac Sim
OpenVLA
Sim-to-Real
Teleoperation
Apply
Report

Nimble Robotics

nimble.ai
Nimble (Nimble Robotics) is an AI-powered robotics and autonomous third-party logistics (3PL) company headquartered in San Francisco, California. Founded in 2017 by Simon Kalouche, the company develops fully autonomous fulfillment networks powered by intelligent picking, packing, and sorting robots ("Superhumanoids") alongside cloud supply chain management software.
nimble.ai • HQ: San Francisco, United States • Transportation & Logistics • Commerce • Logistics • Est. 2017
HQ: San Francisco, United States • Transportation & Logistics • Commerce • Logistics • Est. 2017
Top 25% payVerified live · 2 hours ago 14 days ago

Senior/Staff Software Engineer, Infrastructure (ML)

$210k – $300k per year • Staff+ • 4+ years expIn office (San Francisco, United States) • Bachelor's Degree • Staff
C++
Python
Rust
C++
PyTorch C++
AI/ML
CUDA Toolkit
JAX
PyTorch
DeepSpeed
FSDP
Megatron-LM
NCCL
DevOps
Kubernetes
Apply
Report

d-Matrix

d-matrix.ai
d-Matrix is a semiconductor technology company based in Santa Clara, California, and was founded in 2019. The company develops specialized AI inference hardware and software platforms, including its Corsair platform and 3DIMC architecture, designed to accelerate generative AI workloads in data centers. It operates as a venture-backed enterprise serving global hyperscalers and enterprises with a focus on reducing latency, power consumption, and total cost of ownership for large language models.
d-matrix.ai • HQ: Santa Clara, United States • Data Centers • Information Technology • LLM & Generative AI • Artificial Intelligence • AI Infrastructure • Hardware • Semiconductors • Est. 2019
HQ: Santa Clara, United States • Data Centers • Information Technology • LLM & Generative AI • Artificial Intelligence • AI Infrastructure • Hardware • Semiconductors • Est. 2019
Top 25% payVerified live · 25 min ago 14 days ago

Senior Staff LLM Inference Engineer

$195k – $285k per year • Staff+ • 10+ years expRemote/Hybrid (Santa Clara, United States) • Bachelor's Degree • Full-Time • Staff
C++
Python
AI/ML
CUDA
CUDA Toolkit
LLM
ONNX
Quantization
SGLang
TensorRT
TensorRT-LLM
Triton
vLLM
JAX
Mixture of Experts
Apply
$198k – $370k per year (Estimated) • Staff+ • 10+ years expSanta Clara • Remote (United States) • Visa sponsorship • Bachelor's Degree • Staff
C++
Python
AI/ML
CUDA Toolkit
LLM
ONNX
Quantization
SGLang
TensorRT
TensorRT-LLM
vLLM
JAX
Mixture of Experts
Apply
Report

RTB House

rtbhouse.com
RTB House is a global advertising technology company that provides a demand-side platform (DSP) powered entirely by proprietary deep learning algorithms. The company specializes in personalized retargeting, brand advertising, and full-funnel performance marketing for e-commerce and enterprise advertisers across global markets. By leveraging advanced AI to analyze user behavior in real time, RTB House delivers highly tailored display, video, and programmatic ad campaigns to optimize conversion rates and ROI.
rtbhouse.com • Warsaw • Dubai • São Paulo • Milan • Prague • Artificial Intelligence • Sales & Marketing • Advertising • 51-200 employees • Est. 2010
Warsaw • Dubai • São Paulo • Milan • Prague • Artificial Intelligence • Sales & Marketing • Advertising • 51-200 employees • Est. 2010
Verified live · 26 min ago 14 days ago

Staff Machine Learning Researcher

$109k – $195k per year (Estimated) • Staff+ • 5+ years expIn office (Warsaw, Poland) • Internship • Staff
Java
Python
Scala
Databases
Google BigQuery
AI/ML
JAX
NumPy
Pandas
PyTorch
TPU
DevOps
GCP
Analytics
A/B Testing
Apply
Report

Gilead Sciences

gilead.com
Gilead Sciences is a American biopharmaceutical company that discovers, develops, and commercializes innovative medicines for life-threatening diseases. Founded in 1987 and headquartered in Foster City, California, the company is globally recognized for transforming HIV treatment and advancing curative therapies for hepatitis C. Listed on the NASDAQ, it also focuses on developing targeted therapies in oncology, cell therapy, and inflammatory conditions worldwide.
gilead.com • Foster City • Cambridge • Mississauga • Shanghai • Madrid • Biopharma • Health Care • Pharmaceuticals • 1001-5000 employees • Est. 1996
Foster City • Cambridge • Mississauga • Shanghai • Madrid • Biopharma • Health Care • Pharmaceuticals • 1001-5000 employees • Est. 1996
Top 25% payVerified live · 11 hours ago 14 days ago

Senior Scientist, AI/ML (Biologics Design)

$169k – $219k per year • Senior • 2+ years expRemote/Hybrid (Foster City, United States) • PhD • Full-Time • Senior
Python
AI/ML
JAX
Multimodal AI
NumPy
PyTorch
Apply
Report
JAX Jobs - Remote & On-site
Frequently asked questions
JAX: How many jobs are available now?
There are 491 JAX jobs listed on Alion right now.
JAX: What is the average salary for these roles?
The average salary for JAX roles on Alion is $285,074 USD.
JAX: Are remote, relocation, or visa sponsorship options available?
Many JAX listings allow remote or hybrid schedules, and some positions may offer relocation or visa sponsorship-verify on each job page.
JAX: How do I apply for JAX jobs on Alion?
Open the JAX job posting and click Apply or follow the employer’s link; sign in to save jobs and enable alerts for new listings.