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

Figure AI

figure.ai
figure.ai · HQ: San Jose, United States · Multimodal AI · Humanoid Robots · Robotics AI · 1001-5000 employees · Est. 2022 · Trust A · Hiring 93, 3 new
HQ: San Jose, United States · Multimodal AI · Humanoid Robots · Robotics AI · 1001-5000 employees · Est. 2022 · Trust A · Hiring 93, 3 new
Figure AI is an American robotics company founded in 2022 that develops general-purpose humanoid robots for commercial and household work. Its Figure 02 and Figure 03 machines combine custom actuators, battery systems and hands with the in-house Helix vision-language-action model, which lets one neural network drive both perception and motor control from natural-language instructions. Headquartered in San Jose, California, the company runs the BotQ manufacturing facility for high-volume production and has piloted its robots on logistics and automotive assembly work with commercial partners.
Brett Adcock · Founder & Chief Executive Officer · decision-maker
Jin Woo Park · Executive Officer, Director · decision-maker
Logan Berkowitz · Executive Officer · decision-maker
Dana Berlin · Executive Officer · decision-maker
Careers mailbox is known
Top 25% payVerified live · 2 hours ago Visa sponsor 2 months ago Likely open · 64%

Helix AI Engineer, Training Performance

$200k – $400k per year · Middle · 3+ years exp · In office (San Jose, United States) · Bachelor's Degree · Full-Time · Middle
C++
Python
AI/ML
DeepSpeed
vLLM
JAX
CUDA
Triton
Megatron-LM
FSDP
TPU
NCCL
InfiniBand
NVLink
AWS Trainium
PyTorch
AI Agents
+8
Apply
Verified live · 2 hours ago Visa sponsor 4 months ago Likely open · 52%

Staff AI Inference and Acceleration Engineer

$180k – $275k per year · Staff+ · 8+ years exp · In office (San Jose, United States) · Master's Degree · Full-Time · Staff
C++
Python
AI/ML
Quantization
JAX
ONNX
TensorRT
TFLite
CUDA
ROCm
LiteRT
MLIR
Apache TVM
TensorFlow
+5
Apply

AgiBot

agibot.com.cn
agibot.com.cn · HQ: Shanghai, China · Robotic Software & Control Systems · Robotics AI · Humanoid Robots · 1001-5000 employees · Est. 2023 · Trust A · Hiring 339, 1 new
HQ: Shanghai, China · Robotic Software & Control Systems · Robotics AI · Humanoid Robots · 1001-5000 employees · Est. 2023 · Trust A · Hiring 339, 1 new
Headquartered in Shanghai, China, AgiBot - also known as Zhiyuan Robotics - is an innovative artificial intelligence and robotics company specializing in general-purpose humanoid robots and embodied AI systems. The firm develops advanced hardware platforms and specialized operating software that integrate computer vision, motion control, and deep learning capabilities. By producing both bipedal and wheeled robotic systems, it provides scalable automation solutions designed for industrial manufacturing, commercial environments, and research applications.
No named decision-maker yet.
Careers mailbox is known
Verified live · 12 min ago 2 months ago Likely open · 61%

强化学习工程师(真机操作方向)

In office (Shanghai, China) · Full-Time · Chinese: C1
JAX
PPO
PyTorch
DevOps
Linux
Apply
Verified live · 12 min ago 7 months ago Fading · 36%

AI Infrastructure Lead

≈ $50k – $125k per year (Estimated) · Lead · 5+ years exp · In office (Shanghai, China) · Full-Time · Staff
DeepSpeed
RLHF
Reinforcement Learning
JAX
Multimodal AI
Ray
CUDA
Megatron-LM
FSDP
NCCL
Vision-Language-Action
World Models
Embodied AI
TensorFlow
PyTorch
LLM
DevOps
SLURM
Kubernetes
+10
Apply
Verified live · 12 min ago 7 months ago Fading · 30%

多模态运控算法研究员/专家

In office (Shenzhen, China) · Full-Time · Senior · Chinese: C1
C++
Python
AI/ML
JAX
ONNX
TensorRT
PyTorch
Robotics
Gazebo
MuJoCo
Apply

Inversion

inversionspace.com
inversionspace.com · HQ: Los Angeles, United States · Space Defense · Space Services · 51-200 employees · Est. 2021 · Trust A · Hiring 87
HQ: Los Angeles, United States · Space Defense · Space Services · 51-200 employees · Est. 2021 · Trust A · Hiring 87
Inversion is an aerospace and defense company headquartered in Los Angeles, California, that builds reentry spacecraft designed to deliver cargo from low Earth orbit to anywhere on Earth in under an hour, led by its autonomous, reusable Arc vehicle. Founded in 2021 by former SpaceX and ABL engineers Justin Fiaschetti and Austin Briggs, it flew its first in-house spacecraft Ray, runs a test site in Mojave and is backed by Y Combinator, Spark Capital and Lockheed Martin Ventures. It hires GNC, flight software, avionics, harness, RF, mechanical and aerothermal engineers, integration and test technicians, and business development staff.
No named decision-maker yet.
Careers mailbox is known
Verified live · 4 hours ago 2 months ago Likely open · 50%

Staff GNC Engineer (State Estimation)

$161k – $221k per year · Staff+ · 9+ years exp · In office (Vista, United States) · Bachelor's Degree · Top Secret · Citizenship or vetting required · Staff
MATLAB
C++
Python
AI/ML
JAX
Human-in-the-Loop
Spark
PyTorch
AI Agents
Robotics
Sensor Fusion
+1
Apply

Digital Turbine

digitalturbine.com
digitalturbine.com · HQ: Austin, United States · Advertising · 201-500 employees · Est. 1998 · NASDAQ: APPS · Hiring 23, 1 new
HQ: Austin, United States · Advertising · 201-500 employees · Est. 1998 · NASDAQ: APPS · Hiring 23, 1 new
Digital Turbine is an Austin mobile technology company listed on Nasdaq. Its platform delivers app installs, advertising and content directly through carrier and device partnerships. The company works with operators and phone makers worldwide.
No named decision-maker yet.
No careers mailbox known
Verified live · 12 hours ago Visa sponsor 2 months ago Fading · 35%

Senior Machine Learning Engineer

≈ $174k – $335k per year (Estimated) · Equity · Senior · 8+ years exp · In office (New York, United States) · Master's Degree · Full-Time · Senior
Scala
C++
Java
Python
C++
TensorFlow C++
Databases
Databricks
AI/ML
JAX
Flink
Feature Store
Recommender Systems
TensorFlow
Spark
PyTorch
Machine Learning
AI Agents
+7
Apply

Pantograph

pantograph.com
pantograph.com · HQ: San Francisco, United States · Trust B · Hiring 5
HQ: San Francisco, United States · Trust B · Hiring 5
Research and hardware for an abundant future.. Pantograph is a research lab developing generally intelligent robots with simple, scalable methods.
No named decision-maker yet.
Careers mailbox is known
Verified live · 4 hours ago 2 months ago Long shot · 10%

Research Engineer

≈ $170k – $378k per year (Estimated) · In office (San Francisco, United States) · Full-Time
Reinforcement Learning
JAX
CUDA
DevOps
Kubernetes
Linux
Robotics
Reinforcement Learning
Apply
Open 107 daysVerified live · 4 hours ago 3 months ago Long shot · 7%

Research Scientist

≈ $167k – $372k per year (Estimated) · In office (San Francisco, United States) · Full-Time
Self-Supervised Learning
Pre-training
JAX
DevOps
Kubernetes
Apply

Origamics

origamics.ai
origamics.ai · HQ: San Francisco, United States · Motors, Drives & Motion Control · Machine Learning · Hiring 2
HQ: San Francisco, United States · Motors, Drives & Motion Control · Machine Learning · Hiring 2
General-purpose AI doesn't understand how signals and fields behave on a board. We built our own models to do exactly that.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago 2 months ago Long shot · 10%

Founding Researcher

≈ $165k – $367k per year (Estimated) · In office (San Francisco, United States) · PhD · Full-Time
JAX
TensorFlow
PyTorch
AI Agents
Apply

Latitude AI

lat.ai
lat.ai · HQ: Pittsburgh, United States · Smart Mobility · Autonomous Driving · 501-1000 employees · Est. 2023 · Startup · Trust B · Hiring 29, 1 new
HQ: Pittsburgh, United States · Smart Mobility · Autonomous Driving · 501-1000 employees · Est. 2023 · Startup · Trust B · Hiring 29, 1 new
Latitude AI is a wholly owned subsidiary of Ford Motor Company formed in 2023 from former Argo AI staff to develop automated driving technology for Ford vehicles, starting with hands-free, eyes-off driving on highways. It employs engineers in software, perception, simulation and vehicle testing, along with contract test operators at sites in Michigan, California, Texas and South Carolina. The company is headquartered in Pittsburgh, Pennsylvania, United States, with engineering centers in Dearborn, Michigan and Palo Alto, California.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 1 hour ago Visa sponsorship 2 months ago Long shot · 18%

Senior Software Engineering Manager - Multi-Object Tracking & State Estimation

$253k – $380k per year · Equity · Lead · 10+ years exp · In office (Palo Alto, Pittsburgh, United States) · Bachelor's Degree · Full-Time · Senior
C++
Python
C++
TensorFlow C++
AI/ML
JAX
TensorFlow
Computer Vision
PyTorch
Machine Learning
Robotics
Motion Planning
MedTech
Verification & Validation
+2
Apply

Palona

palona.ai
palona.ai · HQ: Palo Alto, United States · Catering & Food Services · AI Agents · 51-200 employees · Est. 2024 · Startup · Hiring 6
HQ: Palo Alto, United States · Catering & Food Services · AI Agents · 51-200 employees · Est. 2024 · Startup · Hiring 6
Palona is the AI operating layer for physical businesses, proven first in restaurants. Capture demand, convert revenue, and learn where growth is next.
Maria Zhang · Executive Officer, Director · decision-maker
Flora Dong · Director
No careers mailbox known
Verified live · 1 hour ago 2 months ago Long shot · 10%

AI Modeling Engineer

≈ $143k – $317k per year (Estimated) · Equity · In office (Los Altos, New York, United States, Toronto, Canada) · Bachelor's Degree · Full-Time
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
Knowledge Distillation
Speech Recognition
Hugging Face
Post-training
Text-to-Speech
Structured Outputs
LLM Evaluation
LLM Guardrails
Model Distillation
PyTorch
Machine Learning
AI Agents
+8
Apply

Armada

armada.ai
armada.ai · HQ: San Francisco, United States · Artificial Intelligence · AI Compute & Inference · Edge AI · 501-1000 employees · Startup · Trust B · Hiring 34, 3 new
HQ: San Francisco, United States · Artificial Intelligence · AI Compute & Inference · Edge AI · 501-1000 employees · Startup · Trust B · Hiring 34, 3 new
Headquartered in San Francisco, California, Armada is an edge computing and artificial intelligence technology company that delivers advanced computing infrastructure to remote and industrial environments. The enterprise develops modular, ruggedized data centers and cloud orchestration software integrated with satellite connectivity to process high-bandwidth data directly on-site. Serving critical sectors such as energy, defense, mining, and manufacturing, the platform enables organizations to deploy real-time analytics and AI capabilities regardless of physical location.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 2 hours ago 2 months ago Fading · 30%

AI Engineer

$155k – $193k per year · Middle · 3+ years exp · In office (Bellevue, United States) · Master's Degree · Middle
C++
Java
Python
C++
TensorFlow C++
AI/ML
Fine-tuning
Reinforcement Learning
Quantization
JAX
Multimodal AI
Knowledge Distillation
Anomaly Detection
Time Series
Synthetic Data
Model Distillation
TensorFlow
Computer Vision
PyTorch
RAG
Machine Learning
LoRA
vLLM
Triton Inference Server
TensorRT
PEFT
CUDA
Human-in-the-Loop
ONNX Runtime
DevOps
Kubernetes
CI/CD
Robotics
Reinforcement Learning
+22
Apply

Plumerai

plumerai.com
plumerai.com · HQ: London, United Kingdom · Processors, MCUs & AI Chips · Edge AI · 1-10 employees · Est. 2017 · Startup · Hiring 4
HQ: London, United Kingdom · Processors, MCUs & AI Chips · Edge AI · 1-10 employees · Est. 2017 · Startup · Hiring 4
Plumerai builds highly efficient inference software for tiny edge devices. Its person and object detection runs on low cost microcontrollers and cameras. The company licenses its models to consumer and security device makers.
No named decision-maker yet.
No careers mailbox known
Verified live · 2 hours ago Visa sponsorship 2 months ago Long shot · 14%

Senior Deep Learning Research Engineer

≈ $76k – $146k per year (Estimated) · Equity · Senior · Hybrid (Amsterdam, Netherlands) · Relocation · Full-Time · Senior
Python
Databases
Snowflake
AI/ML
JAX
Multimodal AI
Keras
Streamlit
Edge AI
TensorFlow
Computer Vision
PyTorch
Reinforcement Learning
DevOps
GCP
Kubernetes
+5
Apply
Verified live · 2 hours ago Visa sponsorship 2 months ago Long shot · 7%

Deep Learning Research Engineer

≈ $56k – $124k per year (Estimated) · Equity · Hybrid (Amsterdam, Netherlands) · Relocation · Full-Time
Python
Databases
Snowflake
AI/ML
JAX
Multimodal AI
Keras
Streamlit
Edge AI
TensorFlow
Computer Vision
PyTorch
Reinforcement Learning
DevOps
GCP
Kubernetes
+5
Apply

Basis

withbasis.co
withbasis.co · HQ: San Francisco, United States · AI Training Data & Annotation · Startup · Trust A · Hiring 28
HQ: San Francisco, United States · AI Training Data & Annotation · Startup · Trust A · Hiring 28
Basis is a company that supplies human expertise and data to frontier AI developers and describes itself as the human foundation behind frontier AI. Its customers include leading research institutions, and it hires founding engineers, machine learning engineers and forward deployed engineers. The company is headquartered in San Francisco, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago 2 months ago Fading · 30%

Founding ML Engineer

≈ $192k – $427k per year (Estimated) · In office (San Francisco, United States) · Full-Time
Reinforcement Learning
JAX
Multimodal AI
ONNX
TensorRT
Anomaly Detection
CUDA
Triton
Vision-Language-Action
World Models
PyTorch
Machine Learning
Robotics
Isaac Sim
MuJoCo
Sim-to-Real
Imitation Learning
Reinforcement Learning
+9
Apply

Nuro

nuro.ai
nuro.ai · HQ: Mountain View, United States · Delivery · Smart Mobility · Autonomous Driving · 1001-5000 employees · Est. 2016 · Trust B · Hiring 94
HQ: Mountain View, United States · Delivery · Smart Mobility · Autonomous Driving · 1001-5000 employees · Est. 2016 · Trust B · Hiring 94
Nuro is an autonomous driving technology company headquartered in Mountain View, California, that develops the Nuro Driver, an AI-based self-driving system, and licenses it to automakers and mobility partners. Founded in 2016, it has partnered with Uber and Lucid on a robotaxi program planned for Houston in 2027 and announced a European base in Germany in 2026. It hires software engineers for collision avoidance, planning, cloud infrastructure and tooling, embedded engineers, technical lead managers, program managers for validation and compliance, and legal counsel.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 15 min ago Visa sponsor 2 months ago Long shot · 24%

Staff/Lead Machine Learning Engineer, Behavior & Planning

$235k – $352k per year · Lead · In office (Mountain View, United States) · Master's Degree · Staff
C++
Python
C++
TensorFlow C++
AI/ML
Diffusion Models
VLM
World Models
LLM
Machine Learning
JAX
TensorFlow
PyTorch
Robotics
Imitation Learning
+4
Apply

Quantbot Technologies

quantbot.com
quantbot.com · New York · London · Miami · Financial AI · Machine Learning · Professional Services · 1-10 employees · Hiring 7
New York · London · Miami · Financial AI · Machine Learning · Professional Services · 1-10 employees · Hiring 7
UNLOCKING OPPORTUNITIES, MAXIMIZING RETURNS. Explore the intricacies of quantitative investment strategies tailored for global markets.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 3 hours ago Visa sponsor 2 months ago Long shot · 7%

Machine Learning Research Engineer Internship - 2027 [New York]

$180k – $200k per year · Intern · In office (New York, United States) · Internship · Intern
Rust
C++
Python
C++
TensorFlow C++
AI/ML
Machine Learning
JAX
TensorFlow
PyTorch
Apply

Rosendahl Nextrom

rosendahlnextrom.com
rosendahlnextrom.com · HQ: Vienna, Austria · Electrical Equipment & Cables · Industrial Machinery · 11-50 employees · Est. 1958 · Hiring 3
HQ: Vienna, Austria · Electrical Equipment & Cables · Industrial Machinery · 11-50 employees · Est. 1958 · Hiring 3
Rosendahl Nextrom designs and supplies production equipment for the battery, cable and wire, and optical fibre industries. Its lines integrate robotic handling and automated testing, and the company hires field service technicians and physical AI research engineers. The company is based in Austria.
No named decision-maker yet.
No careers mailbox known
Verified live · 5 hours ago 2 months ago Long shot · 8%

Physical AI Research Engineer (f/m/x)

$60k per year (gross) · Hybrid (Pischelsdorf, Austria) · Relocation · Master's Degree · Full-Time
C++
Python
C++
TensorFlow C++
AI/ML
Reinforcement Learning
JAX
Multimodal AI
Edge AI
Vision-Language-Action
TensorFlow
Computer Vision
PyTorch
Machine Learning
Robotics
ROS
PyBullet
Isaac Sim
MuJoCo
Perception
Sim-to-Real
Imitation Learning
Reinforcement Learning
Digital Twin
+13
Apply

ClickPost

clickpost.ai
clickpost.ai · HQ: Gurgaon, India · Supply Chain & Logistics Software · 51-200 employees · Est. 2015 · Startup · Trust A · Hiring 13, 1 new
HQ: Gurgaon, India · Supply Chain & Logistics Software · 51-200 employees · Est. 2015 · Startup · Trust A · Hiring 13, 1 new
ClickPost provides logistics intelligence software for ecommerce shipping. Its platform allocates carriers, tracks shipments and manages returns across providers. Retailers use it to reduce failed deliveries and support enquiries.
No named decision-maker yet.
No careers mailbox known
Verified live · 4 hours ago 2 months ago Long shot · 10%

Data Scientist

≈ $13k – $30k per year (Estimated) · Junior · 2+ years exp · In office (Bengaluru, India) · Bachelor's Degree · Full-Time · Junior
Java
SQL
Python
Python
pySpark
FastAPI
Databases
Delta Lake
Google BigQuery
Databricks
Apache Kafka
AI/ML
LangGraph
AutoGen
Airflow
LlamaIndex
XGBoost
Fine-tuning
Embeddings
Scikit-learn
JAX
NLP
NER
Llama
Flink
NumPy
CrewAI
CLIP
Tokenization
Anomaly Detection
Image Segmentation
OCR
Voice Agents
Agentic Workflows
TensorFlow
Computer Vision
LangChain
Prompt Engineering
Spark
PyTorch
Claude
RAG
Machine Learning
AI Agents
Vertex AI
AWS Bedrock
Amazon SageMaker
Feature Store
DevOps
GitLab
Amazon S3
Git
GCP
Azure
AWS
Docker
Kubernetes
CI/CD
Analytics
Matplotlib
Plotly
A/B Testing
Tableau
Power BI
Dimensional Modeling
Management
Confluence
+53
Apply

Svitla Systems

svitla.com
svitla.com · HQ: Corte Madera, United States · DevOps & Platform Engineering · IT Outsourcing & Dedicated Teams · Custom Software Development · Est. 2003 · Trust B · Hiring 91, 15 new
HQ: Corte Madera, United States · DevOps & Platform Engineering · IT Outsourcing & Dedicated Teams · Custom Software Development · Est. 2003 · Trust B · Hiring 91, 15 new
Svitla Systems is a global digital solutions and custom software engineering company headquartered in California, with delivery centers across North America, Latin America, Europe, and Asia. Founded in 2003, the company offers end-to-end IT services, including cloud architecture, AI/machine learning integration, data engineering, DevOps, and mobile and web application development.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago 2 months ago Long shot · 19%

Physics-Informed Machine Learning Engineer

Argentina · Remote (Argentina)
Python
AI/ML
Scikit-learn
JAX
Pandas
NumPy
Time Series
PyTorch
Machine Learning
Apply

Boston Dynamics

bostondynamics.com
bostondynamics.com · HQ: Waltham, United States · Industrial Robotics · Logistics & Warehouse Robotics · Humanoid Robots · 1001-5000 employees · Est. 1992 · Trust B · Hiring 84, 9 new
HQ: Waltham, United States · Industrial Robotics · Logistics & Warehouse Robotics · Humanoid Robots · 1001-5000 employees · Est. 1992 · Trust B · Hiring 84, 9 new
Boston Dynamics is a robotics company founded in 1992 as a spin-off from the Massachusetts Institute of Technology and now owned by Hyundai Motor Group. It builds the Spot quadruped, the Stretch warehouse robot and the Atlas humanoid, and has shifted from research demonstrations to commercial deployment in inspection and logistics. Recent work combines its control expertise with reinforcement learning and large behaviour models developed with partners.
No named decision-maker yet.
No careers mailbox known
Verified live · 11 hours ago Visa sponsor 2 months ago Fading · 28%

Staff Reinforcement Learning Research Engineer

$155k – $200k per year · Staff+ · 3+ years exp · In office (Waltham, United States) · PhD · Full-Time · Staff
Reinforcement Learning
JAX
Multimodal AI
ONNX
TensorRT
Stable-Baselines3
RLlib
Triton
PyTorch
DevOps
Bazel
Docker
Kubernetes
CI/CD
Robotics
MuJoCo
Isaac Lab
Sim-to-Real
Reinforcement Learning
Isaac Sim
+10
Apply
Top 25% payVerified live · 11 hours ago Visa sponsor 4 months ago Long shot · 9%

Atlas Research Engineer, SLAM & Spatial AI

$177k – $225k per year · In office (Waltham, United States) · PhD · Full-Time
C++
Python
AI/ML
VLM
Vision-Language-Action
Computer Vision
Machine Learning
JAX
PyTorch
Robotics
SLAM
Sensor Fusion
+2
Apply

SignalFire

signalfire.com
signalfire.com · HQ: San Francisco, United States · Venture Capital · Est. 2013 · Hiring 18
HQ: San Francisco, United States · Venture Capital · Est. 2013 · Hiring 18
SignalFire is a venture capital firm that invests in technology startups from seed to growth stage. It runs a proprietary data platform called Beacon that tracks companies and talent to source investments, and it gives portfolio companies in-house support with recruiting, marketing and advisory networks, with a focus on AI, healthcare, enterprise and consumer software. The firm is headquartered in San Francisco, California, United States.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 3 hours ago 2 months ago Long shot · 17%

Head of AI/ML (Director/VP) - VC Backed Startups

$250k – $300k per year · Head+ · 10+ years exp · New York · Remote (United States) · Bachelor's Degree · Full-Time · Staff
Python
Databases
Snowflake
Databricks
Apache Kafka
AI/ML
Weights & Biases
MLFlow
Vertex AI
Fine-tuning
Scikit-learn
JAX
Multimodal AI
Anthropic
Hugging Face
Amazon SageMaker
Feature Store
TensorFlow
Prompt Engineering
OpenAI
Spark
PyTorch
RAG
Machine Learning
AI Agents
DevOps
Docker
GCP
Kubernetes
Azure
AWS
+20
Apply
Verified live · 3 hours ago 2 months ago Long shot · 15%

Applied AI Scientist/Researcher (Senior/Staff) - VC Backed Startups

≈ $176k – $331k per year (Estimated) · Staff+ · 5+ years exp · Remote (United States) · Bachelor's Degree · Full-Time · Staff
Python
Databases
Snowflake
Databricks
AI/ML
Fine-tuning
Reinforcement Learning
Scikit-learn
JAX
Multimodal AI
Knowledge Distillation
Function Calling
Diffusion Models
Synthetic Data
Anthropic
Hugging Face
Post-training
Red Teaming
Recommender Systems
TensorFlow
Computer Vision
OpenAI
Spark
PyTorch
RAG
Machine Learning
DevOps
Kubernetes
+17
Apply
Verified live · 3 hours ago 2 months ago Long shot · 15%

Founding AI Engineer (Senior/Staff) - VC Backed Startups

$170k – $250k per year · Staff+ · 3+ years exp · In office (New York, United States) · Full-Time · Staff
Python
SQL
Python
Flask
FastAPI
Databases
Apache Kafka
Google BigQuery
AI/ML
Hadoop
JAX
TensorFlow
Spark
PyTorch
RAG
LLM
Machine Learning
Airflow
MLFlow
Vertex AI
Scikit-learn
Kubeflow
Flink
Amazon SageMaker
DevOps
Docker
GCP
Kubernetes
Azure
AWS
AWS Lambda
Amazon S3
+20
Apply

Omakase Robotics

omakase.ai
omakase.ai · HQ: Meguro, Japan · Retail & Commercial Robotics · Robotics AI · Robotic Software & Control Systems · Service & Hospitality Robotics · 51-200 employees · Hiring 5
HQ: Meguro, Japan · Retail & Commercial Robotics · Robotics AI · Robotic Software & Control Systems · Service & Hospitality Robotics · 51-200 employees · Hiring 5
Omakase Robotics is a robotics division of the Japanese company Zeals that develops Omakase OS, a software and systems stack for running robots reliably in human environments. It combines autonomy and manipulation research, hardware and field operations, with forward deployed engineers putting robots to work in hospitals, hotels and retail stores. The company is headquartered in Meguro, Tokyo, Japan.
No named decision-maker yet.
No careers mailbox known
Verified live · 4 hours ago 2 months ago Long shot · 12%

Robotics Researcher — Manipulation (Omakase Zen)

≈ $111k – $267k per year (Estimated) · In office (Meguro, Japan) · PhD · Full-Time
Fine-tuning
JAX
Post-training
Pre-training
LLM Evaluation
Vision-Language-Action
World Models
PyTorch
Robotics
MuJoCo
Teleoperation
Sim-to-Real
Imitation Learning
OpenVLA
+5
Apply

TD SYNNEX

tdsynnex.com
tdsynnex.com · HQ: Clearwater, United States · Servers & Data Center Hardware · Wholesale · IT Resellers & VARs · 5000+ employees · Est. 2021 · NYSE: SNX · Trust B · Hiring 636, 66 new
HQ: Clearwater, United States · Servers & Data Center Hardware · Wholesale · IT Resellers & VARs · 5000+ employees · Est. 2021 · NYSE: SNX · Trust B · Hiring 636, 66 new
TD SYNNEX is an American technology distributor formed in 2021 by merging Tech Data with SYNNEX, and it is one of the two companies through which most of the world's technology hardware and software reaches resellers and end customers. Its business is enormous in revenue and thin in margin: it buys from thousands of manufacturers, warehouses and finances the inventory, and sells onward to hundreds of thousands of resellers who lack the scale to deal with vendors directly. Headquartered in Clearwater, Florida and Fremont, California and listed on the New York Stock Exchange, it also assembles data centre systems through its Hyve division.
No named decision-maker yet.
No careers mailbox known
Verified live · 9 hours ago Visa sponsor 2 months ago Long shot · 14%

Physical AI Engineer

≈ $109k – $243k per year (Estimated) · In office (Clearwater, United States) · PhD · Full-Time
Python
C++
AI/ML
Reinforcement Learning
JAX
Stable-Baselines3
Synthetic Data
Human-in-the-Loop
Physical AI
PyTorch
Machine Learning
Multimodal AI
Accelerate
Data Augmentation
Vision-Language-Action
Game Dev
NVIDIA PhysX
Robotics
NVIDIA Omniverse
Isaac Sim
Isaac Lab
Sim-to-Real
Imitation Learning
Reinforcement Learning
ROS
MuJoCo
Localization
+16
Apply
Machines can read this selection · checked 15:12 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.