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

Mozn

mozn.ai
MOZN is an enterprise AI company that has helped 100+ organizations make critical and informed decisions through specialized AI, in two key areas: Financial Crime Prevention and Enterprise Knowledge Intelligence
mozn.ai • Riyadh • Cairo • Lagos • Dubai • Tunis • Law Enforcement • Fraud Detection • Artificial Intelligence
Riyadh • Cairo • Lagos • Dubai • Tunis • Law Enforcement • Fraud Detection • Artificial Intelligence
Verified live · 3 hours ago 2 months ago

AI Infrastructure Engineer III

Middle • 4+ years expCairo • Remote (Egypt) • Full-Time • Middle
Python
Databases
OpenSearch
AI/ML
CUDA
CUDA Toolkit
KServe
Kubeflow
MLFlow
Ray
Ray Serve
Triton
Triton Inference Server
DeepSpeed
JAX
LLM
PyTorch
RAG
TensorFlow
Hugging Face
NCCL
DevOps
Ansible
AWS
Azure
CI/CD
GCP
GitOps
Grafana
Helm
Kubernetes
OpenTelemetry
Platform Engineering
Prometheus
Terraform
Vector
Apply
Report

AE Studio

ae.studio
AE Studio is a bootstrapped, ~150-person applied AI studio founded in 2016. We make high-stakes AI reliable: evals, red-teaming, guardrails, and continuous observability, with the data foundation that fixes failures at the root.
ae.studio • Los Angeles • Artificial Intelligence • Est. 2016
Los Angeles • Artificial Intelligence • Est. 2016
Open 1967 daysVerified live · 11 hours ago 5 years ago

Applied AI Data Scientist (Brazil)

Remote (Brazil)
Python
AI/ML
Computer Vision
Cursor
Fine-tuning
JAX
LLM
NLP
PyTorch
RAG
TensorFlow
Time Series Forecasting
Apply
Open 1967 daysVerified live · 11 hours ago 5 years ago

Applied AI Data Scientist

$136k – $297k per year (Estimated) • Remote (United States, Brazil)
Python
AI/ML
Computer Vision
Cursor
Fine-tuning
JAX
LLM
NLP
PyTorch
RAG
TensorFlow
Time Series Forecasting
Apply
Report

Odyssey AI

odyssey.com
Odyssey is a deep-tech artificial intelligence startup building multimodal "world models" for interactive 3D simulations, headquartered in Palo Alto, California. Founded in 2023 by Oliver Cameron (former VP of Product at Cruise and co-founder of Voyage) and Dr. Jeffrey Hawke (former VP of Technology at Wayve), the venture has raised over $337 million in funding - highlighted by a $310 million Series B at a $1.45 billion valuation led by Google Ventures, EQT Ventures, and Natural Capital.
odyssey.com • HQ: Palo Alto, United States • Blockchain & Crypto • NFT • Internet Services • Domain Names • 201-500 employees • Est. 2023
HQ: Palo Alto, United States • Blockchain & Crypto • NFT • Internet Services • Domain Names • 201-500 employees • Est. 2023
Verified live · 3 hours ago 2 months ago

Member of Technical Staff, Foundation Models

$185k – $374k per year (Estimated) • Staff+ • 4+ years expIn office (Palo Alto, United States, Zurich, Switzerland, London, United Kingdom) • Full-Time • Staff
JAX
Multimodal AI
PyTorch
OpenAI
Apply
Verified live · 3 hours ago 2 months ago

Member of Technical Staff, Diffusion World Models & Robotics

$148k – $325k per year (Estimated) • Staff+ • 4+ years expIn office (Zurich, Switzerland, London, United Kingdom, Palo Alto, United States) • PhD • Full-Time • Staff
Fine-tuning
Multimodal AI
OpenAI
AI Agents
JAX
PyTorch
Robotics
Sim-to-Real
Apply
Open 173 daysVerified live · 3 hours ago 6 months ago

Member of Technical Staff, ML Performance

$185k – $374k per year (Estimated) • Staff+ • 8+ years expIn office (Palo Alto, United States, Zurich, Switzerland, London, United Kingdom) • Full-Time • Staff
JAX
Multimodal AI
PyTorch
Triton
OpenAI
Apply
Report

Luma AI

lumalabs.ai
Luma AI is an artificial intelligence company that specializes in building advanced 3D visual technology and generative AI tools. Founded in 2021, the company is best known for its realistic text-to-3D models, NeRF (Neural Radiance Fields) capture app, and its high-quality video generation model, Dream Machine. By enabling creators and developers to easily capture, generate, and edit photorealistic 3D assets and video, it aims to make next-generation digital media creation accessible to everyone.
lumalabs.ai • HQ: Palo Alto, United States • Artificial Intelligence • Generative Image • Generative Video • Est. 2021
HQ: Palo Alto, United States • Artificial Intelligence • Generative Image • Generative Video • Est. 2021
Verified live · 12 hours ago 2 months ago

Research Engineer - Evaluations

$178k – $440k per year (Estimated)Remote/Hybrid (Redwood City, United States) • Master's Degree • Full-Time
Python
AI/ML
JAX
Multimodal AI
PyTorch
Reinforcement Learning
TensorFlow
Human-in-the-Loop
DevOps
CI/CD
Apply
Report

PhysicsX

physicsx.ai
PhysicsX is a deep technology company that develops artificial intelligence and simulation software to accelerate engineering and advanced manufacturing. Founded by former Formula One engineers and computer scientists, the organization uses physics-informed AI and deep learning to optimize complex designs in fields like aerospace, automotive, and renewable energy. By replacing traditional, time-intensive physics simulations with real-time machine learning models, it enables companies to innovate faster and dramatically improve performance.
physicsx.ai • HQ: London, United Kingdom • Industrial Engineering • Artificial Intelligence • Simulation & Digital Twin
HQ: London, United Kingdom • Industrial Engineering • Artificial Intelligence • Simulation & Digital Twin
Verified live · 1 day ago 5 months ago

Senior Machine Learning Software Engineer, Research

$92k – $190k per year (Estimated) • Equity • Senior • 4+ years expRemote/Hybrid (London, United Kingdom) • Master's Degree • Senior
C++
Python
C++
PyTorch C++
Python
Dask
AI/ML
Computer Vision
CUDA Toolkit
Federated Learning
JAX
NumPy
OpenMP
Pandas
PyTorch
SciPy
Spark
DevOps
AWS
Azure
CI/CD
Docker
GCP
Kubernetes
Octopus Deploy
SLURM
Apply
Report

Nomagic

nomagic.ai
Nomagic is a warehouse robotics company headquartered in Warsaw, Poland, and founded in 2017. The company builds AI controlled robotic arms that pick and place individual items in ecommerce fulfilment and logistics operations, handling mixed goods without per-item programming. It sells the system on a robot as a service model to European retailers and third party logistics providers, integrating with existing conveyors and warehouse management software.
nomagic.ai • HQ: Warsaw, Poland • Packaging & Inspection Robotics • Robotics AI • Artificial Intelligence • Robotics • Transportation & Logistics • Logistics & Warehouse Robotics • Warehousing • Est. 2017
HQ: Warsaw, Poland • Packaging & Inspection Robotics • Robotics AI • Artificial Intelligence • Robotics • Transportation & Logistics • Logistics & Warehouse Robotics • Warehousing • Est. 2017
Verified live · 1 day ago 2 months ago

Research Scientist

$68k – $116k per year (Estimated)Remote/Hybrid (Warsaw, Poland) • Relocation • Full-Time
JavaScript
AI/ML
Fine-tuning
JAX
Knowledge Distillation
Multimodal AI
PyTorch
Reinforcement Learning
RLHF
Transformers
Post-training
Frontend
Bootstrap
Robotics
Imitation Learning
Reinforcement Learning
Apply
Verified live · 1 day ago 2 months ago

Research Engineer

$67k – $115k per year (Estimated)Remote/Hybrid (Warsaw, Poland) • Relocation • Full-Time
Python
JavaScript
AI/ML
Fine-tuning
JAX
Multimodal AI
PyTorch
Transformers
Frontend
Bootstrap
Robotics
Imitation Learning
Apply
Verified live · 1 day ago 4 months ago

Research Scientist

$125k – $208k per year (Estimated)Remote/Hybrid (Zurich, Switzerland) • Relocation • Full-Time • English
JavaScript
AI/ML
Fine-tuning
JAX
Knowledge Distillation
Multimodal AI
PyTorch
Reinforcement Learning
RLHF
Transformers
Post-training
Frontend
Bootstrap
Robotics
Imitation Learning
Reinforcement Learning
Apply
Report

Cerebras Systems

cerebras.ai
Cerebras Systems is an American computer hardware company founded in 2016 and headquartered in Sunnyvale, California that builds accelerators for artificial intelligence at wafer scale. Instead of assembling clusters from many small chips, it manufactures a single processor the size of an entire silicon wafer, the Wafer Scale Engine, which removes most of the communication overhead in large model training and inference. The company sells CS-series systems to research laboratories and enterprises, operates its own inference cloud known for very high token throughput, and has built large supercomputers with partners including the Gulf technology group G42.
cerebras.ai • HQ: Sunnyvale, United States • Machine Learning • Artificial Intelligence • Hardware • Semiconductors • AI Infrastructure • 1001-5000 employees • Est. 2016
HQ: Sunnyvale, United States • Machine Learning • Artificial Intelligence • Hardware • Semiconductors • AI Infrastructure • 1001-5000 employees • Est. 2016
$171k – $400k per year (Estimated) • Junior • In office (Sunnyvale, United States) • Visa sponsorship • Bachelor's Degree • Internship • Junior
C++
Go
Java
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
Cerebras
Edge AI
OpenAI
AI Agents
JAX
Multimodal AI
PyTorch
TensorFlow
DevOps
Docker
Kubernetes
Apply
Verified live · 6 hours ago 2 months ago

Director/Sr. Manager, AI Inference Model Scaling

$224k – $478k per year (Estimated) • Executive • 5+ years expRemote/Hybrid (Sunnyvale, United States) • Bachelor's Degree • Full-Time • Senior
C++
Python
C++
LLVM
PyTorch C++
TensorFlow C++
AI/ML
Cerebras
LLM
AI Agents
Edge AI
OpenAI
JAX
ONNX
PyTorch
TensorFlow
Apply
Report

Xebia

xebia.com
Xebia is a global, award-winning, AI-first consulting, software engineering, and training company. We shape tomorrow with AI today helping organizations turn AI strategy into production-ready solutions that drive real transformation, faster.
xebia.com • HQ: Gurgaon, India • LLM & Generative AI • Information Technology • Artificial Intelligence • Est. 2001
HQ: Gurgaon, India • LLM & Generative AI • Information Technology • Artificial Intelligence • Est. 2001
2 months ago

Lead Data Scientist

$97k – $119k per year • Lead • Wrocław • Remote (Poland) • Full-Time • Staff
Python
JavaScript
Java
TypeScript
Python
FastAPI
Pydantic
SQLAlchemy
Celery
Hypothesis
Pandera
Java
Testcontainers
Databases
PostgreSQL
Redis
AI/ML
Claude
Claude Code
Copilot
Cursor
LLM
NumPy
Pandas
Scikit-learn
SciPy
Statsmodels
Gemini
JAX
Kedro
TensorFlow
Frontend
React.js
DevOps
Docker
GCP
Kubernetes
GitHub
AWS
Azure
OpenTelemetry
Apply
Report
Carnegie Mellon University is a private research university located in Pittsburgh, Pennsylvania, world-renowned for its top-tier programs in computer science, robotics, engineering, business, and the arts. Founded in 1900 by Andrew Carnegie, the institution is famous for its interdisciplinary approach and pioneering role in fields like artificial intelligence and software engineering. Beyond academics, it operates major technology and research facilities, driving innovation and spinoff ventures across global industries.
cmu.edu • Pittsburgh • Doha • Arlington • Cybersecurity • Education • Artificial Intelligence • Higher Education • 1001-5000 employees • Est. 1900
Pittsburgh • Doha • Arlington • Cybersecurity • Education • Artificial Intelligence • Higher Education • 1001-5000 employees • Est. 1900
Verified live · 1 day ago 2 months ago

AI Engineer - Mission Innovation Lab

$125k – $242k per year (Estimated) • Senior • 8+ years expIn office (Pittsburgh, Arlington, United States) • Bachelor's Degree • Full-Time • Staff
C++
Java
Python
C++
PyTorch C++
TensorFlow C++
Python
FastAPI
AI/ML
Computer Vision
Fine-tuning
JAX
LangChain
LLM
LoRA
Model Context Protocol
PEFT
Prefect
PyTorch
RAG
Ray
Reinforcement Learning
TensorFlow
Transformers
Edge AI
Hugging Face
AI Agents
TPU
DevOps
CI/CD
Docker
Kubernetes
Vector
GitOps
gRPC
Analytics
ETL/ELT
Apply
Verified live · 1 day ago 2 months ago

Senior Data Scientist

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

Data Scientist

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

Figma

figma.com
Figma is a leading collaborative interface design and prototyping platform that enables teams to build digital products together in real time. Operating entirely in the cloud, the platform integrates vector graphics editing, interactive prototyping, design system management, and developer handoff into a seamless, browser-based workflow.
figma.com • New York • Berlin • Sydney • London • Washington • Design & Creative • Digital Design • UI/UX Design • 1001-5000 employees
New York • Berlin • Sydney • London • Washington • Design & Creative • Digital Design • UI/UX Design • 1001-5000 employees
Verified live · 2 min ago 9 months ago

AI Applied Scientist

$139k – $289k per year (Estimated) • Senior • 6+ years expRemote/Hybrid (San Francisco, United States) • Internship • Senior
C++
Java
Python
C++
PyTorch C++
AI/ML
Computer Vision
Fine-tuning
JAX
Prompt Engineering
PyTorch
Reinforcement Learning
Synthetic Data
DPO
GRPO
Hugging Face
PPO
SFT
Design
Figma
Apply
Report

Radical Numerics

radicalnumerics.ai
Radical Numerics is an artificial intelligence research lab dedicated to developing general biological intelligence and generative genomics models. Headquartered in San Francisco, California, the company builds multimodal platforms capable of reading, writing, and engineering biological sequences across DNA, RNA, and proteins. Its technology aims to accelerate biopharmaceutical research, enhance early disease diagnostics, and establish robust biodefense capabilities.
radicalnumerics.ai • HQ: San Francisco, United States
HQ: San Francisco, United States
Verified live · 1 day ago 2 months ago

Member of Technical Staff, Protein Design

$195k – $394k per year (Estimated) • Staff+ • In office (San Francisco, United States) • Full-Time • Staff
Python
AI/ML
Diffusion Models
Fine-tuning
JAX
PyTorch
Apply
Open 167 daysVerified live · 1 day ago 6 months ago

Member of Technical Staff, AI Bio

$196k – $396k per year (Estimated) • Staff+ • In office (San Francisco, United States) • Full-Time • Staff
Python
AI/ML
Diffusion Models
Fine-tuning
JAX
Multimodal AI
PyTorch
Post-training
Apply
Report

ELCASecurity

elcasecurity.ch
Headquartered in Pully, Switzerland, ELCASecurity is a specialized cybersecurity company and part of the Swiss IT services firm ELCA Group. The company offers a broad range of offensive and defensive security services, including cybersecurity consulting, risk assessment, and identity and access management solutions like trustID. Through its specialized divisions and local Security Operations Centers (SOC), it provides continuous threat detection and data protection tailored to Swiss businesses and public sector organizations.
elcasecurity.ch • Geneva • Zurich • Bern • Basel • Palermo • Incident Response • Information Security • Cybersecurity • 501-1000 employees
Geneva • Zurich • Bern • Basel • Palermo • Incident Response • Information Security • Cybersecurity • 501-1000 employees
Verified live · 4 hours ago 2 months ago

Agentic Java Engineer

$102k – $191k per year (Estimated) • Senior • 5+ years expIn office (Zurich, Switzerland) • Senior • German: C1 • English: C1
Java
TypeScript
JavaScript
Java
Hibernate
Spring Boot
AI/ML
Claude
Claude Code
Copilot
JAX
AI Agents
Frontend
Angular
DevOps
CI/CD
GitHub
Apply
Report
The U.S. Securities and Exchange Commission (SEC) is an independent federal agency tasked with protecting investors and maintaining fair, orderly, and efficient financial markets. Established in the wake of the 1929 stock market crash, its primary mandate is to enforce federal securities laws, facilitate capital formation, and ensure that companies provide accurate information to the public. The commission oversees key market participants, including stock exchanges, brokers, and investment advisors, to promote transparency and prevent fraud within the financial system.
sec.gov • HQ: Washington, United States • Financial Services • Digital Government • Government • 5000+ employees
HQ: Washington, United States • Financial Services • Digital Government • Government • 5000+ employees
Verified live · 2 hours ago 2 months ago

Research Scientist

$96k – $201k per year (Estimated)In office (Cambridge, United Kingdom) • PhD
C++
Python
C++
PyTorch C++
TensorFlow C++
AI/ML
AI Agents
Fine-tuning
JAX
Knowledge Distillation
LLM
Multimodal AI
PyTorch
Quantization
TensorFlow
DevOps
Git
GitHub
Apply
Report

Zoox

zoox.com
Zoox is an American technology company headquartered in Foster City, California, that was founded in 2014. The company develops a purpose-built, fully autonomous electric vehicle designed specifically for ride-hailing services rather than individual ownership. Operating as an independent subsidiary of Amazon, the firm is currently testing its robotaxi fleet in several major United States cities including San Francisco and Las Vegas.
zoox.com • HQ: Foster City, United States • Robotics • Smart Mobility • Automotive Manufacturing • Manufacturing • Transportation & Logistics • Artificial Intelligence • Ride Hailing • Autonomous Driving • Est. 2014
HQ: Foster City, United States • Robotics • Smart Mobility • Automotive Manufacturing • Manufacturing • Transportation & Logistics • Artificial Intelligence • Ride Hailing • Autonomous Driving • Est. 2014
$182k – $257k per year • Equity • Junior • 2+ years exp • Remote (United States) • Visa sponsorship • Bachelor's Degree • Junior
DeepSpeed
JAX
PyTorch
Ray
Reinforcement Learning
Edge AI
DevOps
AWS
Robotics
Reinforcement Learning
Apply
Top 25% payVerified live · 2 hours ago 2 months ago

Software Engineer, ML Platform (ML Training)

$192k – $257k per year • Junior • 2+ years expRemote/Hybrid (Foster City, United States) • Full-Time • Junior
DeepSpeed
JAX
PyTorch
Ray
Reinforcement Learning
Edge AI
DevOps
AWS
Apply
Report

Capgemini

capgemini.com
Capgemini is a global leader in partnering with companies to transform and manage their business by harnessing the power of technology. Headquartered in Paris, France, the multinational corporation offers a comprehensive suite of services that includes consulting, digital transformation, technology, and engineering.
capgemini.com • HQ: Paris, France • Professional Services • Information Technology • IT Consulting • Est. 1967
HQ: Paris, France • Professional Services • Information Technology • IT Consulting • Est. 1967
2 months ago

GCP AI Developer

$64k – $110k per year (Estimated)Remote/Hybrid (Warsaw, Poland) • Bachelor's Degree • Full-Time • Polish
Python
Databases
Google BigQuery
AI/ML
Gemini
JAX
Prompt Engineering
PyTorch
RAG
TensorFlow
Vertex AI
DevOps
GCP
AWS
Azure
CI/CD
Docker
Google Cloud Run
Kubernetes
Rest API
Apply
Report

Gateworth Group

gateworth.com
Gateworth Group is an executive search and talent acquisition firm based in Dubai, specializing in recruitment across technology, finance, engineering, and corporate operations. The company connects multinational corporations, single-family offices, and emerging tech enterprises with specialized professionals and leadership talent. Through tailored recruitment services and market intelligence, it helps organizations source critical skill sets to drive regional and global business growth.
gateworth.com • Dubai • Professional Services
Dubai • Professional Services
2 months ago

Senior AI Engineer

Senior • Dubai • Remote (United Arab Emirates) • Relocation • Master's Degree • Full-Time • Senior
Fine-tuning
JAX
LLM
Multimodal AI
PyTorch
RAG
TensorFlow
Transformers
Hugging Face
AI Agents
Apply
Report

Nubank

nubank.com.br
Nubank (Nu Holdings Ltd.) is the largest digital banking platform and fintech company in Latin America, headquartered in São Paulo, Brazil. Founded in 2013 by David Vélez, Cristina Junqueira, and Edward Wible, Nubank launched as a pioneer in fee-free credit cards managed entirely through a mobile app. It has since grown into a full-scale financial services platform serving over 100 million customers across Brazil, Mexico, and Colombia.
nubank.com.br • HQ: São Paulo, Brazil • Financial Services • FinTech • Banking • 1001-5000 employees • Est. 2013
HQ: São Paulo, Brazil • Financial Services • FinTech • Banking • 1001-5000 employees • Est. 2013
Top 25% payVerified live · 2 hours ago 2 months ago

Staff Machine Learning Engineer

$230k – $345k per year • Equity • Staff+ • 7+ years expRemote/Hybrid (Palo Alto, United States) • Relocation • Full-Time • Staff
Python
AI/ML
JAX
Multimodal AI
PyTorch
TensorFlow
Transformers
Analytics
A/B Testing
Apply
Report

Adobe

adobe.com
Adobe is an American software company founded in 1982 by John Warnock and Charles Geschke and headquartered in San Jose, California. It created the PostScript and PDF formats and built the industry-standard creative toolset around Photoshop, Illustrator, Premiere Pro, After Effects and InDesign, now delivered as the Creative Cloud subscription. The company also runs Document Cloud for electronic signatures and document workflows, Experience Cloud for enterprise marketing analytics and personalisation, and the commercially safe Firefly family of generative AI models.
adobe.com • HQ: San Jose, United States • Sales & Marketing • Document Management • Artificial Intelligence • LLM & Generative AI • Software • Design & Creative • Graphic Design • Productivity Software • Digital Design • 5000+ employees • Est. 1982
HQ: San Jose, United States • Sales & Marketing • Document Management • Artificial Intelligence • LLM & Generative AI • Software • Design & Creative • Graphic Design • Productivity Software • Digital Design • 5000+ employees • Est. 1982
Top 25% payVerified live · 2 hours ago 2 months ago

Principal Engineer AI/ML

$206k – $379k per year • Equity • Lead • 14+ years expIn office (San Jose, United States) • Bachelor's Degree • Full-Time • Principal
Python
AI/ML
Computer Vision
Contrastive Learning
CUDA
CUDA Toolkit
Fine-tuning
JAX
Keras
LLM
LoRA
Multimodal AI
NLP
Prompt Engineering
PyTorch
QLoRA
RAG
Reinforcement Learning
TensorFlow
PEFT
Feature Store
GraphRAG
Hugging Face
Human-in-the-Loop
LLMOps
AI Agents
DevOps
AWS
Azure
GCP
Platform Engineering
Vector
Analytics
A/B Testing
Apply
Top 25% pay 2 months ago

Lead Customer Facing Applied AI Engineer

$159k – $302k per year • Equity • Lead • 8+ years expIn office (San Jose, United States) • Bachelor's Degree • Full-Time • Staff
Java
Python
Scala
SQL
AI/ML
JAX
LLM
PyTorch
RAG
TensorFlow
A2A
LLM Evaluation
LLM Guardrails
LLMOps
DevOps
AWS
Azure
CI/CD
Datadog
Docker
GCP
Git
Grafana
gRPC
OpenTelemetry
Prometheus
Analytics
A/B Testing
Apply
Top 25% payVerified live · 2 hours ago 3 months ago

Staff Machine Learning Engineer

$173k – $307k per year • Equity • Staff+ • In office (San Jose, United States) • Master's Degree • Full-Time • Staff
Python
AI/ML
AI Agents
JAX
Multimodal AI
PyTorch
Ray
Spark
TensorFlow
Hugging Face
Recommender Systems
RAG
DevOps
Vector
Analytics
A/B Testing
Apply
Report

Comarch

comarch.pl
Comarch is a global software house and systems integrator headquartered in Poland that develops and implements proprietary IT solutions across international markets. The company serves a broad range of industries - including telecommunications, banking and finance, retail, healthcare, and public administration - offering enterprise software such as ERP systems, EDI, e-invoicing, and customer loyalty platforms. Leveraging cloud infrastructure, artificial intelligence, and cybersecurity expertise, it helps organizations worldwide modernize operations, automate complex business processes, and maintain regulatory compliance.
comarch.pl • HQ: Kraków, Poland • Commerce • Banking • Financial Services • 1001-5000 employees
HQ: Kraków, Poland • Commerce • Banking • Financial Services • 1001-5000 employees
$66k – $114k per year (Estimated)In office (Kraków, Poland) • Full-Time • Polish: C1
Python
AI/ML
JAX
PyTorch
RAG
TensorFlow
Transformers
DevOps
CI/CD
Management
Google Workspace
Apply
Report
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.
allenai.org • HQ: Seattle, United States • Science & Engineering • Machine Learning • Artificial Intelligence
HQ: Seattle, United States • Science & Engineering • Machine Learning • Artificial Intelligence
Top 25% payVerified live · 2 hours ago 3 months ago

Senior Research Engineer, Olmo + Molmo

$147k – $220k per year • Senior • 4+ years expIn office (Seattle, United States) • Bachelor's Degree • Internship • Senior
Python
AI/ML
Fine-tuning
JAX
LLaVA
LLM
Multimodal AI
PyTorch
TensorFlow
OpenAI
AI Agents
Function Calling
DevOps
AWS
Docker
GCP
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.