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
Software
Data
Security
DevOps & Cloud
Leadership
Product
Design
Fintech
Hardware & Chips
Defense
Aerospace
Biotech
Energy
Robotics
Automotive
MedTech
Science
allenai.org · HQ: Seattle, United States · Research Institutes · Foundation Models · Natural Language Processing · AI Research Labs · Est. 2014 · Trust A · Hiring 16, 1 new
HQ: Seattle, United States · Research Institutes · Foundation Models · Natural Language Processing · AI Research Labs · Est. 2014 · Trust A · Hiring 16, 1 new
Headquartered in Seattle, Washington, the Allen Institute for AI (AI2) is a non-profit research institute founded by Microsoft co-founder Paul Allen to conduct high-impact artificial intelligence research and engineering in the public interest. The organization focuses on advancing open science, developing large-scale natural language processing models, and building foundational AI systems for scientific discovery and environmental conservation. Additionally, it freely releases its datasets, open-source models, and research papers, alongside creating tools like Semantic Scholar to accelerate global scientific research.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 1 hour ago 1 month ago Hiring now · 89%

Research Engineer, Climate Modeling

$147k – $220k per year · Junior · 2+ years exp · In office (Seattle, United States) · Bachelor's Degree · Junior
Python
AI/ML
JAX
TPU
Machine Learning
DevOps
GitHub
GCP
Apply
Top 25% payVerified live · 1 hour ago 1 month ago Hiring now · 90%

Research Scientist, Climate Modeling

$167k – $261k per year · In office (Seattle, United States) · PhD
Python
AI/ML
JAX
TPU
Machine Learning
Apply
Top 25% payVerified live · 1 hour ago 4 months ago Fading · 32%

Senior Research Engineer, Olmo + Molmo

$174k – $261k per year · Senior · 4+ years exp · In office (Seattle, United States) · Bachelor's Degree · Senior
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
Function Calling
LLaVA
Agentic Workflows
TensorFlow
PyTorch
LLM
AI Agents
OpenAI
DevOps
Docker
GCP
AWS
+7
Apply

Amgen

amgen.com
amgen.com · HQ: Thousand Oaks, United States · Immunology & Inflammation Therapeutics · Cardiometabolic Therapeutics · Prescription Drugs · Biologics & Biosimilars · 5000+ employees · Est. 1980 · Trust A · Hiring 1133, 168 new
HQ: Thousand Oaks, United States · Immunology & Inflammation Therapeutics · Cardiometabolic Therapeutics · Prescription Drugs · Biologics & Biosimilars · 5000+ employees · Est. 1980 · Trust A · Hiring 1133, 168 new
Amgen is an American biotechnology company founded in Thousand Oaks in 1980 that became one of the first firms to turn recombinant protein research into large commercial medicines. Its history rests on Epogen and Neupogen, and its current portfolio spans bone health with Prolia and Evenity, inflammation with Otezla and Tezspire, cardiovascular disease with Repatha, oncology including the bispecific antibody Blincyto, and a biosimilars business built on its own manufacturing capacity. The company is listed on Nasdaq, is a component of the Dow Jones Industrial Average, and expanded further into rare disease through its acquisition of Horizon Therapeutics.
Jay Bradner · Chief Scientific Officer · decision-maker
Robert Bradway · Chairman and Chief Executive Officer · decision-maker
George Rathmann · Founding Chief Executive Officer · founder
Careers mailbox is known
Top 25% payVerified live · 1 hour ago Visa sponsor 1 month ago Likely open · 46%

Director, AI & Machine Learning

$243k – $328k per year · Head+ · 12+ years exp · Thousand Oaks · Remote (United States) · Bachelor's Degree · Full-Time · Architect
Python
Databases
Databricks
Snowflake
AI/ML
JAX
TensorFlow
PyTorch
Machine Learning
AI Agents
Function Calling
LLM Guardrails
DevOps
GCP
Azure
CI/CD
AWS
+6
Apply
Verified live · 1 hour ago 1 month ago Long shot · 24%

Senior Data Scientist - Protein Structure ML models

≈ $25k – $55k per year (Estimated) · Senior · 6+ years exp · In office (Hyderabad, India) · Bachelor's Degree · Full-Time · Senior
Python
AI/ML
Machine Learning
JAX
PyTorch
Life Sciences
Antibody Engineering
Apply

Target

target.com
target.com · HQ: Minneapolis, United States · Retail & Shop Systems · Online Groceries · 5000+ employees · Est. 1902 · NYSE: TGT · Trust A · Hiring 3675, 188 new
HQ: Minneapolis, United States · Retail & Shop Systems · Online Groceries · 5000+ employees · Est. 1902 · NYSE: TGT · Trust A · Hiring 3675, 188 new
Target is an American general merchandise retailer founded in 1902 as the Dayton Dry Goods Company, which opened the first discount store under the Target name in 1962. It runs roughly two thousand stores across the United States selling apparel, home goods, beauty, electronics, toys and groceries, and has built its position on design partnerships and owned brands that give it products competitors cannot stock. Headquartered in Minneapolis and listed on the New York Stock Exchange, it fulfils the large majority of digital orders from its own stores rather than from separate warehouses.
Marshall Gibbs · Executive Officer · decision-maker
Ross Shelleman · Executive Officer, Director · decision-maker
Heiko Hildebrandt · Executive Officer · decision-maker
Jim Lee · Chief Financial Officer · decision-maker
No careers mailbox known
Top 25% payVerified live · 9 hours ago 1 month ago Long shot · 17%

Lead Applied Data Scientist - Recommendations (Contextual Bandits, Reinforcement Learning)

$132k – $238k per year · Lead · 5+ years exp · Hybrid (Minneapolis, United States) · Master's Degree · Full-Time · Staff
SQL
Python
Python
bandit
AI/ML
Reinforcement Learning
JAX
Spark
PyTorch
Machine Learning
Apply
Top 25% payVerified live · 9 hours ago 2 months ago Long shot · 6%

Sr Applied Data Scientist - Personalization (applied ML, PyTorch, RecSys)

$98k – $176k per year · Senior · 3+ years exp · Hybrid (Brooklyn Park, United States) · Master's Degree · Full-Time · Senior
SQL
Python
AI/ML
JAX
Spark
Machine Learning
Management
Agile
Apply
Top 25% payVerified live · 9 hours ago 2 months ago Long shot · 6%

Sr Applied Data Scientist - Personalization (applied ML, PyTorch, ML Ops)

$98k – $176k per year · Senior · 3+ years exp · Hybrid (Brooklyn Park, United States) · Master's Degree · Full-Time · Senior
SQL
Python
AI/ML
JAX
Spark
Machine Learning
Management
Agile
Apply

Ec Lyon

ec-lyon.fr
ec-lyon.fr · Écully · Hiring 9, 1 new
Écully · Hiring 9, 1 new
No named decision-maker yet.
No careers mailbox known
$38k per year · In office (Écully, France) · PhD
JAX
Diffusion Models
PyTorch
Machine Learning
DevOps
Error Budget
SLI/SLO/SLA
Apply

Crick

crick.ac.uk
crick.ac.uk · London · 1001-5000 employees · Hiring 8, 4 new
London · 1001-5000 employees · Hiring 8, 4 new
No named decision-maker yet.
No careers mailbox known
Verified live · 10 hours ago 1 month ago Long shot · 19%

AI/ML Postdoctoral Fellow – F Rouhani lab

$64k per year · Lead · In office (London, United Kingdom) · PhD · Principal
Python
AI/ML
JAX
Multimodal AI
NumPy
PyTorch
Machine Learning
DevOps
HPC
Apply

Chai Discovery

chaidiscovery.com
chaidiscovery.com · HQ: San Francisco, United States · AI for Science · Foundation Models · Drug Discovery AI · 201-500 employees · Est. 2024 · Startup · Trust C · Hiring 13, 1 new
HQ: San Francisco, United States · AI for Science · Foundation Models · Drug Discovery AI · 201-500 employees · Est. 2024 · Startup · Trust C · Hiring 13, 1 new
Chai Discovery is a San Francisco research company founded in 2024 that builds foundation models for molecular structure and interaction prediction. Its Chai-1 model predicts complexes of proteins, small molecules, nucleic acids and antibodies, and was released for free non-commercial use alongside a commercial offering. The company is backed by OpenAI and Thrive Capital and targets the design of antibodies and other biologics.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago Visa sponsor 1 month ago Long shot · 21%

Research Engineer, ML Infrastructure

≈ $164k – $365k per year (Estimated) · In office (San Francisco, United States) · Full-Time
Python
AI/ML
Quantization
JAX
PyTorch
Apply
Likely inactiveVerified live · 3 hours ago Visa sponsor 2 years ago Long shot · 3%

Research Engineer

≈ $164k – $365k per year (Estimated) · In office (San Francisco, United States) · Full-Time
Python
AI/ML
Quantization
JAX
PyTorch
Apply

Slate Auto

slate.auto
slate.auto · HQ: Troy, United States · Automotive Manufacturing · 1001-5000 employees · Est. 2022 · Trust A · Hiring 136, 13 new
HQ: Troy, United States · Automotive Manufacturing · 1001-5000 employees · Est. 2022 · Trust A · Hiring 136, 13 new
Slate Auto is an American electric vehicle start-up headquartered in Troy, Michigan that is developing a low-cost, customizable electric pickup that converts into an SUV, using wraps and accessories in place of paint, with a starting price under $25,000. Founded in 2022 inside Re:Build Manufacturing and backed by investors including Jeff Bezos, it is preparing production at a plant in Warsaw, Indiana and has satellite offices in Louisville, Chicago, Phoenix and Los Angeles. It hires manufacturing process and vehicle integration engineers, plant and quality leaders, AI and mobile software engineers, product managers and supply chain and facilities staff.
No named decision-maker yet.
No careers mailbox known
1 month ago Long shot · 16%

Senior AI/ML Engineer

≈ $111k – $255k per year (Estimated) · Senior · 3+ years exp · Remote (United States) · Bachelor's Degree · Full-Time · Senior
Python
AI/ML
Fine-tuning
Embeddings
Scikit-learn
JAX
NLP
Gemini
Time Series
Anthropic
Hugging Face
Agentic Workflows
Physical AI
Computer Vision
OpenAI
PyTorch
RAG
LLM
Machine Learning
AI Agents
DevOps
GCP
Azure
AWS
+14
Apply
Verified live · 11 hours ago 1 month ago Fading · 27%

AI/ML Engineer

≈ $116k – $231k per year (Estimated) · Middle · 3+ years exp · United States · Remote (United States) · Bachelor's Degree · Full-Time · Middle
Python
AI/ML
Fine-tuning
Embeddings
Scikit-learn
JAX
NLP
Gemini
Time Series
Anthropic
Hugging Face
Agentic Workflows
Physical AI
Computer Vision
OpenAI
PyTorch
RAG
LLM
Machine Learning
DevOps
GCP
Azure
AWS
+13
Apply

Foxconn

foxconn.com
foxconn.com · HQ: New Taipei, Taiwan · Electronics Manufacturing · Contract Manufacturing · 1001-5000 employees · Est. 1974 · Hiring 389, 3 new
HQ: New Taipei, Taiwan · Electronics Manufacturing · Contract Manufacturing · 1001-5000 employees · Est. 1974 · Hiring 389, 3 new
Foxconn, formally Hon Hai Precision Industry, is a Taiwanese electronics contract manufacturer that assembles smartphones, servers, computers and other devices for global technology brands. Founded in 1974 and listed on the Taiwan Stock Exchange, it runs large manufacturing campuses in China, India, Vietnam, Mexico and other countries, and has expanded into AI servers, electric vehicles and semiconductors. The company is headquartered in New Taipei, Taiwan.
No named decision-maker yet.
No careers mailbox known
Verified live · 2 hours ago 1 month ago Fading · 25%

機器學習工程師 AI/ML Engineer(Data Science & AI Team)

In office (New Taipei, Taiwan)
C++
Java
Python
Python
pySpark
C++
TensorFlow C++
Databases
Databricks
AI/ML
Scikit-learn
JAX
SciPy
NLP
Gensim
MLlib
Transformers
Pandas
NumPy
Keras
BERT
NLTK
Time Series
Hugging Face
Amazon SageMaker
TensorFlow
Spark
PyTorch
LLM
Machine Learning
DevOps
Azure
AWS
Analytics
Matplotlib
Azure Data Factory
+22
Apply

ALTEN

alten.com
alten.com · HQ: Boulogne-Billancourt, France · IT Consulting & Digital Transformation · Engineering Services · 5000+ employees · Est. 1988 · Staffing · EOR · Trust B · Hiring 796, 194 new
HQ: Boulogne-Billancourt, France · IT Consulting & Digital Transformation · Engineering Services · 5000+ employees · Est. 1988 · Staffing · EOR · Trust B · Hiring 796, 194 new
ALTEN is a French engineering and technology consultancy founded in 1988 that supplies project engineers to companies developing complex products and systems. Its consultants work inside client organisations across automotive, aerospace, defence, rail, energy, telecommunications, life sciences and finance, covering embedded software, mechanical and electrical design, simulation, testing, certification and information systems. Listed on Euronext Paris and headquartered in Boulogne-Billancourt, the group employs tens of thousands of engineers across more than thirty countries and has grown through a long sequence of acquisitions of specialist engineering firms.
Simon Azoulay · Founder, Chairman and Chief Executive Officer · decision-maker
Gerald Attia · Deputy Chief Executive Officer · decision-maker
No careers mailbox known
Verified live · 2 hours ago 1 month ago Long shot · 23%

Stage Innovation : Ingénieur Génie Logiciel / Intelligence Artificielle

up to $17k per year · Intern · In office (Meudon, France) · Internship · Intern · French: C1
Python
Julia
Julia
SciML
AI/ML
JAX
TensorFlow
PyTorch
Machine Learning
DevOps
Azure DevOps
Git
Docker
Linux
CI/CD
Azure
Robotics
Digital Twin
+6
Apply
Hiring 7
Hiring 7
No named decision-maker yet.
No careers mailbox known
1 month ago Long shot · 21%

AI Researcher – Multilingual Data

Remote (location not specified) · Full-Time
Python
AI/ML
JAX
NLP
Transfer Learning
Tokenization
PyTorch
Apply
1 month ago Long shot · 21%

Machine Learning Engineer — Distillation

Remote (location not specified) · Full-Time
JAX
Knowledge Distillation
Model Distillation
PyTorch
Machine Learning
Quantization
Apply

Accenture

accenture.com
accenture.com · HQ: Dublin, Ireland · Business Process Outsourcing (BPO) · Systems Integrators · Management Consulting · IT Consulting & Digital Transformation · 5000+ employees · Est. 1989 · IT services · Trust A · Hiring 5423, 335 new
HQ: Dublin, Ireland · Business Process Outsourcing (BPO) · Systems Integrators · Management Consulting · IT Consulting & Digital Transformation · 5000+ employees · Est. 1989 · IT services · Trust A · Hiring 5423, 335 new
Accenture is a professional services company that began as the consulting arm of the accounting firm Arthur Andersen, separated as Andersen Consulting in 1989 and took its present name in 2001. It is one of the largest technology services organisations in the world, employing well over seven hundred thousand people and running strategy, consulting, technology, operations and industry work for most of the Fortune Global 500. Incorporated in Dublin and built on a delivery network concentrated in India and the Philippines, it has reoriented around cloud migration, cybersecurity and generative AI, which it now books as a distinct multi-billion dollar revenue line.
Angie Park · Chief Financial Officer · decision-maker
Julie Sweet · Chair & Chief Executive Officer · decision-maker
George Shaheen · Founding Chief Executive of Andersen Consulting · founder
Careers mailbox is known
Verified live · 24 min ago 1 month ago Long shot · 6%

AI Engineer / Agentic AI Engineer

≈ $57k – $131k per year (Estimated) · Middle · 4+ years exp · In office (Paris, France) · Bachelor's Degree · Full-Time · Middle · French: C1
SQL
Python
Databases
Neo4j
Amazon Neptune
Snowflake
Databricks
AI/ML
LangGraph
AutoGen
DSPy
LlamaIndex
Prefect
Embeddings
JAX
Haystack
ONNX
Pydantic AI
Semantic Kernel
AgentOps
Mistral
CrewAI
Gemini
Anthropic
LLMOps
GraphRAG
Human-in-the-Loop
TensorFlow
LangChain
Prompt Engineering
OpenAI
PyTorch
RAG
LLM
Machine Learning
AI Agents
DevOps
GCP
Azure
CI/CD
AWS
+30
Apply

Mach9

mach9.ai
mach9.ai · HQ: San Francisco, United States · Artificial Intelligence · Space & Aerospace · Geospatial Analytics · Computer Vision · 51-200 employees · Est. 2021 · Startup · Hiring 15
HQ: San Francisco, United States · Artificial Intelligence · Space & Aerospace · Geospatial Analytics · Computer Vision · 51-200 employees · Est. 2021 · Startup · Hiring 15
Mach9 is a geospatial AI software company specializing in automated 3D feature extraction and CAD vectorization for civil engineering and infrastructure surveying. Headquartered in San Francisco, California, United States, the platform processes raw LiDAR point clouds, 3D reality-capture scans, and imagery into precise engineering drawings. By automating spatial data analysis, the enterprise enables civil engineering firms, utility providers, and infrastructure operators to reduce manual CAD drafting, accelerate mapping timelines, and scale project capacity.
Alexander Baikovitz · Executive Officer, Director · decision-maker
Jaime Suarez Marquez · Executive Officer, Director · decision-maker
Lee Woodruff · Director
Deepak Mishra · Director
No careers mailbox known
Top 25% payVerified live · 1 hour ago Visa sponsor 1 month ago Fading · 31%

ML Engineer, Research

$180k – $300k per year · In office (San Francisco, United States) · Master's Degree · Full-Time
Python
AI/ML
Weights & Biases
Quantization
JAX
Knowledge Distillation
TensorRT
ONNX Runtime
Model Distillation
TensorFlow
Computer Vision
PyTorch
Machine Learning
+4
Apply

GyanSys

gyansys.com
gyansys.com · HQ: Indianapolis, Carmel, United States · Systems Integrators · Enterprise Apps Implementation Partners · 51-200 employees · Est. 2005 · Hiring 99, 21 new
HQ: Indianapolis, Carmel, United States · Systems Integrators · Enterprise Apps Implementation Partners · 51-200 employees · Est. 2005 · Hiring 99, 21 new
GyanSys is a global systems integrator headquartered in Indianapolis, Indiana, that implements SAP, Salesforce, Microsoft and ServiceNow platforms along with data, analytics and AI solutions for sales, finance, supply chain, manufacturing and HR processes. Founded in 2005 with a focus on SAP services, it has more than 3,000 employees in 12 countries serving about 160 active customers in manufacturing, automotive, high-tech, consumer goods and life sciences, with large delivery teams in Bengaluru. It hires SAP functional and technical consultants and architects, AI and ML engineers, data scientists and architects, .NET developers and project managers.
No named decision-maker yet.
Careers mailbox is known
≈ $33k – $70k per year (Estimated) · Lead · 7+ years exp · In office (Bengaluru, India) · Staff
Python
AI/ML
JAX
TensorFlow
PyTorch
RAG
LLM
Machine Learning
AI Agents
Reinforcement Learning
NLP
Multi-Agent Systems
+3
Apply
1 month ago Long shot · 23%

Lead/Senior AI/ML Engineer

≈ $27k – $70k per year (Estimated) · Lead · 7+ years exp · In office (Bengaluru, India) · Staff
SQL
Python
AI/ML
Reinforcement Learning
JAX
ONNX
TensorRT
Transformers
Triton
PPO
TensorFlow
PyTorch
Machine Learning
DevOps
Azure AKS
Docker
Kubernetes
Azure
CI/CD
+9
Apply
1 month ago Long shot · 23%

Lead/Senior Data Scientist

≈ $33k – $70k per year (Estimated) · Lead · 7+ years exp · In office (Bengaluru, India) · Staff
Python
AI/ML
JAX
TensorFlow
PyTorch
RAG
LLM
Machine Learning
AI Agents
Reinforcement Learning
Multi-Agent Systems
+2
Apply
3ds.com · HQ: Vélizy-Villacoublay, France · Clinical Trial Software · Manufacturing & Industrial Software · Simulation & Digital Twin Software · CAD, PLM & BIM Software · 5000+ employees · Est. 1981 · Trust B · Hiring 556, 37 new
HQ: Vélizy-Villacoublay, France · Clinical Trial Software · Manufacturing & Industrial Software · Simulation & Digital Twin Software · CAD, PLM & BIM Software · 5000+ employees · Est. 1981 · Trust B · Hiring 556, 37 new
Dassault Systemes is a French software company founded in 1981 as a spin-off from the aircraft maker Dassault Aviation, created to commercialise the three-dimensional design tools its engineers had built in house. It sells the CATIA and SOLIDWORKS design suites, SIMULIA simulation, ENOVIA lifecycle management and DELMIA manufacturing planning, all now delivered through the 3DEXPERIENCE platform that treats a product as one connected model from concept to service. The group has extended the same virtual twin idea into life sciences through the Medidata clinical trial platform and into urban and healthcare simulation, and it is headquartered in Velizy-Villacoublay near Paris.
Robert Walmsley · Executive Officer, Director · decision-maker
Roger Blanchette · Executive Officer · decision-maker
Barry Morris · Executive Officer, Director · decision-maker
Rouven Bergmann · Chief Financial Officer · decision-maker
No careers mailbox known
Verified live · 10 hours ago 1 month ago Fading · 36%

Ingénieur développement AI Physique (H/F)

≈ $53k – $115k per year (Estimated) · In office (Vélizy-Villacoublay, France) · PhD · Full-Time · French: C1
C++
Python
C++
TensorFlow C++
AI/ML
JAX
CUDA
ROCm
TensorFlow
PyTorch
Machine Learning
Management
Agile
+2
Apply
ekfrazo.in · HQ: Bengaluru, India · Custom Software Development · IT Consulting & Digital Transformation · 1001-5000 employees · Est. 2015 · Hiring 81, 4 new
HQ: Bengaluru, India · Custom Software Development · IT Consulting & Digital Transformation · 1001-5000 employees · Est. 2015 · Hiring 81, 4 new
Ekfrazo Technologies is an IT services company based in Bengaluru that delivers enterprise AI, data engineering, ServiceNow, Salesforce and cloud automation work for large corporate clients. Operating since 2015, it runs delivery teams in Bengaluru, Hyderabad, Noida and Pune and staffs engineers into client development programs, including data platform and generative AI projects for European and North American customers. Its openings cover Azure and senior data engineers, Java, .NET and Python full stack developers, agentic AI developers, solutions architects, database administrators, QA leads and technical project and program managers.
No named decision-maker yet.
No careers mailbox known
1 month ago Fading · 25%

Applied Researcher

≈ $24k – $51k per year (Estimated) · Senior · 5+ years exp · In office (Bengaluru, India) · Senior
Python
AI/ML
LoRA
Fine-tuning
JAX
Interpretability
PyTorch
RAG
LLM
Machine Learning
PEFT
+2
Apply
1 month ago Long shot · 12%

Research Lead - AI Security

≈ $28k – $72k per year (Estimated) · Lead · 8+ years exp · In office (Bengaluru, India) · Staff
LoRA
Fine-tuning
JAX
Anomaly Detection
Interpretability
PyTorch
RAG
Machine Learning
PEFT
+1
Apply
tii.ae · HQ: Abu Dhabi, United Arab Emirates · Research Institutes · Foundation Models · AI Research Labs · Quantum Science & Computing · 201-500 employees · Est. 2020 · Trust B · Hiring 15
HQ: Abu Dhabi, United Arab Emirates · Research Institutes · Foundation Models · AI Research Labs · Quantum Science & Computing · 201-500 employees · Est. 2020 · Trust B · Hiring 15
The Technology Innovation Institute is an Abu Dhabi research organisation founded in 2020 under the Advanced Technology Research Council, set up as the applied research pillar of the United Arab Emirates' science strategy. It runs a dozen dedicated research centres covering artificial intelligence, quantum computing, autonomous robotics, cryptography, advanced materials, propulsion and secure systems, staffed by researchers recruited internationally. Its highest-profile output is the Falcon family of large language models, released with open weights and for a period the strongest open model available anywhere, which established the Gulf as a serious participant in frontier AI research.
Faisal Al Bannai · Secretary General, Advanced Technology Research Council · decision-maker
Hakim Hacid · Chief Researcher, AI and Digital Science · decision-maker
Najwa Aaraj · Chief Executive Officer · decision-maker
Careers mailbox is known
Verified live · 1 hour ago 1 month ago Fading · 41%

Senior Software Engineer

Senior · In office (United Arab Emirates) · Master's Degree · Full-Time · Senior
Python
AI/ML
LoRA
Fine-tuning
RLHF
JAX
TensorFlow
PyTorch
PEFT
DevOps
Docker
Kubernetes
+2
Apply
Open 191 daysVerified live · 1 hour ago 6 months ago Long shot · 4%

Rl Robotics Engineer

In office (United Arab Emirates) · Master's Degree · Full-Time
C++
Python
C++
TensorFlow C++
AI/ML
Reinforcement Learning
JAX
Transfer Learning
Stable-Baselines3
Ray
Federated Learning
RLlib
CUDA
Multi-Agent Systems
TensorFlow
PyTorch
AI Agents
DevOps
Docker
Robotics
ROS2
Gazebo
PyBullet
MAVSDK
Isaac Gym
MuJoCo
AirSim
Swarm Robotics
Sim-to-Real
Reinforcement Learning
MAVLink
+19
Apply
Open 191 daysVerified live · 1 hour ago 6 months ago Long shot · 9%

Senior Robotics RL Engineer

≈ $42k – $100k per year (Estimated) · Senior · In office (United Arab Emirates) · Master's Degree · Full-Time · Senior
C++
Python
C++
TensorFlow C++
AI/ML
Reinforcement Learning
JAX
Transfer Learning
Stable-Baselines3
Ray
Federated Learning
RLlib
CUDA
Multi-Agent Systems
TensorFlow
PyTorch
AI Agents
DevOps
Docker
Robotics
ROS2
Gazebo
PyBullet
MAVSDK
Isaac Gym
MuJoCo
AirSim
Swarm Robotics
Sim-to-Real
Reinforcement Learning
MAVLink
+19
Apply

Writer

writer.com
writer.com · HQ: San Francisco, United States · Vector Search & RAG · Foundation Models · LLM & Generative AI · AI Agents · 5000+ employees · Est. 2020 · Hiring 46
HQ: San Francisco, United States · Vector Search & RAG · Foundation Models · LLM & Generative AI · AI Agents · 5000+ employees · Est. 2020 · Hiring 46
Writer is an enterprise generative artificial intelligence company headquartered in San Francisco, California, and founded in 2020. The company provides a full-stack platform that includes proprietary large language models, an integrated graph-based retrieval-augmented generation system, and tools for building autonomous AI agents. It serves Fortune 500 companies across sectors such as financial services, healthcare, and retail, focusing on delivering secure and compliant AI applications for marketing, sales, and operations.
May Habib · Executive Officer, Director · decision-maker
Waseem Alshikh · Executive Officer, Director · decision-maker
Jennifer Fonstad · Director
Doug Pepper · Director
Careers mailbox is known
Verified live · 1 hour ago 1 month ago Likely open · 63%

AI engineer (UK)

≈ $123k – $219k per year (Estimated) · Equity · Senior · 5+ years exp · Hybrid (London, United Kingdom) · Full-Time · Senior
Python
AI/ML
JAX
TensorFlow
PyTorch
Machine Learning
AI Agents
DevOps
GCP
Azure
AWS
+1
Apply
Top 25% payVerified live · 1 hour ago Visa sponsor 1 month ago Likely open · 63%

AI research scientist

$199k – $325k per year · Equity · Senior · 7+ years exp · Hybrid (San Francisco, Seattle, New York, United States) · PhD · Full-Time · Senior
Python
AI/ML
RLHF
Reinforcement Learning
JAX
Function Calling
NLP
Synthetic Data
DPO
SFT
GRPO
Post-training
Pre-training
Agentic Workflows
RLAIF
PyTorch
LLM
Machine Learning
AI Agents
+10
Apply
Top 25% payVerified live · 1 hour ago Visa sponsor 1 month ago Likely open · 63%

AI engineer

$198k – $259k per year · Equity · Senior · 5+ years exp · Hybrid (San Francisco, New York, United States) · Full-Time · Senior
Python
AI/ML
JAX
TensorFlow
PyTorch
Machine Learning
AI Agents
DevOps
GCP
Azure
AWS
+1
Apply

Gina's Tech Jobs

ginastechjobs.com
ginastechjobs.com · HQ: Charlotte, United States · Recruiting & Staffing · Est. 1999 · Hiring 10
HQ: Charlotte, United States · Recruiting & Staffing · Est. 1999 · Hiring 10
Gina's Tech Jobs is a technology recruitment agency headquartered in Charlotte, North Carolina, and founded in 1999. The agency places engineers into permanent roles at United States technology employers, with a heavy concentration in machine learning, artificial intelligence, and software engineering positions. It works largely on remote and hybrid searches and represents both established employers and smaller AI companies hiring technical staff.
No named decision-maker yet.
No careers mailbox known
Verified live · 24 min ago 1 month ago Hiring now · 71%

Staff Machine Learning Engineer, Artificial Intelligence (AI) Required, Work From Home

$150k – $170k per year · Staff+ · San Francisco · Remote (United States) · Full-Time · Senior · English
Python
AI/ML
LoRA
JAX
Knowledge Distillation
QLoRA
DPO
SFT
Model Distillation
PyTorch
Machine Learning
PEFT
+3
Apply
Verified live · 24 min ago 1 month ago Hiring now · 76%

Machine Learning Technical Lead, Artificial Intelligence (AI) Required, Work From Home

$190k – $225k per year · Lead · San Francisco · Remote (United States) · Full-Time · Senior · English
Python
AI/ML
LoRA
JAX
Knowledge Distillation
QLoRA
DPO
SFT
Model Distillation
PyTorch
Machine Learning
PEFT
+3
Apply
Verified live · 24 min ago 1 month ago Hiring now · 76%

LLM Application Engineer, Artificial Intelligence (AI) Required, Work From Home

$130k – $175k per year · San Francisco · Remote (United States) · Full-Time · Senior · English
Python
AI/ML
JAX
Function Calling
Structured Outputs
Context Engineering
Agentic Workflows
OpenAI
PyTorch
LLM
AI Agents
+2
Apply

Rivian

rivian.com
rivian.com · HQ: Irvine, United States · Automotive Manufacturing · 1001-5000 employees · Est. 2009 · Trust A · Hiring 531, 40 new
HQ: Irvine, United States · Automotive Manufacturing · 1001-5000 employees · Est. 2009 · Trust A · Hiring 531, 40 new
Headquartered in Irvine, California, Rivian is an American electric vehicle manufacturer and automotive technology company. The enterprise designs and builds all-electric pickup trucks, sport utility vehicles, and commercial delivery vans tailored for both consumer adventure and commercial fleet operations. Through continuous innovation in battery technology and software-defined vehicle platforms, the business aims to accelerate the global transition to sustainable transportation.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago Visa sponsor 1 month ago Likely open · 46%

Staff Software Engineer, ML Infra, Autonomy

$207k – $258k per year · Equity · Staff+ · 5+ years exp · In office (Palo Alto, United States) · Staff
Rust
Go
C++
Python
AI/ML
JAX
Ray
TPU
AWS Trainium
PyTorch
MLFlow
NCCL
LLM
DevOps
SLURM
Kubernetes
AWS
+7
Apply

Gen Digital

gendigital.com
gendigital.com · HQ: Tempe, United States · Consumer & Utility Software · Identity Management · Endpoint Security · 1001-5000 employees · Est. 2022 · NASDAQ: GEN · Trust B · Hiring 99, 3 new
HQ: Tempe, United States · Consumer & Utility Software · Identity Management · Endpoint Security · 1001-5000 employees · Est. 2022 · NASDAQ: GEN · Trust B · Hiring 99, 3 new
Gen Digital is the consumer cybersecurity company formed in 2022 when NortonLifeLock merged with the Czech antivirus firm Avast, bringing together several of the best known security brands sold to individuals. Its portfolio spans Norton, Avast, AVG, Avira, LifeLock identity protection, the CCleaner utility and, since 2025, the consumer finance app MoneyLion, all sold as subscriptions rather than through enterprise contracts. Headquartered jointly in Tempe, Arizona and Prague and listed on Nasdaq, it serves several hundred million users and competes for consumer attention rather than for corporate security budgets.
Natalie Derse · Chief Financial Officer · decision-maker
Vincent Pilette · Chief Executive Officer · decision-maker
Careers mailbox is known
Verified live · 3 hours ago 1 month ago Likely open · 54%

Senior Manager, Applied AI Research

≈ $50k – $130k per year (Estimated) · Senior · 5+ years exp · Hybrid (Prague, Brno, Czech Republic) · Master's Degree · Full-Time · Senior · English: C1
Python
AI/ML
Fine-tuning
Reinforcement Learning
JAX
PPO
TensorFlow
Computer Vision
PyTorch
LLM
Machine Learning
AI Agents
Weights & Biases
DeepSpeed
MLFlow
Quantization
Accelerate
Hugging Face
FSDP
LLM Evaluation
DevOps
Kubernetes
CI/CD
+13
Apply
Machines can read this selection · checked 11:26 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.