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

Rockwell Automation

rockwellautomation.com
rockwellautomation.com · HQ: Milwaukee, United States · Artificial Intelligence · Manufacturing · Motors, Drives & Motion Control · Manufacturing & Industrial Software · Industrial Automation · 5000+ employees · Est. 1903 · NYSE: ROK · Trust A · Hiring 372, 43 new
HQ: Milwaukee, United States · Artificial Intelligence · Manufacturing · Motors, Drives & Motion Control · Manufacturing & Industrial Software · Industrial Automation · 5000+ employees · Est. 1903 · NYSE: ROK · Trust A · Hiring 372, 43 new
Rockwell Automation is an American industrial automation company whose origins go back to 1903, and which became a pure automation business after separating from Rockwell International in the 1990s. It supplies the programmable controllers, drives, motion control, safety systems and industrial software that run factory floors, sold under the Allen-Bradley hardware brand and the FactoryTalk software family, largely through a distributor and system integrator channel. Headquartered in Milwaukee and listed on the New York Stock Exchange, it is unusually concentrated on North American manufacturing compared with its European competitors.
Blake Moret · Chairman and Chief Executive Officer · decision-maker
Lynn Stellrecht · Founder · founder
No careers mailbox known
Verified live · 4 hours ago Visa sponsor 1 month ago Long shot · 24%

Robotics Researcher

$87k – $130k per year · Senior · 5+ years exp · Hybrid (Waterloo, Kitchener, Canada) · Bachelor's Degree · Full-Time · Senior
Reinforcement Learning
JAX
Multimodal AI
Synthetic Data
CUDA
TPU
World Models
Physical AI
Embodied AI
TensorFlow
PyTorch
Robotics
ROS
ROS2
Isaac Lab
Localization
Sensor Fusion
Perception
Sim-to-Real
Reinforcement Learning
Isaac Sim
+12
Apply

RTB House

rtbhouse.com
rtbhouse.com · HQ: Warsaw, Poland · Machine Learning · Marketing & Sales AI · Advertising · 1001-5000 employees · Est. 2010 · Trust B · Hiring 64, 4 new
HQ: Warsaw, Poland · Machine Learning · Marketing & Sales AI · Advertising · 1001-5000 employees · Est. 2010 · Trust B · Hiring 64, 4 new
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.
No named decision-maker yet.
Careers mailbox is known
Verified live · 27 min ago 1 month ago Fading · 31%

Staff Research Team Lead

≈ $78k – $169k per year (Estimated) · Lead · 5+ years exp · Warsaw · Remote (Poland) · Staff
Scala
Java
Python
Databases
Google BigQuery
AI/ML
JAX
Pandas
NumPy
TPU
PyTorch
Machine Learning
DevOps
GCP
Analytics
A/B Testing
+4
Apply
Verified live · 27 min ago 2 months ago Long shot · 21%

Staff Machine Learning Researcher

≈ $104k – $185k per year (Estimated) · Staff+ · 5+ years exp · Warsaw · Remote (Poland) · Staff
Scala
Java
Python
Databases
Google BigQuery
AI/ML
JAX
Pandas
NumPy
TPU
PyTorch
Machine Learning
DevOps
GCP
Analytics
A/B Testing
+4
Apply
pi.website · HQ: San Francisco, United States · Foundation Models · AI Research Labs · Robotics AI · 501-1000 employees · Est. 2024 · Startup · Trust A · Hiring 41, 1 new
HQ: San Francisco, United States · Foundation Models · AI Research Labs · Robotics AI · 501-1000 employees · Est. 2024 · Startup · Trust A · Hiring 41, 1 new
Physical Intelligence is a robotics research company building general-purpose foundation models and learning algorithms for robots. Its models are trained on data from many robot embodiments and tasks, and its job ads cover machine learning infrastructure, robot interfaces, mechatronics and operations. The company is headquartered in San Francisco, California, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 2 hours ago 1 month ago Fading · 28%

ML Infra Engineer, Modeling

≈ $162k – $361k per year (Estimated) · In office (San Francisco, United States) · Full-Time
JAX
TPU
PyTorch
Multimodal AI
DevOps
SLURM
Google GKE
GCP
Kubernetes
AWS
+1
Apply

Centific

centific.com
centific.com · HQ: Redmond, United States · AI Consulting & Integration · AI Evaluation & Observability · AI Training Data & Annotation · Translation Services · 1001-5000 employees · Est. 2020 · IT services · Trust C · Hiring 69, 4 new
HQ: Redmond, United States · AI Consulting & Integration · AI Evaluation & Observability · AI Training Data & Annotation · Translation Services · 1001-5000 employees · Est. 2020 · IT services · Trust C · Hiring 69, 4 new
Centific is an American company spun out of the localisation group Pactera EDGE that supplies the data, evaluation and engineering services artificial intelligence systems need in production. Its work covers multilingual training data collection and annotation, human evaluation of model outputs, red teaming, and the engineering to deploy models into enterprise and retail environments, drawing on a contributor network spanning many languages and markets. Headquartered in Redmond, Washington, it works with large technology companies and enterprises, and its heritage in localisation is what gave it the global human capacity this work requires.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 4 hours ago 1 month ago Long shot · 21%

AI Research Engineer: Vision AI / VLM / Physical AI-1

$180k – $240k per year · United States · Remote (United States) · Master's Degree · Full-Time
Python
Python
FastAPI
AI/ML
JAX
Multimodal AI
VLM
Transformers
Ray
LLaVA
CUDA
Edge AI
Agentic Workflows
Multi-Agent Systems
Physical AI
Embodied AI
Computer Vision
PyTorch
RAG
AI Agents
LangGraph
Weights & Biases
YOLO
ONNX
TensorRT
xFormers
CLIP
Torchvision
Triton
Hugging Face
NeRF
LangChain
LLM
DevOps
Docker
Kubernetes
GitHub
Robotics
Isaac Sim
MuJoCo
ROS2
Open3D
SLAM
COLMAP
+32
Apply
Top 25% payVerified live · 4 hours ago 1 month ago Long shot · 24%

Senior AI Engineer

$160k – $220k per year · Senior · 5+ years exp · In office (Redmond, United States) · Master's Degree · Full-Time · Senior
Python
Python
FastAPI
AI/ML
Vertex AI
JAX
Multimodal AI
Function Calling
AWS Bedrock
RAPIDS
Triton
NVIDIA NeMo
Agentic Workflows
World Models
Physical AI
Embodied AI
TensorFlow
PyTorch
RAG
AI Agents
Weights & Biases
MLFlow
TensorRT
Ray
DevOps
GCP
AWS
Docker
Kubernetes
Robotics
Digital Twin
Isaac Sim
Sensor Fusion
+21
Apply
Top 25% payVerified live · 4 hours ago 1 month ago Fading · 25%

Principal Research Scientist – Vision AI, Simulation & Physical AI

$200k – $300k per year · Lead · 5+ years exp · United States · Remote (United States) · PhD · Full-Time · Principal
Python
Python
FastAPI
AI/ML
Fine-tuning
JAX
Multimodal AI
Knowledge Distillation
Vision-Language-Action
World Models
Physical AI
Embodied AI
Model Distillation
Computer Vision
PyTorch
RAG
Machine Learning
AI Agents
Weights & Biases
Reinforcement Learning
TensorRT
xFormers
Ray
CLIP
Triton
Hugging Face
NeRF
Knowledge Graph
DevOps
GitHub
Docker
Kubernetes
Robotics
Isaac Sim
MuJoCo
Isaac Lab
Digital Twin
Open3D
Sensor Fusion
Teleoperation
Imitation Learning
Reinforcement Learning
+30
Apply
ju.st · HQ: Emeryville, United States · Food Technology · Vegan Food · Likely startup · Hiring 4
HQ: Emeryville, United States · Food Technology · Vegan Food · Likely startup · Hiring 4
Just Food Company develops plant protein products with clean ingredients, positioned as better for people and the planet. Its job ads cover a marketing director, a machine learning research engineer, a protein chemist, a food safety director and a head of sales for Germany. The company is based in Emeryville, California, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 18 min ago 2 months ago Long shot · 16%

Machine Learning Research Engineer, Scientific AI

$160k – $200k per year · In office (Emeryville, United States) · Bachelor's Degree · Full-Time
Python
AI/ML
JAX
Multimodal AI
Diffusion Models
Accelerate
TensorFlow
PyTorch
Machine Learning
Apply

RCT Systems

rct-systems.com
rct-systems.com · HQ: Reston, United States · IT Consulting & Digital Transformation · Custom Software Development · Hiring 1
HQ: Reston, United States · IT Consulting & Digital Transformation · Custom Software Development · Hiring 1
RCT Systems is a technology contractor in Reston, Virginia that supports a sponsor's web-based applications through a technical development program of more than 200 full-time staff. Its teams include developers, DevOps, security, systems and data engineers, and it is seeking a senior software developer for backend work such as web services. The company is based in Reston, Virginia, United States.
No named decision-maker yet.
No careers mailbox known
Top 25% pay 2 months ago Long shot · 8%

Software Developer (Senior) - $164K – $182K

$164k – $182k per year · Senior · 6+ years exp · In office (Reston, United States, 9am - 3pm) · Full-Time · TS/SCI · Senior
Bash
JavaScript
SQL
Python
Databases
Oracle
ElasticSearch
AI/ML
JAX
Frontend
Vue.js
Bootstrap
Vuetify
Mobile
Material Design
DevOps
Puppet
Kibana
Logstash
Nagios
Jenkins
GitHub
Git
Linux
AWS
Management
Agile
+13
Apply
Report

Talentoj

talentoj.com
talentoj.com · HQ: Gurgaon, India · IT Staffing & Staff Augmentation · Recruiting & Staffing · Hiring 10
HQ: Gurgaon, India · IT Staffing & Staff Augmentation · Recruiting & Staffing · Hiring 10
Talentoj is an Indian talent and recruitment company that posts technology roles in artificial intelligence, machine learning, backend and full-stack engineering and quality assurance. Its vacancies include AI technical leads and biometrics machine learning engineers in Gurgaon, Delhi, Bengaluru and Noida. The company is based in India.
No named decision-maker yet.
No careers mailbox known
2 months ago Long shot · 11%

Staff/Senior Machine Learning Engineer - Python

≈ $34k – $77k per year (Estimated) · Staff+ · 5+ years exp · In office (India) · Staff
Python
Databases
Milvus
FAISS
AI/ML
OpenCV
Quantization
JAX
ONNX
TensorRT
Pillow
Synthetic Data
Amazon SageMaker
Model Distillation
TensorFlow
Computer Vision
PyTorch
Machine Learning
Diffusion Models
TFLite
LiteRT
Mobile
Core ML
DevOps
Amazon EKS
Amazon EC2
AWS
Kubernetes
+16
Apply
≈ $33k – $74k per year (Estimated) · Staff+ · 5+ years exp · India · Remote (India) · Staff
Python
Databases
Milvus
FAISS
AI/ML
OpenCV
Quantization
JAX
ONNX
TensorRT
Pillow
Synthetic Data
Amazon SageMaker
Model Distillation
TensorFlow
Computer Vision
PyTorch
Machine Learning
Diffusion Models
TFLite
LiteRT
Mobile
Core ML
DevOps
Amazon EKS
Amazon EC2
AWS
Kubernetes
+16
Apply

Silurian AI

silurian.ai
silurian.ai · HQ: Seattle, United States · Energy AI · AI for Science · Foundation Models · 51-200 employees · Est. 2024 · Startup · Hiring 1
HQ: Seattle, United States · Energy AI · AI for Science · Foundation Models · 51-200 employees · Est. 2024 · Startup · Hiring 1
Silurian AI is an artificial intelligence company that builds foundation models simulating the physical systems of the Earth, starting with weather forecasting. Its Generative Forecasting Model produces high-resolution forecasts for customers exposed to extreme weather risk, among them energy and insurance companies. It was founded in 2024 by researchers who had worked on large-scale weather models at Microsoft. The company is headquartered in Seattle, Washington, United States.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 1 hour ago 2 months ago Long shot · 15%

Machine Learning Research Engineer / Scientist

$120k – $250k per year · Equity 0.2–1% · In office (Seattle, United States) · Master's Degree · Full-Time
Python
AI/ML
JAX
PyTorch
Machine Learning
Fine-tuning
Diffusion Models
Self-Supervised Learning
Apply
Report

Build AI

build.ai
build.ai · HQ: San Francisco, United States · AI Training Data & Annotation · 201-500 employees · Likely startup · Hiring 26
HQ: San Francisco, United States · AI Training Data & Annotation · 201-500 employees · Likely startup · Hiring 26
About Build AI. Build AI is the data hyperscaler for Physical AI. We're vertically integrated across hardware, manufacturing, logistics, collection, and model training to scale the physical labor dataset orders of magnitude more than anyone in the world.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 13 min ago 2 months ago Long shot · 12%

ML Engineer, Inference Optimization

$200k – $320k per year · In office (San Francisco, United States) · Relocation · Full-Time
Rust
C++
Python
AI/ML
Quantization
JAX
CUDA
PyTorch
TensorRT
MLIR
Apache TVM
Claude
+3
Apply

SF Tensor

sf-tensor.com
sf-tensor.com · HQ: San Francisco, United States · Cloud Platforms (IaaS & PaaS) · AI Compute & Inference · Startup · Hiring 7
HQ: San Francisco, United States · Cloud Platforms (IaaS & PaaS) · AI Compute & Inference · Startup · Hiring 7
Train the models only you can build. SF Tensor provides one optimized, cross-vendor training stack for enterprise post-training and frontier pre-training.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 1 hour ago 2 months ago Long shot · 22%

Member of Technical Staff, Post-Training & Applied Research

$275k – $315k per year · Staff+ · In office (San Francisco, United States) · Relocation · Full-Time · Staff
JAX
Knowledge Distillation
Synthetic Data
DPO
SFT
PPO
GRPO
Post-training
TPU
AWS Trainium
LLM Evaluation
Model Distillation
PyTorch
Quantization
Function Calling
Speculative Decoding
+8
Apply
Top 25% payVerified live · 1 hour ago 2 months ago Long shot · 22%

Member of Technical Staff, AI-Driven Compilation

$275k – $315k per year · Staff+ · In office (San Francisco, United States) · Relocation · Full-Time · Staff
Embeddings
Reinforcement Learning
JAX
Function Calling
Post-training
Pre-training
TPU
AWS Trainium
PyTorch
LLM
AI Agents
RLHF
Triton
MLIR
Apache TVM
XLA
Reward Modeling
Robotics
Reinforcement Learning
+10
Apply

Brave

brave.com
brave.com · HQ: San Francisco, United States · AI Search · Cryptocurrencies · Advertising · Search Engines · 501-1000 employees · Est. 2015 · Trust A · Hiring 10
HQ: San Francisco, United States · AI Search · Cryptocurrencies · Advertising · Search Engines · 501-1000 employees · Est. 2015 · Trust A · Hiring 10
Brave Software is an internet browser that is focused in on being privacy-oriented and is combined with its blockchain-based digital advertising platform. The company is based out of San Francisco, California.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago Visa sponsor 2 months ago Likely open · 55%

AI Security Researcher/Research Engineer

≈ $72k – $163k per year (Estimated) · In office (United Kingdom) · PhD
Python
AI/ML
Fine-tuning
JAX
LLM Guardrails
Browser Agents
PyTorch
LLM
Cybersecurity
Least Privilege
Apply

QuantCo

quantco.com
quantco.com · HQ: Cambridge, United States · Machine Learning · Analytics & BI Consulting · 51-200 employees · Est. 2016 · Trust D · Hiring 12
HQ: Cambridge, United States · Machine Learning · Analytics & BI Consulting · 51-200 employees · Est. 2016 · Trust D · Hiring 12
QuantCo leverages expertise in data science, engineering, and economics to help organizations turn data into decisions.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago Visa sponsor 2 months ago Long shot · 8%

Applied Scientist, AI/ML

≈ $133k – $297k per year (Estimated) · Hybrid (United States) · Bachelor's Degree · Full-Time
Python
AI/ML
Polars
XGBoost
Scikit-learn
JAX
Multimodal AI
LightGBM
PyTorch
Machine Learning
+1
Apply
Likely inactiveVerified live · 2 hours ago Visa sponsor 2 years ago Long shot · 4%

Quantitative Researcher

≈ $107k – $234k per year (Estimated) · Hybrid (United States) · Bachelor's Degree · Full-Time
Python
AI/ML
Polars
XGBoost
Scikit-learn
JAX
LightGBM
PyTorch
Machine Learning
Apply

Mindfire Solutions

mindfiresolutions.com
mindfiresolutions.com · HQ: Bhubaneswar, India · IT Consulting & Digital Transformation · Custom Software Development · 1-10 employees · Est. 1999 · Hiring 2
HQ: Bhubaneswar, India · IT Consulting & Digital Transformation · Custom Software Development · 1-10 employees · Est. 1999 · Hiring 2
Mindfire Solutions provides software development and quality engineering services. Its teams build products for healthcare, finance and independent software vendors. The company works largely through dedicated offshore engineering teams.
No named decision-maker yet.
Careers mailbox is known
2 months ago Long shot · 6%

Senior AI Research Scientist

≈ $23k – $48k per year (Estimated) · Senior · 5+ years exp · In office (India) · Senior
Python
JavaScript
AI/ML
DeepSpeed
vLLM
Fine-tuning
Reinforcement Learning
Quantization
JAX
Multimodal AI
Knowledge Distillation
Function Calling
Diffusion Models
SGLang
Mamba
TensorRT-LLM
MLX ML
Transformers
Ray
Mixture of Experts
Synthetic Data
CUDA
Triton
SFT
Post-training
ROCm
Red Teaming
Edge AI
ONNX Runtime
Speculative Decoding
KV Cache
Model Distillation
PyTorch
LLM
Machine Learning
AI Agents
TensorRT
Frontend
WebGPU
+29
Apply

Hudson River Trading

hudsonrivertrading.com
hudsonrivertrading.com · HQ: New York, United States · Financial Services · Hedge Funds & Alternative Investments · 1001-5000 employees · Est. 2002 · Trust C · Hiring 78, 2 new
HQ: New York, United States · Financial Services · Hedge Funds & Alternative Investments · 1001-5000 employees · Est. 2002 · Trust C · Hiring 78, 2 new
Hudson River Trading is a New York quantitative trading firm that uses automated, research-driven strategies to trade equities, futures, options, currencies and other instruments on markets around the world. Founded in 2002, it builds its own low-latency trading systems, FPGA and ASIC hardware and machine learning models through its HRT AI Labs team, and has offices in cities including London, Chicago, Singapore, Austin, Boulder and Seattle. It hires software, hardware and design verification engineers, AI researchers, data scientists, network architects, electronic trading support engineers, tax, treasury and accounting specialists and interns.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 4 hours ago Visa sponsor 2 months ago Long shot · 8%

AI Researcher

$200k – $300k per year · Junior · 2+ years exp · In office (London, United Kingdom, New York, United States) · Junior
C++
AI/ML
JAX
PyTorch
LLM
DevOps
HPC
Apply
Likely inactiveTop 25% pay Visa sponsor 4 months ago Long shot · 4%

AI Research Engineer, Pre-Training

$250k – $300k per year · Junior · 2+ years exp · In office (London, United Kingdom, New York, United States) · Junior
JAX
CUDA
Triton
Pre-training
XLA
PyTorch
LLM
Apply
Likely inactiveTop 25% pay Visa sponsor 4 months ago Long shot · 4%

AI Research Engineer, Inference

$250k – $300k per year · Junior · 2+ years exp · In office (London, United Kingdom, New York, United States) · Junior
JAX
CUDA
Triton
XLA
PyTorch
LLM
Apply

Hestus

hestus.co
hestus.co · HQ: San Francisco, United States · Machine Learning · CAD, PLM & BIM Software · 51-200 employees · Startup · Hiring 2
HQ: San Francisco, United States · Machine Learning · CAD, PLM & BIM Software · 51-200 employees · Startup · Hiring 2
Hestus is a software company building AI-powered CAD to speed up hardware development. Its small team designs, trains and evaluates machine learning models and embeddings behind the product. The company is headquartered in San Francisco, United States.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 4 hours ago 2 months ago Long shot · 19%

Machine Learning Engineer

$180k – $240k per year · Equity · Senior · 6+ years exp · In office (San Mateo, United States) · Master's Degree · Full-Time · Senior
Python
SQL
Python
Flask
Django
AI/ML
Embeddings
JAX
TensorFlow
PyTorch
LLM
Machine Learning
Amazon SageMaker
DevOps
GCP
Azure
AWS
+6
Apply

Goaly

goaly.ai
goaly.ai · HQ: Palo Alto, United States · AI Compute & Inference · Reinforcement Learning · AI Development Tools · Likely startup · Hiring 7
HQ: Palo Alto, United States · AI Compute & Inference · Reinforcement Learning · AI Development Tools · Likely startup · Hiring 7
Goaly is an AI startup building tools that make it easier and cheaper for companies to train and continuously improve their own proprietary AI models. It was founded by former Meta Superintelligence Labs engineers and researchers and hires AI researchers in reinforcement learning, applied AI engineers and distributed infrastructure engineers. The company is headquartered in Palo Alto, California, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 2 hours ago 2 months ago Long shot · 14%

Founding Senior AI Infrastructure Engineer

≈ $145k – $277k per year (Estimated) · Senior · In office (Palo Alto, United States) · Full-Time · Senior
Rust
C++
Python
AI/ML
DeepSpeed
vLLM
Reinforcement Learning
JAX
SGLang
TensorRT-LLM
Ray
Mixture of Experts
CUDA
Triton
Post-training
Megatron-LM
FSDP
NCCL
InfiniBand
NVLink
PyTorch
Machine Learning
TensorRT
DevOps
HPC
Linux
Kubernetes
+17
Apply
Verified live · 2 hours ago Visa sponsorship 2 months ago Long shot · 14%

Founding Senior AI Researcher, RL

≈ $171k – $330k per year (Estimated) · Senior · Hybrid (Palo Alto, United States) · Full-Time · Senior
Python
AI/ML
DeepSpeed
Fine-tuning
RLHF
Reinforcement Learning
JAX
Function Calling
Ray
Post-training
Megatron-LM
FSDP
PyTorch
AI Agents
+5
Apply
Verified live · 2 hours ago Visa sponsorship 2 months ago Long shot · 10%

AI Research Scientist - Early Career (PHD/MS)

≈ $128k – $286k per year (Estimated) · Junior · Hybrid (Palo Alto, United States) · PhD · Full-Time · Junior
Python
AI/ML
Reinforcement Learning
JAX
Function Calling
Post-training
PyTorch
Machine Learning
AI Agents
Apply

Mercari

mercari.com
mercari.com · HQ: Tokyo, Japan · Secondhand · Marketplaces · 1001-5000 employees · Est. 2013 · Trust A · Hiring 123, 8 new
HQ: Tokyo, Japan · Secondhand · Marketplaces · 1001-5000 employees · Est. 2013 · Trust A · Hiring 123, 8 new
Mercari is a Japanese company that operates the Mercari marketplace app, where individuals buy and sell new and pre-owned items such as fashion, electronics, collectibles and home goods, with built-in shipping and payment services. Founded in 2013 and listed on the Tokyo Stock Exchange, it also runs the Merpay mobile payment service and a marketplace in the United States. The company is headquartered in Tokyo, Japan.
Fumiaki Koizumi · Executive Officer, Director · decision-maker
Shintaro Yamada · Executive Officer, Director · decision-maker
Masashi Namatame · Director
Makiko Shinoda · Director
No careers mailbox known
Verified live · 4 hours ago 2 months ago Likely open · 54%

ML Operations Engineer (AI/LLM) - Mercari

In office (Minato, Japan) · Full-Time · Japanese (Optional) · English: C1 (Optional)
Python
Databases
Google BigQuery
Google Bigtable
AI/ML
Triton Inference Server
Quantization
JAX
TensorRT-LLM
TPU
LLM
Machine Learning
AI Agents
TensorRT
LLM Guardrails
RAG
DevOps
SLI/SLO/SLA
Terraform
Docker
Kubernetes
CI/CD
GitHub
+12
Apply

AfterQuery

afterquery.com
afterquery.com · HQ: San Francisco, United States · LLM & Generative AI · Foundation Models · Artificial Intelligence · 201-500 employees · Startup · Trust A · Hiring 29
HQ: San Francisco, United States · LLM & Generative AI · Foundation Models · Artificial Intelligence · 201-500 employees · Startup · Trust A · Hiring 29
AfterQuery captures how experts reason and turns real-world professional work into high-quality training data for frontier foundation models.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 4 hours ago 2 months ago Long shot · 9%

Research Engineer

$210k – $450k per year · Junior · 2+ years exp · In office (San Francisco, United States) · Full-Time · Junior
Python
AI/ML
Fine-tuning
JAX
Ray
Anthropic
SFT
Post-training
OpenAI
PyTorch
LLM
Machine Learning
DevOps
SLURM
+4
Apply

Torus

usetorus.com
usetorus.com · HQ: San Francisco, United States · Construction Management Software · AI Agents · 51-200 employees · Startup · Hiring 3
HQ: San Francisco, United States · Construction Management Software · AI Agents · 51-200 employees · Startup · Hiring 3
AI agents that think in P&IDs, HMBs, and spec sheets. Torus understands every parameter across your projects and adopts how your firm works.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 21 min ago 2 months ago Long shot · 10%

AI Engineer (interested in physical infrastructure)

$150k – $250k per year · Equity 0.2–0.5% · Junior · 1+ year exp · In office (San Francisco, United States) · Full-Time · Junior
Rust
TypeScript
C++
Python
AI/ML
JAX
Function Calling
PyTorch
RAG
LLM
Machine Learning
AI Agents
+3
Apply

AMI Labs

amilabs.xyz
amilabs.xyz · HQ: Paris, France · Foundation Models · AI Research Labs · Startup · Trust B · Hiring 5
HQ: Paris, France · Foundation Models · AI Research Labs · Startup · Trust B · Hiring 5
AMI Labs, short for Advanced Machine Intelligence, builds world-model-based AI systems that understand the real world, have persistent memory, can reason and plan, and are controllable and safe. It applies them to industry, robotics and healthcare and has offices in Paris, Montreal, New York and Singapore. The company is headquartered in Paris, France.
No named decision-maker yet.
Careers mailbox is known
Verified live · 4 hours ago 2 months ago Long shot · 8%

AMI Scientist/Interns - Geometry and 3D Vision

≈ $30k – $46k per year (Estimated) · Intern · In office (Paris, France, New York, United States, Montreal, Canada) · Internship · Intern
Python
AI/ML
Self-Supervised Learning
TPU
World Models
Machine Learning
AI Agents
JAX
PyTorch
Robotics
Isaac Sim
Localization
Design
Blender
+3
Apply
Open 120 daysVerified live · 4 hours ago 4 months ago Long shot · 7%

AMI Software Engineer

≈ $46k – $128k per year (Estimated) · In office (Paris, France) · Full-Time
Python
AI/ML
Self-Supervised Learning
TPU
World Models
Machine Learning
AI Agents
JAX
Multimodal AI
PyTorch
+1
Apply
Open 223 daysVerified live · 4 hours ago 7 months ago Long shot · 4%

AMI Research Engineer

≈ $57k – $123k per year (Estimated) · In office (Paris, France, New York, United States, Montreal, Canada) · Full-Time
Python
AI/ML
Self-Supervised Learning
TPU
World Models
Machine Learning
AI Agents
JAX
PyTorch
Apply
Machines can read this selection · checked 14:10 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.