JAX Jobs - Remote & On-site

JAX jobs at vetted startups and product companies focused on high-performance ML and research, updated daily. These roles often pay competitively and support remote or hybrid work arrangements. Browse and apply today.

AI/ML
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Executive
Networking
Product
Design
Analytics
Support
Enterprise Apps
Quantum

Basis

basis.ai
Basis is a nonprofit applied research organization with two mutually reinforcing goals. Our Mission Basis is a 501(c)(3) nonprofit applied research organization with two mutually reinforcing goals.
basis.ai • HQ: New York, United States • Biotechnology • Information Technology • Artificial Intelligence
HQ: New York, United States • Biotechnology • Information Technology • Artificial Intelligence
Verified live · 2 days ago 9 months ago

ML Systems Engineer, Infrastructure & Cloud

$144k – $315k per year (Estimated)In office (New York, Cambridge, United States) • Full-Time
JAX
PyTorch
Anthropic
FSDP
NCCL
OpenAI
DevOps
AWS
Azure
GCP
Kubernetes
Terraform
Apply
Verified live · 2 days ago 9 months ago

Research Engineer, Platform

$149k – $325k per year (Estimated)In office (New York, United States) • PhD • Full-Time
Python
AI/ML
JAX
PyTorch
TensorFlow
Apply
Report

Lovelace

lovelace.ai
Lovelace is the only provider of enterprise-scale context engines that enable autonomous AI agents at the speed, scale, and accuracy required for mission-critical analysis.
lovelace.ai • Higher Education • AI Agents • Artificial Intelligence
Higher Education • AI Agents • Artificial Intelligence
Open 285 daysVerified live · 2 days ago 9 months ago

Software Engineer

$132k – $269k per year (Estimated)In office • Bachelor's Degree • Full-Time
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
JAX
PyTorch
TensorFlow
AI Agents
Knowledge Graph
DevOps
GCP
Apply
Open 326 daysVerified live · 2 days ago 10 months ago

Machine Learning Engineer

$134k – $294k per year (Estimated)In office • Bachelor's Degree • Full-Time
JAX
PyTorch
Knowledge Graph
AI Agents
DevOps
AWS
Azure
GCP
Apply
Report

Awarri

awarri.com
Awarri builds artificial intelligence infrastructure and data annotation capacity in Nigeria, including work on a national large language model. Founded in 2019 in Lagos, it also runs robotics and technology education programmes for young people. The company partnered with the Nigerian government on local language model development.
awarri.com • HQ: Lagos, Nigeria • Education • STEM Education • Artificial Intelligence • 51-200 employees • Est. 2019
HQ: Lagos, Nigeria • Education • STEM Education • Artificial Intelligence • 51-200 employees • Est. 2019
Open 303 daysVerified live · 1 day ago 10 months ago

Senior Audio AI Engineer – TTS / Speech Synthesis

$92k – $190k per year (Estimated) • Senior • 3+ years expIn office • Contractor • Senior • English (Optional)
Python
TypeScript
Python
FastAPI
Flask
Databases
MySQL
AI/ML
Fine-tuning
GAN
JAX
NLP
PyTorch
TensorFlow
Text-to-Speech
Apply
Report
AI Robot Association (AIRoA) is a Japanese non-profit organization dedicated to advancing AI-powered robotics through open data platforms and collaborative research. The association develops scalable robot data ecosystems, foundation models, and shared software tools to accelerate the deployment of humanoid and general-purpose robots worldwide.
airoa.org • Tokyo • Robotics AI • Artificial Intelligence • Robotics • 501-1000 employees • Est. 2023
Tokyo • Robotics AI • Artificial Intelligence • Robotics • 501-1000 employees • Est. 2023
Open 306 daysVerified live · 1 day ago 10 months ago

R&D-023 MLOps Engineer

$46k – $135k per year (Estimated) • Senior • 5+ years expIn office • Master's Degree • Full-Time • Senior
SQL
AI/ML
Dagster
JAX
Multimodal AI
PyTorch
Fine-tuning
Knowledge Distillation
VLM
SFT
DevOps
AWS
Azure
GCP
Robotics
Sim-to-Real
ROS
ROS2
Sensor Fusion
Analytics
ETL/ELT
Apply
Report

Substrate

substrate.com
substrate.com • San Francisco • Artificial Intelligence • Manufacturing • Hardware
San Francisco • Artificial Intelligence • Manufacturing • Hardware
Open 306 daysTop 25% pay 10 months ago

Research Scientist, AI

$150k – $275k per year • Senior • 5+ years expIn office (San Francisco, United States) • Visa sponsorship • PhD • Full-Time • Senior
Python
AI/ML
JAX
PyTorch
TensorFlow
Apply
Report

Lancedb

lancedb.com
The multimodal lakehouse for AI, accelerating large-scale data curation and feature engineering so teams can build better models faster.
lancedb.com • San Francisco • Machine Learning • Artificial Intelligence • Data & Analytics
San Francisco • Machine Learning • Artificial Intelligence • Data & Analytics
Open 309 daysTop 25% pay 10 months ago

Senior Software Engineer

$180k – $250k per year • Senior • 10+ years expIn office • Full-Time • Senior
C++
Java
Rust
Scala
C++
PyTorch C++
Databases
Apache Hudi
Apache Iceberg
ClickHouse
Delta Lake
LanceDB
Presto
Trino
HBase
AI/ML
Flink
Hadoop
JAX
Multimodal AI
PyTorch
RAG
Ray
Spark
Apply
Report

Cobot

cobot.me
Run your coworking space with one platform – mobile app, bookings, invoicing, access control, analytics, and member tools. Try Cobot free or book a quick demo.
cobot.me • Santa Clara • Seattle • Software • Real Estate • Coworking & Shared Office • Est. 2010
Santa Clara • Seattle • Software • Real Estate • Coworking & Shared Office • Est. 2010
Open 311 days 10 months ago

AI Research Engineer

$151k – $282k per year (Estimated) • Middle • 4+ years expIn office (Seattle, United States) • Relocation • Master's Degree • Full-Time • Middle
Python
AI/ML
Diffusion Models
JAX
PyTorch
Reinforcement Learning
TensorFlow
Multimodal AI
Robotics
Reinforcement Learning
Imitation Learning
Apply
Report

Simular

simular.ai
Simular is an artificial intelligence company based in Palo Alto, California, and was founded in 2023. The company develops autonomous AI agents, such as its Sai co-worker and S2 framework, which automate complex digital tasks by interacting directly with computer interfaces and web browsers. It operates as a venture-backed startup focusing on the research and deployment of computer-use models for knowledge workers and enterprise teams.
simular.ai • HQ: Palo Alto, United States • Productivity Software • Business Process Automation (BPA) • Software • AI Agents • Artificial Intelligence • Est. 2023
HQ: Palo Alto, United States • Productivity Software • Business Process Automation (BPA) • Software • AI Agents • Artificial Intelligence • Est. 2023
Open 324 daysVerified live · 2 hours ago 10 months ago

Research Scientist

$147k – $322k per year (Estimated)In office (Palo Alto, United States) • PhD • Full-Time
AI Agents
Computer Vision
Embeddings
Fine-tuning
JAX
LLM
Multimodal AI
PyTorch
Reinforcement Learning
Post-training
Pre-training
Apply
Report

Pear VC

pear.vc
Pear VC is a venture capital firm specializing in pre-seed and seed-stage investments in startups. Founded by Pejman Nozad and Mar Hershenson, the firm provides hands-on support in areas such as product development, recruiting, fundraising, and go-to-market strategy. Pear has backed companies including DoorDash, Gusto, Aurora Solar, Vanta, and Guardant Health.
pear.vc • HQ: San Francisco, United States • Financial Services • Venture Capital
HQ: San Francisco, United States • Financial Services • Venture Capital
Open 325 daysVerified live · 5 hours ago 10 months ago

AI / ML Engineer - Known

$200k – $374k per year (Estimated) • Middle • 3+ years expIn office (San Francisco, United States) • Full-Time • Middle
Python
AI/ML
Embeddings
JAX
LLM
PyTorch
TensorFlow
Hugging Face
Human-in-the-Loop
LLM Guardrails
AI Agents
Reinforcement Learning
LLM Evaluation
Analytics
A/B Testing
Apply
Report

Nagarro

nagarro.com
Nagarro is a global digital engineering and technology solutions firm that specializes in custom software development, cloud transformation, and AI-driven innovation for enterprise clients. Headquartered in Germany, the company operates across more than 30 countries with an agile, non-hierarchical organizational model tailored for rapid digital product engineering. Partnering with leading organizations across industries such as automotive, financial services, and healthcare, it helps businesses accelerate digital transformation and modernize complex IT systems.
nagarro.com • HQ: Gurgaon, India • HQ: Munich, Germany • Software • Artificial Intelligence • Information Technology • 1001-5000 employees
HQ: Gurgaon, India • HQ: Munich, Germany • Software • Artificial Intelligence • Information Technology • 1001-5000 employees
Open 361 dayVerified live · 2 hours ago 11 months ago

Staff Engineer, Java

$46k – $99k per year (Estimated) • Staff+ • 7+ years expIn office (Bengaluru, India) • Bachelor's Degree • Full-Time • Staff
Java
SQL
Java
Hibernate
Spring Framework
Databases
Apache Kafka
Redis
AI/ML
JAX
DevOps
CI/CD
GCP
Rest API
Apply
Report

Etched

etched.com
Etched. We co-design chips, racks, software, and manufacturing methods so frontier models can run with best-in-class throughput, latency, cost, and power efficiency for both prefill and decode workloads.
etched.com • San Jose • Taipei • Austin • Artificial Intelligence • AI Infrastructure • Semiconductors
San Jose • Taipei • Austin • Artificial Intelligence • AI Infrastructure • Semiconductors
Top 25% payVerified live · 5 hours ago 1 year ago

Inference Software Engineer

$175k – $275k per year • In office (San Jose, United States) • Relocation • Full-Time
C++
Rust
C++
PyTorch C++
AI/ML
JAX
PyTorch
InfiniBand
NVLink
TPU
Transformers
Mixture of Experts
Apply
Report

Netic

netic.com
Net Improvement Consulting (NETic) was founded in 1997 with a view to create specialized products to cater to the Network Management professional. Our staff have over 30 years experience in the network management industry.
netic.com • San Francisco • Managed Network Services • Information Technology • IT Management
San Francisco • Managed Network Services • Information Technology • IT Management
Open 458 daysVerified live · 7 hours ago 1 year ago

Applied AI Research Engineer

$161k – $302k per year (Estimated) • Middle • 4+ years expIn office (San Francisco, United States) • Full-Time • Middle
Databricks
AI/ML
JAX
Multimodal AI
PyTorch
Feature Store
LLM Guardrails
AI Agents
Analytics
ETL/ELT
Apply
Report

Character.AI

character.ai
Character.AI is an American company founded in 2021 by former Google researchers that operates a platform for creating and talking to personality-driven AI characters. Users build characters with a description, greeting and example dialogue, then share them publicly, and the service has grown into one of the most used consumer AI products by session time. Headquartered in Redwood City, California, the company runs its own dialogue-tuned models and has added voice calls, animated avatars and multi-character scenes, alongside age assurance and safety controls for younger users.
character.ai • HQ: Redwood City, United States • Internet Services • Media & Entertainment • Conversational AI • LLM & Generative AI • Artificial Intelligence • 51-200 employees • Est. 2021
HQ: Redwood City, United States • Internet Services • Media & Entertainment • Conversational AI • LLM & Generative AI • Artificial Intelligence • 51-200 employees • Est. 2021
Open 493 daysTop 25% pay 1 year ago

Machine Learning Infrastructure Engineer

$150k – $350k per year • Middle • 4+ years expRemote/Hybrid (Redwood City, United States) • Full-Time • Middle
JAX
PyTorch
TensorFlow
DevOps
Kubernetes
Apply
Report

Gauss Labs

gausslabs.ai
Gauss Labs is an industrial artificial intelligence company founded in 2020 by SK hynix to apply machine learning to semiconductor manufacturing. Its products improve metrology, defect detection and yield analysis inside fabrication plants. The company operates research teams in Korea and the United States.
gausslabs.ai • HQ: Seoul, South Korea • Chemical Engineering • Machine Learning • Artificial Intelligence • Est. 2020
HQ: Seoul, South Korea • Chemical Engineering • Machine Learning • Artificial Intelligence • Est. 2020
Open 495 daysVerified live · 1 day ago 1 year ago

AI Scientist - Machine Learning (KR)

Middle • 3+ years expRemote/Hybrid (Seoul, South Korea) • PhD • Full-Time • Middle
Python
AI/ML
JAX
PyTorch
TensorFlow
Transformers
Time Series Forecasting
Apply
Report

Atria Group

atriagroup.com
Atria Group (Atria Plc) is a leading Nordic food industry company that specializes in meat processing, ready meals, and convenience food production. Headquartered in Finland, the enterprise operates extensively across Northern Europe, including Finland, Sweden, Denmark, and Estonia, serving retail, food service, and industrial customers.
atriagroup.com • HQ: Plano, United States • Prepared Foods • Food & Beverages • Meat & Poultry • 1001-5000 employees
HQ: Plano, United States • Prepared Foods • Food & Beverages • Meat & Poultry • 1001-5000 employees
Open 4076 daysVerified live · 7 hours ago 11 years ago

Java Architect

$168k – $316k per year (Estimated) • Staff+ • In office (New York, United States) • Middle • English
Java
Node JS
Python
SQL
Kotlin
JavaScript
Java
Gradle
Hazelcast
Hibernate
Jackson
Maven
Spring Boot
Spring Security
Kotlin
Mockito
Databases
ElasticSearch
Memcached
Oracle
Redis
AI/ML
JAX
Apply
Report
JAX Jobs - Remote & On-site
Frequently asked questions
JAX: How many jobs are available now?
There are 491 JAX jobs listed on Alion right now.
JAX: What is the average salary for these roles?
The average salary for JAX roles on Alion is $285,074 USD.
JAX: Are remote, relocation, or visa sponsorship options available?
Many JAX listings allow remote or hybrid schedules, and some positions may offer relocation or visa sponsorship-verify on each job page.
JAX: How do I apply for JAX jobs on Alion?
Open the JAX job posting and click Apply or follow the employer’s link; sign in to save jobs and enable alerts for new listings.