JAX Jobs - Remote & On-site

JAX jobs hiring now at vetted startups and product companies on Alion, updated daily. Average salary around $276,623/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
Leadership
Networking
Product
Design
Analytics
Support
Solutions
Enterprise Apps
Quantum

Apple

apple.com
apple.com • HQ: Cupertino, United States • Smartwatches & Fitness Trackers • Operating Systems • Tablets • Laptops • Smartphones • 5000+ employees • Est. 1976 • Trust B • Hiring 3247, 312 new
HQ: Cupertino, United States • Smartwatches & Fitness Trackers • Operating Systems • Tablets • Laptops • Smartphones • 5000+ employees • Est. 1976 • Trust B • Hiring 3247, 312 new
Apple is an American multinational technology company founded in 1976 by Steve Jobs, Steve Wozniak and Ronald Wayne, and headquartered in Cupertino, California. It designs and sells consumer hardware including the iPhone, Mac, iPad, Apple Watch, AirPods and Vision Pro, together with the operating systems and silicon that run them. A growing services division built around the App Store, iCloud, Apple Music, Apple TV+ and Apple Pay now contributes a large share of profit, making Apple one of the most valuable companies in the world.
Sabih Khan · Chief Operating Officer · decision-maker
Kevan Parekh · Chief Financial Officer · decision-maker
Tim Cook · Chief Executive Officer · decision-maker
Steve Wozniak · Co-Founder · founder
No careers mailbox known
Verified live · 1 hour ago Visa sponsor 2 days ago Hiring now · 70%

Machine Learning Engineer, Creative Intelligence Systems

≈ $174k – $385k per year (Estimated) • Junior • In office (Cupertino, United States) • Bachelor's Degree
Python
C++
Objective-C
C++
PyTorch C++
AI/ML
JAX
Computer Vision
PyTorch
Machine Learning
CUDA Toolkit
AI Agents
Gaussian Splatting
CUDA
Mobile
ARKit
Metal
Robotics
Localization
+7
Apply
Verified live · 1 hour ago Visa sponsor 2 days ago Hiring now · 70%

ML Data & Eval Engineer, Siri Agent Modeling

≈ $223k – $420k per year (Estimated) • Middle • 4+ years exp • In office (Cupertino, United States) • Bachelor's Degree • Middle
Python
AI/ML
Spark
JAX
AI Agents
PyTorch
Ray
Edge AI
Quantization
Kubeflow
LLM
Synthetic Data
DevOps
Docker
Kubernetes
+5
Apply
Verified live · 1 hour ago Visa sponsor 4 days ago Hiring now · 70%

Machine Learning Engineer, Data for Foundation Models

≈ $223k – $420k per year (Estimated) • Middle • 3+ years exp • In office (Seattle, United States) • Master's Degree • Middle
Hadoop
JAX
PyTorch
Ray
Synthetic Data
Machine Learning
Apply

YO AI Labs

yoailabs.com
yoailabs.com • HQ: San Francisco, United States • AI Evaluation & Observability • AI Training Data & Annotation • 5000+ employees • Trust A • Hiring 2183, 273 new
HQ: San Francisco, United States • AI Evaluation & Observability • AI Training Data & Annotation • 5000+ employees • Trust A • Hiring 2183, 273 new
YO AI Labs is an artificial intelligence platform specializing in AI model training, evaluation, and data annotation services. Headquartered in San Francisco, California, United States, the company connects domain experts and remote specialists across the globe to review, test, and benchmark generative AI model outputs. Through its decentralized workforce solutions and structured evaluation workflows, it enables frontier technology developers to enhance the accuracy, safety, and reliability of next-generation artificial intelligence systems.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 2 days ago Hiring now · 90%

ML Engineer - Remote

≈ $183k – $401k per year (Estimated) • New York • Remote (United States) • Master's Degree
Python
AI/ML
llama.cpp
vLLM
JAX
SciPy
SGLang
Transformers
NumPy
PyTorch
Hugging Face
Machine Learning
+3
Apply

Crusoe

crusoe.ai
crusoe.ai • HQ: Denver, United States • Data Centers & Colocation • AI Compute & Inference • 1001-5000 employees • Est. 2018 • Trust B • Hiring 291, 12 new
HQ: Denver, United States • Data Centers & Colocation • AI Compute & Inference • 1001-5000 employees • Est. 2018 • Trust B • Hiring 291, 12 new
Crusoe is an energy and cloud company founded in 2018 that builds data centres next to stranded and low-carbon power sources. It started by converting flared natural gas into computing capacity and has become a large supplier of graphics processing capacity for artificial intelligence training and inference. The company develops sites, energy systems and its own managed AI cloud.
Cully Cavness · Co-founder and President · decision-maker
Chase Lochmiller · Co-founder and Chief Executive Officer · decision-maker
Michael Gordon · Executive Officer · decision-maker
Matthew DeNezza · Executive Officer · decision-maker
Careers mailbox is known
Top 25% payVerified live · 11 min ago 2 days ago Likely open · 63%

Senior Staff Software Engineer, Storage

$245k – $290k per year • Equity • Staff+ • 12+ years exp • In office (Sunnyvale, United States) • Master's Degree • Full-Time • Staff
Rust
C++
C++
TensorFlow C++
PyTorch C++
Databases
Redis
Apache Kafka
AI/ML
JAX
TensorFlow
PyTorch
DevOps
Amazon S3
GCP
Azure
AWS
+5
Apply

Apptronik

apptronik.com
apptronik.com • HQ: Austin, United States • Robotics AI • Robotic Components & Sensors • Humanoid Robots • 501-1000 employees • Est. 2016 • Startup • Trust B • Hiring 79, 3 new
HQ: Austin, United States • Robotics AI • Robotic Components & Sensors • Humanoid Robots • 501-1000 employees • Est. 2016 • Startup • Trust B • Hiring 79, 3 new
Apptronik is an American robotics company founded in 2016 out of the Human Centered Robotics Laboratory at the University of Texas, and it develops Apollo, a general-purpose humanoid robot for logistics and manufacturing work. Its engineering heritage is in actuators: the team spent years building the force-controlled actuators that let a robot work safely alongside people, which is a different problem from position-controlled industrial arms behind a cage. Headquartered in Austin and backed by Google and Mercedes-Benz among others, it is piloting Apollo with manufacturers and logistics operators rather than selling it openly.
Jeffery Cardenas · Executive Officer, Director · decision-maker
Nicholas Paine · Executive Officer, Director · decision-maker
Howard Morgan · Director
Clarence Daugherty IV · Director
No careers mailbox known
Top 25% payVerified live · 1 hour ago Visa sponsor 2 days ago Hiring now · 90%

Senior Perception Learning Engineer

$220k – $250k per year • Senior • In office (Sunnyvale, Austin, United States) • Master's Degree • Senior
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
OpenCV
YOLO
Reinforcement Learning
JAX
Multimodal AI
Computer Vision
Detectron2
TensorFlow
PyTorch
CLIP
DINOv2
Synthetic Data
World Models
Quantization
TensorRT
ONNX Runtime
DevOps
CI/CD
Robotics
Isaac Sim
SLAM
Localization
Sensor Fusion
Visual-Inertial Odometry
Reinforcement Learning
ROS
ORB-SLAM3
Cartographer
GStreamer
Kimera
+24
Apply
Top 25% payVerified live · 1 hour ago Visa sponsor 2 months ago Likely open · 50%

Senior Reinforcement Learning Engineer

$230k – $260k per year • Senior • 5+ years exp • In office (Sunnyvale, Austin, United States) • Master's Degree • Senior
Python
C++
C++
PyTorch C++
AI/ML
Fine-tuning
Reinforcement Learning
JAX
PyTorch
Robotics
Isaac Gym
MuJoCo
Teleoperation
Sim-to-Real
Imitation Learning
Reinforcement Learning
+5
Apply

ADBI

adbi.fr
adbi.fr • HQ: Paris, France • Analytics & BI Consulting • Data Integration & ETL Tools • Hiring 1, 1 new
HQ: Paris, France • Analytics & BI Consulting • Data Integration & ETL Tools • Hiring 1, 1 new
ADBI is a French data management and data integration consultancy that improves the quality, governance and visualisation of client data. It works with tools such as Power BI, Qlik, Snowflake, Informatica and Talend, and staffs developers on client platforms. The company is headquartered in Paris, France.
No named decision-maker yet.
Careers mailbox is known
$45k – $51k per year • Hybrid (France) • Freelance
Java
TypeScript
JavaScript
Kotlin
Java
Maven
Spring Boot
Hibernate
Spring Security
Gradle
Liquibase
Kotlin
Mockito
Databases
MySQL
AI/ML
JAX
Frontend
RxJS
Angular
Mobile
Clean Architecture
DevOps
Rest API
Ansible
GitLab CI
CI/CD
Jenkins
Git
AWS
Kubernetes
Amazon EKS
QA
Swagger
Jest
Rest-Assured
+20
Apply
Report

Clariti

clariticloud.com
clariticloud.com • HQ: Vancouver, Canada • Digital Government • 51-200 employees • Est. 1985 • Hiring 11, 4 new
HQ: Vancouver, Canada • Digital Government • 51-200 employees • Est. 1985 • Hiring 11, 4 new
Clariti builds community development software for cities and counties, covering permitting, licensing and plan review workflows for local governments. It is adding AI to its products and hires software engineers, engagement managers, professional services directors and product designers. The company is headquartered in Vancouver, Canada.
No named decision-maker yet.
No careers mailbox known
Verified live · 2 hours ago 2 days ago Hiring now · 90%

Senior Applied AI Engineer

≈ $106k – $201k per year (Estimated) • Senior • 7+ years exp • Remote (Canada) • Full-Time • Senior
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
Function Calling
Diffusion Models
Computer Vision
AI Agents
NLP
TensorFlow
PyTorch
LLM
LLM Guardrails
Interpretability
Tool Use
Model Distillation
Machine Learning
DevOps
GCP
Azure
CI/CD
Git
AWS
+14
Apply
tri.global • HQ: Los Altos, United States • Autonomous Driving • Robotics AI • Research Institutes • Est. 2016 • Trust B • Hiring 12, 1 new
HQ: Los Altos, United States • Autonomous Driving • Robotics AI • Research Institutes • Est. 2016 • Trust B • Hiring 12, 1 new
Toyota Research Institute conducts research in robotics, automated driving and materials discovery. Its teams work on large behaviour models for manipulation and on automated driving projects together with Woven by Toyota. The institute is headquartered in Los Altos, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago Visa sponsor 2 days ago Likely open · 63%

Postdoctoral Researcher, Post-Training and Continual Adaptation for Robotics

≈ $117k – $281k per year (Estimated) • Hybrid (Cambridge, United States) • PhD
Reinforcement Learning
JAX
PyTorch
Post-training
Machine Learning
Robotics
Imitation Learning
Reinforcement Learning
Apply

Kapital Bank

kapitalbank.az
kapitalbank.az • HQ: Baku, Azerbaijan • Commercial & Retail Banks • Est. 2004 • Hiring 26, 18 new
HQ: Baku, Azerbaijan • Commercial & Retail Banks • Est. 2004 • Hiring 26, 18 new
Headquartered in Baku, Azerbaijan, Kapital Bank is the country’s largest commercial and retail financial institution by customer base and branch network. Tracing its heritage back to 1874 as the Baku Savings Bank and officially rebranded as Kapital Bank in 2004 following the restructuring and privatization of post-Soviet universal banking assets, the bank operates as a core financial services subsidiary of PASHA Holding. The institution delivers a full spectrum of retail, SME, and corporate banking solutions, acting as a critical pillar for public finance by managing state pensions, social payments, and treasury disbursement programs.
No named decision-maker yet.
No careers mailbox known
3 days ago Hiring now · 90%

Full-Stack Developer - Future Lab

In office (Baku, Azerbaijan)
Java
Kotlin
SQL
Swift
Java
Maven
Spring Boot
Hibernate
Spring MVC
Spring Data JPA
Spring Security
Spring Cloud
Gradle
Kotlin
Retrofit
Hilt
Dagger
Kotlin Coroutines
OkHttp
Databases
PostgreSQL
Redis
Oracle
Apache Kafka
AI/ML
JAX
Mobile
JUnit
Jetpack Compose
UIKit
SwiftUI
Combine
MVI
Android SDK
Core Data
MVVM
Room
Firebase
CocoaPods
XCTest
Clean Architecture
Espresso
Push Notifications
WorkManager
DataStore
ProGuard
Crashlytics
ViewModel
DevOps
Rest API
Consul
CI/CD
Docker
Kubernetes
Cybersecurity
LDAP
QA
XCUITest
+42
Apply

Mecka

mecka.ai
mecka.ai • New York • Toronto • San Francisco • Synthetic Data • AI Training Data & Annotation • 201-500 employees • Startup • Trust A • Hiring 50, 4 new
New York • Toronto • San Francisco • Synthetic Data • AI Training Data & Annotation • 201-500 employees • Startup • Trust A • Hiring 50, 4 new
Mecka is a data infrastructure company that captures, structures and evaluates real-world activity into datasets for robotics and embodied AI teams. It designs and operates data capture, labeling and hardware-enabled systems used by AI labs, and hires software, firmware, computer vision and manufacturing engineers. The company is based in the United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 min ago 3 days ago Hiring now · 90%

Research Scientist – Multimodal Foundation Models

$185k – $210k per year • In office (Toronto, Canada, New York, San Francisco, United States) • Full-Time
Python
AI/ML
Fine-tuning
RLHF
Reinforcement Learning
JAX
Multimodal AI
PyTorch
Synthetic Data
DPO
SFT
PPO
GRPO
Post-training
Embodied AI
Machine Learning
Reward Modeling
Robotics
Reinforcement Learning
+9
Apply

NTT DATA

nttdata.com
nttdata.com • HQ: Tokyo, Japan • Data Centers & Colocation • Managed IT Services (MSP) • Financial Software & Embedded Finance • IT Consulting & Digital Transformation • Systems Integrators • 5000+ employees • Est. 1988 • Trust B • Hiring 1498, 443 new
HQ: Tokyo, Japan • Data Centers & Colocation • Managed IT Services (MSP) • Financial Software & Embedded Finance • IT Consulting & Digital Transformation • Systems Integrators • 5000+ employees • Est. 1988 • Trust B • Hiring 1498, 443 new
NTT DATA is a Tokyo system integrator founded in 1988 that spun out of the data communications arm of NTT. It is one of the largest information technology services firms in the world, building banking, public sector and payment systems and operating global consulting and managed services. The company runs the CAFIS payment network used across Japanese retail and banking.
Abhijit Dubey · 代表取締役社長 · decision-maker
No careers mailbox known
Verified live · 2 hours ago 3 days ago Likely open · 63%

AI Engineer

Middle • 3+ years exp • Sibiu • Remote (Romania) • Bachelor's Degree • Full-Time • Middle
Python
SQL
Databases
Weaviate
Chroma
pgvector
Pinecone
PostgreSQL
AI/ML
LangGraph
LangChain
LlamaIndex
Fine-tuning
Embeddings
Scikit-learn
JAX
Prompt Engineering
Function Calling
Chain-of-Thought
AI Agents
TensorFlow
PyTorch
LLM
RAG
Google ADK
Tokenization
Human-in-the-Loop
LLM Guardrails
Multi-Agent Systems
Tool Use
Vertex AI
Gemini
Semantic Search
Semantic Search
LLM Evaluation
DevOps
Rest API
gRPC
GCP
Azure
CI/CD
Git
AWS
OpenTelemetry
Datadog
Prometheus
Grafana
Management
Gmail
WhatsApp
+38
Apply
Verified live · 2 hours ago 19 days ago Fading · 37%

Java Developer – Gov&Tech

≈ $38k – $78k per year (Estimated) • In office (Bari, Italy) • Bachelor's Degree • Full-Time • Italian: C1
Java
Java
Spring Boot
Quarkus
Hibernate
AI/ML
JAX
DevOps
Rest API
Apply
Verified live · 2 hours ago 1 month ago Long shot · 21%

Lead AI Solution Architect

≈ $21k – $53k per year (Estimated) • Lead • 8+ years exp • Hybrid (Hanoi, Vietnam, 8am - 12pm) • Bachelor's Degree • Full-Time • Architect • English: B2
Python
Databases
Google BigQuery
BigQuery
AI/ML
Vertex AI
Embeddings
Prompt Engineering
Multimodal AI
Gemini
LLM
RAG
LLM Guardrails
LangGraph
LangChain
JAX
TensorFlow
Machine Learning
DevOps
GCP
CI/CD
Google GKE
Google Cloud Run
IAM
Kubernetes
FinOps
+15
Apply

Delos Data

delosdata.com
delosdata.com • HQ: Palo Alto, United States • Artificial Intelligence • AI Agents • 201-500 employees • Startup • Hiring 11, 4 new
HQ: Palo Alto, United States • Artificial Intelligence • AI Agents • 201-500 employees • Startup • Hiring 11, 4 new
Mixture of "X" Infrastructure (MoXI) Today's Agentic AI requires a mixture of hardware, a mixture of models, and a mixture of switches, links, bandwidths, and topologies, all serving one workload. WHO WE ARE A mixture of software, system and silicon people We're solving AI's interconnect problem.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 3 days ago Hiring now · 90%

System Software Engineer - AI

$85k – $127k per year • Hybrid (Montreal, Canada) • Bachelor's Degree • Full-Time
Python
C++
C++
PyTorch C++
AI/ML
DeepSpeed
vLLM
CUDA Toolkit
JAX
SGLang
PyTorch
LLM
CUDA
DevOps
Linux
+4
Apply
sei.cmu.edu • HQ: Pittsburgh, United States • Engineering Services • Research Institutes • 1001-5000 employees • Est. 1984 • Trust B • Hiring 53, 2 new
HQ: Pittsburgh, United States • Engineering Services • Research Institutes • 1001-5000 employees • Est. 1984 • Trust B • Hiring 53, 2 new
The Software Engineering Institute is a federally funded research and development center operated by Carnegie Mellon University in Pittsburgh, Pennsylvania, advancing software engineering, cybersecurity and AI engineering for the US Department of Defense and other agencies. Established in 1984, it is home to the CERT Division and its Coordination Center, which has handled vulnerability disclosure since 1988, and keeps an office near Washington in Arlington, Virginia. It hires cybersecurity and AI security researchers, AI red team engineers, reverse engineers, autonomous systems research scientists, software and embedded engineers, solutions engineers and data scientists.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 day ago 3 days ago Likely open · 62%

Associate Data Scientist

≈ $76k – $161k per year (Estimated) • Junior • 3+ years exp • In office (Pittsburgh, Arlington, United States) • Bachelor's Degree • Full-Time • Junior
Python
MATLAB
Python
FastAPI
Databases
Snowflake
Databricks
Google BigQuery
Trino
BigQuery
AI/ML
LangChain
CatBoost
Spark
LlamaIndex
XGBoost
Vertex AI
Fine-tuning
JAX
Prompt Engineering
SciPy
Multimodal AI
Computer Vision
AI Agents
NLP
Haystack
LightGBM
Llama
TensorFlow
NumPy
PyTorch
RAG
NeuralProphet
Time Series Forecasting
KServe
Triton
Amazon SageMaker
Human-in-the-Loop
Recommender Systems
Agentic Workflows
Machine Learning
+30
Apply
Open 92 daysVerified live · 1 day ago 3 months ago Long shot · 6%

Senior Data Scientist

≈ $108k – $206k per year (Estimated) • Senior • 10+ years exp • In office (Pittsburgh, Arlington, United States) • Bachelor's Degree • Full-Time • Senior
Python
MATLAB
Python
FastAPI
Databases
Snowflake
Databricks
Google BigQuery
Trino
AI/ML
LangChain
CatBoost
Spark
LlamaIndex
XGBoost
Vertex AI
Fine-tuning
JAX
Prompt Engineering
SciPy
Multimodal AI
Computer Vision
AI Agents
NLP
Haystack
LightGBM
Llama
TensorFlow
NumPy
PyTorch
RAG
NeuralProphet
Time Series Forecasting
KServe
Triton
Amazon SageMaker
Human-in-the-Loop
Recommender Systems
Machine Learning
+28
Apply
Open 162 daysVerified live · 1 day ago 5 months ago Long shot · 4%

Data Scientist

≈ $108k – $206k per year (Estimated) • Senior • 8+ years exp • In office (Pittsburgh, Arlington, United States) • Bachelor's Degree • Full-Time • Staff
Python
MATLAB
Python
FastAPI
Databases
Snowflake
Databricks
Google BigQuery
Trino
BigQuery
AI/ML
LangChain
CatBoost
Spark
LlamaIndex
XGBoost
Vertex AI
Fine-tuning
JAX
Prompt Engineering
SciPy
Multimodal AI
Computer Vision
AI Agents
NLP
Haystack
LightGBM
Llama
TensorFlow
NumPy
PyTorch
RAG
NeuralProphet
Time Series Forecasting
KServe
Triton
Amazon SageMaker
Human-in-the-Loop
Recommender Systems
Agentic Workflows
Machine Learning
+30
Apply

TrainLoop

trainloop.ai
trainloop.ai • HQ: San Francisco, United States • 11-50 employees • Est. 2024 • Startup • Hiring 4, 4 new
HQ: San Francisco, United States • 11-50 employees • Est. 2024 • Startup • Hiring 4, 4 new
Headquartered in San Francisco, California, TrainLoop is an artificial intelligence research and product lab specializing in post-training reasoning models for long-horizon tasks. Founded in 2024, the enterprise develops advanced reinforcement learning and fine-tuning solutions to transform general foundation models into domain-specific experts for industries such as biotechnology, logistics, and financial services. By embedding human expert intuition and rigorous preference alignment into proprietary architectures, the company enables organizations to deploy highly reliable AI agents for complex, mission-critical workflows.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 3 hours ago 3 days ago Hiring now · 90%

Research Engineer

$150k – $250k per year • Equity 0.5–1.5% • Middle • 3+ years exp • In office (San Francisco, United States) • Full-Time • Middle
JAX
PyTorch
Apply

Digital Harbor Foundation

digitalharbor.org
digitalharbor.org • HQ: Baltimore, United States • 11-50 employees • Est. 2011 • Hiring 2, 2 new
HQ: Baltimore, United States • 11-50 employees • Est. 2011 • Hiring 2, 2 new
Founded in 2011, the Digital Harbor Foundation is a non-profit organization dedicated to promoting digital equity, innovative education, and technology for the public good. Headquartered in Baltimore, Maryland, the foundation provides youth programs, makerspace learning, and resources to bridge the digital divide. Through its strategic initiatives and community projects, it empowers local residents and supports mission-driven tech ventures across the country.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 3 days ago Hiring now · 86%

Research Scientist - Digital Harbor Foundation

$120k – $150k per year • Remote (likely United States) • PhD • English (Optional)
Python
Python
Dask
AI/ML
JAX
PyTorch
Machine Learning
Apply

NVIDIA

nvidia.com
nvidia.com • HQ: Santa Clara, United States • Computer Components • Servers & Data Center Hardware • AI Chips & Accelerators • Processors, MCUs & AI Chips • 5000+ employees • Est. 1993 • Trust B • Hiring 2341, 203 new
HQ: Santa Clara, United States • Computer Components • Servers & Data Center Hardware • AI Chips & Accelerators • Processors, MCUs & AI Chips • 5000+ employees • Est. 1993 • Trust B • Hiring 2341, 203 new
NVIDIA is an American technology company founded in 1993 that invented the graphics processing unit and has become the dominant supplier of accelerated computing platforms for artificial intelligence. Its portfolio spans data centre GPUs and systems built on the Hopper and Blackwell architectures, GeForce consumer graphics, automotive and robotics platforms, high-speed networking acquired with Mellanox, and the CUDA software stack that binds the ecosystem together. Headquartered in Santa Clara, California, the company sells to cloud providers, enterprises, research institutions and gamers worldwide and is one of the most valuable listed businesses on the Nasdaq.
Jensen Huang · Co-Founder, President & Chief Executive Officer · decision-maker
Jay Puri · Executive Vice President, Worldwide Field Operations · decision-maker
Colette Kress · Executive Vice President & Chief Financial Officer · decision-maker
Curtis Priem · Co-Founder & former Chief Technical Officer · founder
Careers mailbox is known
Top 25% payVerified live · 1 hour ago 3 days ago Likely open · 63%

Principal AI and ML Infra Software Engineer, GPU Clusters

$272k – $431k per year • Lead • 15+ years exp • In office (Santa Clara, Redmond, United States) • Bachelor's Degree • Full-Time • Principal
Python
AI/ML
JAX
PyTorch
FSDP
NVIDIA NeMo
InfiniBand
DevOps
GCP
SLURM
Azure
AWS
Docker
Kubernetes
HPC
+5
Apply
Top 25% payVerified live · 1 hour ago 4 days ago Likely open · 63%

Senior Software Engineer - Storage

$152k – $242k per year • Senior • 5+ years exp • In office (Santa Clara, Redmond, United States) • Bachelor's Degree • Full-Time • Senior
Python
C++
C++
PyTorch C++
AI/ML
JAX
PyTorch
NVIDIA NeMo
DevOps
GCP
SLURM
Azure
AWS
Kubernetes
HPC
+4
Apply
Verified live · 1 hour ago 5 days ago Likely open · 63%

Senior Software Engineer, JAX

$75k – $131k per year • Senior • 6+ years exp • Switzerland • United Kingdom • Poland • Germany • France • Remote (Switzerland) • Bachelor's Degree • Full-Time • Senior
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
Scikit-learn
JAX
TensorFlow
NumPy
PyTorch
Machine Learning
+2
Apply

GSK

gsk.com
gsk.com • HQ: London, United Kingdom • Oncology Therapeutics • Infectious Disease Therapeutics • Vaccines • Prescription Drugs • 1001-5000 employees • Est. 2000 • Trust A • Hiring 411, 125 new
HQ: London, United Kingdom • Oncology Therapeutics • Infectious Disease Therapeutics • Vaccines • Prescription Drugs • 1001-5000 employees • Est. 2000 • Trust A • Hiring 411, 125 new
GSK is a British multinational pharmaceutical company that develops and manufactures prescription medicines and vaccines, with research focused on infectious diseases, HIV, respiratory and immunology, and oncology. It was formed in 2000 through the merger of Glaxo Wellcome and SmithKline Beecham and spun off its consumer healthcare business as Haleon in 2022. Listed in London and New York, it runs research, manufacturing and commercial operations worldwide. The company is headquartered in London, England, United Kingdom.
Anastasia Maksimova · HR
Careers mailbox is known
Top 25% payVerified live · 11 hours ago 3 days ago Hiring now · 70%

Applied AI Engineer

$136k – $227k per year • Junior • 2+ years exp • Providence • Cambridge • Remote (United States) • Bachelor's Degree • Full-Time • Junior
Python
AI/ML
Scikit-learn
JAX
Computer Vision
AI Agents
NLP
TensorFlow
NumPy
PyTorch
LLM
Knowledge Graph
Edge AI
Multi-Agent Systems
Machine Learning
LangGraph
AutoGen
Weights & Biases
LangChain
MLFlow
RAG
DevOps
GCP
Azure
AWS
Docker
Kubernetes
CI/CD
Git
GitHub
+20
Apply

AstraZeneca

astrazeneca.com
astrazeneca.com • HQ: Cambridge, United Kingdom • Rare Disease Therapeutics • Cardiometabolic Therapeutics • Oncology Therapeutics • Prescription Drugs • 1001-5000 employees • Est. 1999 • Trust A • Hiring 1031, 142 new
HQ: Cambridge, United Kingdom • Rare Disease Therapeutics • Cardiometabolic Therapeutics • Oncology Therapeutics • Prescription Drugs • 1001-5000 employees • Est. 1999 • Trust A • Hiring 1031, 142 new
AstraZeneca is a British-Swedish pharmaceutical company formed in 1999 by the merger of Astra of Sweden and Zeneca of the United Kingdom, and now one of the largest drug makers in the world by revenue. Its portfolio is concentrated in oncology with Tagrisso, Imfinzi and Enhertu, alongside cardiovascular, renal and metabolic products including Farxiga, respiratory and immunology medicines, and a rare disease business built on the acquisition of Alexion. Headquartered in Cambridge in England and listed in London, Stockholm and New York, it reports in United States dollars.
Aradhana Sarin · Chief Financial Officer · decision-maker
Pascal Soriot · Chief Executive Officer · decision-maker
Careers mailbox is known
Top 25% payVerified live · 9 hours ago 3 days ago Hiring now · 90%

Director, AI Research

$203k – $305k per year • Head+ • 8+ years exp • In office (Boston, United States) • PhD • Full-Time • Architect
Python
AI/ML
JAX
Multimodal AI
Function Calling
AI Agents
TensorFlow
PyTorch
Post-training
Interpretability
Tool Use
Machine Learning
Fine-tuning
RAG
Edge AI
+6
Apply
Verified live · 9 hours ago 25 days ago Fading · 31%

Associate Director, AI for Discovery — New Target Discovery

$145k – $217k per year • Head+ • 5+ years exp • In office (Cambridge, United States) • PhD • Full-Time • Architect
Python
AI/ML
JAX
AI Agents
PyTorch
LLM
Edge AI
Multi-Agent Systems
Multimodal AI
DevOps
AWS
+1
Apply
Verified live · 9 hours ago 26 days ago Fading · 27%

AI Science & Engineering Analyst

≈ $94k – $181k per year (Estimated) • Senior • 8+ years exp • In office (Gaithersburg, United States) • PhD • Full-Time • Staff
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
TensorFlow
PyTorch
Self-Supervised Learning
TPU
Machine Learning
Time Series Forecasting
Interpretability
DevOps
GCP
Azure
AWS
Robotics
Digital Twin
+7
Apply
cmu.edu • HQ: Pittsburgh, United States • Higher Education • 1001-5000 employees • Est. 1900 • Trust A • Hiring 171, 13 new
HQ: Pittsburgh, United States • Higher Education • 1001-5000 employees • Est. 1900 • Trust A • Hiring 171, 13 new
Carnegie Mellon University is a private research university in Pittsburgh, Pennsylvania, known for computer science, robotics, engineering, business, public policy and the arts. Founded in 1900 by Andrew Carnegie as the Carnegie Technical Schools, it also operates degree campuses in Doha, Qatar and Kigali, Rwanda, and runs the Software Engineering Institute, a federally funded research center with offices in Pittsburgh and Arlington, Virginia. Staff openings include cybersecurity and solutions engineers, software engineers, Robotics Institute research assistants and postdoctoral fellows, IT analysts, sponsored research accountants and early childhood educators.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 day ago Visa sponsor 3 days ago Hiring now · 90%

Associate Data Scientist

≈ $76k – $161k per year (Estimated) • Junior • 3+ years exp • In office (Pittsburgh, Arlington, United States) • Bachelor's Degree • Full-Time • Junior
Python
MATLAB
Python
FastAPI
Databases
Snowflake
Databricks
Google BigQuery
Trino
BigQuery
AI/ML
LangChain
CatBoost
Spark
LlamaIndex
XGBoost
Vertex AI
Fine-tuning
JAX
Prompt Engineering
SciPy
Multimodal AI
Computer Vision
AI Agents
NLP
Haystack
LightGBM
Llama
TensorFlow
NumPy
PyTorch
RAG
NeuralProphet
Time Series Forecasting
KServe
Triton
Amazon SageMaker
Human-in-the-Loop
Recommender Systems
Agentic Workflows
Machine Learning
+30
Apply
Verified live · 1 day ago Visa sponsor 3 months ago Fading · 25%

Senior Data Scientist

≈ $108k – $206k per year (Estimated) • Senior • 10+ years exp • In office (Pittsburgh, Arlington, United States) • Bachelor's Degree • Full-Time • Senior
Python
MATLAB
Python
FastAPI
Databases
Snowflake
Databricks
Google BigQuery
Trino
BigQuery
AI/ML
LangChain
CatBoost
Spark
LlamaIndex
XGBoost
Vertex AI
Fine-tuning
JAX
Prompt Engineering
SciPy
Multimodal AI
Computer Vision
AI Agents
NLP
Haystack
LightGBM
Llama
TensorFlow
NumPy
PyTorch
RAG
NeuralProphet
Time Series Forecasting
KServe
Triton
Amazon SageMaker
Human-in-the-Loop
Recommender Systems
Agentic Workflows
Machine Learning
+30
Apply
Open 162 daysVerified live · 1 day ago Visa sponsor 5 months ago Long shot · 15%

Data Scientist

≈ $108k – $206k per year (Estimated) • Senior • 8+ years exp • In office (Pittsburgh, Arlington, United States) • Bachelor's Degree • Full-Time • Staff
Python
MATLAB
Python
FastAPI
Databases
Snowflake
Databricks
Google BigQuery
Trino
BigQuery
AI/ML
LangChain
CatBoost
Spark
LlamaIndex
XGBoost
Vertex AI
Fine-tuning
JAX
Prompt Engineering
SciPy
Multimodal AI
Computer Vision
AI Agents
NLP
Haystack
LightGBM
Llama
TensorFlow
NumPy
PyTorch
RAG
NeuralProphet
Time Series Forecasting
KServe
Triton
Amazon SageMaker
Human-in-the-Loop
Recommender Systems
Agentic Workflows
Machine Learning
+30
Apply

Google DeepMind

deepmind.google
deepmind.google • HQ: London, United Kingdom • LLM & Generative AI • AI for Science • Foundation Models • AI Research Labs • 1001-5000 employees • Est. 2010 • Hiring 100, 24 new
HQ: London, United Kingdom • LLM & Generative AI • AI for Science • Foundation Models • AI Research Labs • 1001-5000 employees • Est. 2010 • Hiring 100, 24 new
Google DeepMind is the artificial intelligence research and engineering division of Alphabet, formed in 2023 by merging the London laboratory founded in 2010 with the Google Brain team. It builds frontier foundation models such as the Gemini family, generative media systems including Veo and Imagen, and scientific tools like AlphaFold, whose protein structure predictions earned a share of the 2024 Nobel Prize in Chemistry. The unit pairs long-horizon research on reinforcement learning and general intelligence with product delivery across Google Search, Workspace, Android and Google Cloud.
Shane Legg · Co-Founder & Chief AGI Scientist · decision-maker
Demis Hassabis · Co-Founder & Chief Executive Officer · decision-maker
Lila Ibrahim · Chief Operating Officer · decision-maker
Koray Kavukcuoglu · Chief Technology Officer · decision-maker
No careers mailbox known
Verified live · 1 hour ago 3 days ago Hiring now · 90%

Research Scientist for Gemini Horizon

≈ $129k – $284k per year (Estimated) • Junior • 2+ years exp • In office (Mountain View, United States) • Master's Degree • Junior
Python
C++
C++
PyTorch C++
AI/ML
NLP
Gemini
LLM
Computer Use
Machine Learning
JAX
PyTorch
+2
Apply
Verified live · 1 hour ago 3 days ago Hiring now · 90%

Software Engineer, ML Accelerators

≈ $133k – $295k per year (Estimated) • Junior • 2+ years exp • In office (Mountain View, United States) • Bachelor's Degree • Junior
Python
C++
C++
PyTorch C++
AI/ML
JAX
PyTorch
XLA
Machine Learning
Apply
Verified live · 1 hour ago 3 days ago Hiring now · 90%

Staff Research Scientist, Gemini Research

≈ $211k – $430k per year (Estimated) • Staff+ • 4+ years exp • In office (Mountain View, United States) • PhD • Staff
JAX
AI Agents
PyTorch
Gemini
Flax
Post-training
Pre-training
Machine Learning
Apply

Bayern

bayern.de
bayern.de • Munich • Garching bei München • Digital Government • Hiring 8, 3 new
Munich • Garching bei München • Digital Government • Hiring 8, 3 new
Offizielles Landesportal der Bayerischen Staatsregierung: Aktuelle Meldungen, Infos zum Freistaat Bayern, Politikthemen.
No named decision-maker yet.
Careers mailbox is known
3 days ago Hiring now · 90%

Java Entwickler (w/m/d)

≈ $69k – $133k per year (Estimated) • Hybrid (Freyung, Germany) • Full-Time
Java
AI/ML
JAX
Apply
Machines can read this selection · checked 18:16 UTC
JAX Jobs - Remote & On-site
Frequently asked questions
How many JAX jobs are available on Alion?
There are currently 525 active listings from 241 hiring companies on Alion, refreshed daily.
What salary do JAX jobs pay?
The average salary is around $276,623 per year based on the current listings.
Are remote options available for JAX 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 JAX jobs?
Open any listing on Alion and apply directly with the employer through the provided link - no lengthy sign-up required.