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
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Leadership
Networking
Product
Design
Analytics
Support
Solutions
Enterprise Apps
Quantum

WOOD Federal Solutions

woodfederalsolutions.com
woodfederalsolutions.com • Laurel • Annapolis Junction • Fort Meade • Industrial Engineering • Information Security • Cybersecurity • Hiring 8, 8 new
Laurel • Annapolis Junction • Fort Meade • Industrial Engineering • Information Security • Cybersecurity • Hiring 8, 8 new
WOOD Federal Solutions delivers mission-critical systems engineering, cybersecurity, and advanced automation services to Intelligence Community customers. 28+ years of prime performance supporting secure, high-trust missions.
No named decision-maker yet.
No careers mailbox known
Top 25% payVerified live · 3 hours ago 5 days ago Hiring now · 89%

Software Engineer (Mid or Senior)

$175k – $230k per year • Senior • 14+ years exp • In office (Annapolis Junction, United States) • Bachelor's Degree • TS/SCI • Senior
Python
JavaScript
SQL
C++
Groovy
Java
Kotlin
TypeScript
Java
Spring Framework
Jackson
Maven
Apache Tomcat
Kotlin
Mockito
C++
CMake
Databases
PostgreSQL
AI/ML
JAX
Frontend
Angular
Bootstrap
JQuery
Mobile
JUnit
DevOps
CI/CD
Configuration Management
Linux
Windows
Rest API
OpenShift
Podman
Jenkins
Docker
Kubernetes
SOAP
Management
Agile
Scrum
QA
Selenium
+26
Apply

Fidelity Investments

fidelity.com
fidelity.com • HQ: Boston, United States • Wealth Management & Financial Advisors • Online Brokerage & Trading Platforms • Asset Management & Funds • 1001-5000 employees • Est. 1946 • Hiring 453, 62 new
HQ: Boston, United States • Wealth Management & Financial Advisors • Online Brokerage & Trading Platforms • Asset Management & Funds • 1001-5000 employees • Est. 1946 • Hiring 453, 62 new
Fidelity Investments is a privately held financial services company headquartered in Boston, Massachusetts, offering brokerage and retirement accounts, mutual funds and ETFs, workplace benefits and stock plan administration, wealth management and institutional asset management and custody. Founded in 1946 and owned largely by the Johnson family and employees, it has major US campuses in Boston, Durham, Westlake, Salt Lake City and Greenwood Village, and offices in Ireland since 1996 and India. It hires software, cloud and mainframe engineers, compliance and fraud analysts, quantitative and product analysts, technology risk staff, investment consultants and accountants.
Mindy Burds · Executive Officer · decision-maker
Dan Zitelman · Executive Officer, Director · decision-maker
Bryan Vander Lee · Executive Officer, Director · decision-maker
Vivian Clemen · Executive Officer · decision-maker
No careers mailbox known
Verified live · 6 hours ago 5 days ago Likely open · 59%

Principal Java Full Stack Engineer (Devops, Azure, SQL)

≈ $118k – $203k per year (Estimated) • Equity • Lead • 10+ years exp • United States • Remote (United States) • Bachelor's Degree • Full-Time • Principal
Java
Kotlin
Java
Jackson
Apache Tomcat
Kotlin
Mockito
Databases
Redis
AI/ML
JAX
DevOps
Rest API
Rancher
Azure
CI/CD
Jenkins
Git
Docker
Kubernetes
Blue-Green Deployment
Analytics
A/B Testing
Management
Agile
QA
Swagger
+11
Apply

Perceptyne

perceptyne.com
perceptyne.com • HQ: Hyderabad, India • Computer Vision • Industrial Robotics • Humanoid Robots • 51-200 employees • Est. 2021 • Startup • Trust B • Hiring 27, 15 new
HQ: Hyderabad, India • Computer Vision • Industrial Robotics • Humanoid Robots • 51-200 employees • Est. 2021 • Startup • Trust B • Hiring 27, 15 new
Perceptyne is a deep-tech robotics and AI company developing industrial humanoid and semi-humanoid robots that automate complex, dexterous factory tasks using AI, advanced sensing and vision technologies. Its job ads cover customer project managers, computer vision engineers, field application engineers and production engineers. The company is based in Hyderabad, Telangana, India.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 5 days ago Likely open · 63%

Robotics Research Engineer (L4) — Physical AI & Robot Learning

Senior • 8+ years exp • In office • Bachelor's Degree • Full-Time • Staff
Python
C++
C++
PyTorch C++
AI/ML
CUDA Toolkit
Fine-tuning
Reinforcement Learning
JAX
Multimodal AI
Computer Vision
ONNX
TensorRT
PyTorch
CUDA
Vision-Language-Action
Physical AI
Embodied AI
Machine Learning
Robotics
Motion Planning
Teleoperation
Imitation Learning
Reinforcement Learning
+13
Apply

Hark

hark.com
hark.com • HQ: San Jose, United States • LLM & Generative AI • AI Research Labs • 1001-5000 employees • Est. 2007 • Trust A • Hiring 45, 2 new
HQ: San Jose, United States • LLM & Generative AI • AI Research Labs • 1001-5000 employees • Est. 2007 • Trust A • Hiring 45, 2 new
Hark is a technology company whose operating entity, Hark Labs Inc., develops hardware and desktop software, hiring members of technical staff for its Hark Desktop product, manufacturing quality engineers, information and application security engineers and enterprise sales managers. Its hiring is concentrated at its base on North First Street in San Jose. The company is headquartered in San Jose, California, United States.
Brett Adcock · Executive Officer, Director · decision-maker
Jesse Coors-Blankenship · Director
Colby Adcock · Director
Gregg Hill · Director
No careers mailbox known
Top 25% payVerified live · 24 min ago 5 days ago Hiring now · 90%

Member of Technical Staff, LLM & Production

$180k – $450k per year • Staff+ • 3+ years exp • In office (San Jose, United States) • Full-Time • Staff
Python
AI/ML
vLLM
JAX
Multimodal AI
Function Calling
AI Agents
SGLang
PyTorch
LLM
Synthetic Data
SFT
LLM Evaluation
Tool Use
+5
Apply

Coinis

coinis.com
coinis.com • HQ: Podgorica, Montenegro • Digital Marketing • Advertising • Est. 2009 • Hiring 4, 4 new
HQ: Podgorica, Montenegro • Digital Marketing • Advertising • Est. 2009 • Hiring 4, 4 new
Coinis is a Montenegrin technology company founded in 2009 and headquartered in Podgorica. It operates advertising technology platforms for publishers and advertisers worldwide. The company is one of the largest IT employers in the country.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 6 days ago Hiring now · 90%

Senior AI Engineer , Generative Creative Models

≈ $48k – $124k per year (Estimated) • Senior • Podgorica • Remote (Montenegro, European time zones hours) • Bachelor's Degree • Senior
Python
AI/ML
JAX
Prompt Engineering
Multimodal AI
PyTorch
Stable Diffusion
LoRA
Fine-tuning
Diffusion Models
ComfyUI
ControlNet
PEFT
Edge AI
DevOps
Cloudflare
+6
Apply

DNA325

dna325.com
dna325.com • Information Technology • IT Staffing & Staff Augmentation • 51-200 employees • Est. 2010 • Hiring 17, 17 new
Information Technology • IT Staffing & Staff Augmentation • 51-200 employees • Est. 2010 • Hiring 17, 17 new
Accelerate your hiring with DNA325, your IT recruitment Europe partner. Get qualified candidates, even for hard-to-fill roles, in as little as 3 days!
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago 6 days ago Hiring now · 90%

Senior Machine Learning Engineer

≈ $104k – $230k per year (Estimated) • Senior • Remote (Germany, Indonesia, Ireland, Poland, Spain) • Senior
Python
AI/ML
JAX
PyTorch
Apply
Verified live · 1 hour ago 6 days ago Hiring now · 90%

Applied AI Engineer (Spain, remote)

≈ $50k – $114k per year (Estimated) • Madrid • Remote (Spain)
Python
AI/ML
Fine-tuning
Machine Learning
Qwen
vLLM
JAX
PyTorch
OpenAI
Apply
Verified live · 1 hour ago 6 days ago Hiring now · 90%

Applied AI Engineer

Remote (Germany, Indonesia, Ireland, Poland, Spain)
Python
AI/ML
Multimodal AI
Knowledge Distillation
RAG
Mixture of Experts
LLM Guardrails
Multi-Agent Systems
Model Distillation
Machine Learning
JAX
PyTorch
+3
Apply

Lenovo

lenovo.com
lenovo.com • HQ: Beijing, China • Smartphones • Servers & Data Center Hardware • Laptops • 1001-5000 employees • Est. 1984 • Trust A • Hiring 412, 87 new
HQ: Beijing, China • Smartphones • Servers & Data Center Hardware • Laptops • 1001-5000 employees • Est. 1984 • Trust A • Hiring 412, 87 new
Lenovo Group Limited is a global technology enterprise specializing in personal computing, smart devices, data center infrastructure, and enterprise IT services. Headquartered in Beijing, China, and Morrisville, North Carolina, United States - with its registered office in Hong Kong - the publicly traded corporation operates across three primary business divisions: the Intelligent Devices Group (IDG), the Infrastructure Solutions Group (ISG), and the Solutions and Services Group (SSG). Through an extensive product and services portfolio encompassing ThinkPad and IdeaPad personal computers, Motorola smartphones, ThinkSystem enterprise servers, edge computing hardware, and hybrid cloud solutions, the enterprise provides hardware, software, and managed IT services to commercial clients, public institutions, and retail consumers across more than 180 markets worldwide.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago 7 days ago Fading · 42%

AI Data Infra Senior Researcher

≈ $37k – $95k per year (Estimated) • Senior • In office (Beijing, China) • Senior
Python
Go
Java
Rust
C++
C++
TensorFlow C++
PyTorch C++
Databases
ClickHouse
Apache Kafka
AI/ML
Spark
Fine-tuning
JAX
Flink
TensorFlow
PyTorch
RAG
Machine Learning
CUDA Toolkit
Time Series Forecasting
CUDA
NCCL
ROCm
+14
Apply
bvteck.com • HQ: Bridgewater Township, United States • IT Staffing & Staff Augmentation • HR & Recruiting AI • Est. 2020 • Staffing · EOR • Hiring 368, 103 new
HQ: Bridgewater Township, United States • IT Staffing & Staff Augmentation • HR & Recruiting AI • Est. 2020 • Staffing · EOR • Hiring 368, 103 new
Bright Vision Technologies is a minority-owned technology company that develops Lumina, an AI talent intelligence and enterprise automation platform offering candidate sourcing, semantic resume matching, recruiter assistants and blockchain-based credential verification. Alongside the product, it places IT consultants with clients, with openings for SAP functional consultants, Oracle Cloud architects, data architects, PeopleSoft and Dell Boomi integration specialists, and it runs an office in Chennai, India. The company is headquartered in Bridgewater, New Jersey, United States.
No named decision-maker yet.
Careers mailbox is known
7 days ago Hiring now · 90%

AI Infrastructure Engineer

$100k – $160k per year • Staff+ • 10+ years exp • Remote (United States) • Full-Time • Staff
Python
Go
C++
C++
PyTorch C++
AI/ML
DeepSpeed
JAX
PyTorch
Ray
Megatron-LM
FSDP
NCCL
InfiniBand
DevOps
SLURM
CI/CD
Kubernetes
HPC
Linux
FinOps
+10
Apply
7 days ago Hiring now · 90%

AI Scientist

$100k per year • Senior • 6+ years exp • Remote (United States) • Full-Time • Senior
Python
AI/ML
Fine-tuning
JAX
PyTorch
Machine Learning
Multimodal AI
AI Agents
RAG
Apply
9 days ago Hiring now · 78%

Applied AI Engineer

$130k – $180k per year • Staff+ • 10+ years exp • Remote (United States) • Full-Time • Staff
Python
Databases
Weaviate
Milvus
Pinecone
FAISS
AI/ML
Embeddings
JAX
Prompt Engineering
Multimodal AI
Computer Vision
AI Agents
NLP
TensorFlow
PyTorch
RAG
Recommender Systems
Machine Learning
LangGraph
LangChain
LlamaIndex
Fine-tuning
LLM
Ray
DevOps
GCP
Azure
CI/CD
AWS
Docker
Kubernetes
+21
Apply
argonne.gov • HQ: Lemont, United States • Research Institutes • Nuclear Energy • Quantum Science & Computing • 501-1000 employees • Est. 1946 • Trust A • Hiring 70, 5 new
HQ: Lemont, United States • Research Institutes • Nuclear Energy • Quantum Science & Computing • 501-1000 employees • Est. 1946 • Trust A • Hiring 70, 5 new
Argonne National Laboratory is a premier multidisciplinary science and engineering research center located in Lemont, Illinois (near Chicago). Originating in 1946 from the University of Chicago’s work on the Manhattan Project (Metallurgical Laboratory), Argonne operates under the oversight of the U.S. Department of Energy (DOE) Office of Science and is managed by UChicago Argonne, LLC.
No named decision-maker yet.
No careers mailbox known
Verified live · 7 hours ago 8 days ago Hiring now · 90%

Postdoctoral Researcher - AI for Earth System Sciences

$73k – $121k per year • Junior • In office (Lemont, United States) • PhD • Full-Time
Fine-tuning
JAX
Multimodal AI
PyTorch
GNN
Machine Learning
DevOps
HPC
Robotics
Digital Twin
Apply
Verified live · 7 hours ago 3 months ago Fading · 29%

Assistant Scientist – AI for Autonomous Synthesis and Multimodal Characterization

$94k – $147k per year • Senior • 5+ years exp • In office (Lemont, United States) • Bachelor's Degree • Full-Time • Senior
Python
AI/ML
Scikit-learn
JAX
Multimodal AI
TensorFlow
PyTorch
Ray
Machine Learning
Reinforcement Learning
AI Agents
DevOps
HPC
Robotics
ROS
Reinforcement Learning
Digital Twin
+6
Apply
Verified live · 7 hours ago 6 months ago Long shot · 20%

Staff Scientist – Post-Training and Reinforcement Learning for AI for Science

$94k – $147k per year • Staff+ • 5+ years exp • Hybrid (Lemont, United States) • Bachelor's Degree • Full-Time • Staff
Python
C++
C++
PyTorch C++
AI/ML
Fine-tuning
Reinforcement Learning
JAX
PyTorch
Post-training
Machine Learning
DPO
SFT
Reward Modeling
DevOps
HPC
+5
Apply

Brain Co.

brain.co
brain.co • HQ: San Francisco, United States • Health Insurance & Benefits • Artificial Intelligence • Operating Systems • AI Agents • Startup • Trust B • Hiring 32, 1 new
HQ: San Francisco, United States • Health Insurance & Benefits • Artificial Intelligence • Operating Systems • AI Agents • Startup • Trust B • Hiring 32, 1 new
Brain Co. builds agent native operating systems for modern institutions.
No named decision-maker yet.
Careers mailbox is known
Verified live · 1 hour ago 8 days ago Hiring now · 90%

Machine Learning Engineer, Applied AI Co-Op Intern, Winter 2027

Intern • Hybrid (San Francisco, United States) • Bachelor's Degree • Internship • Intern
Python
AI/ML
JAX
TensorFlow
PyTorch
LLM
RAG
Machine Learning
Apply

Sonata Software

sonata-software.com
sonata-software.com • HQ: Bengaluru, India • IT Outsourcing & Dedicated Teams • IT Consulting & Digital Transformation • Est. 1986 • Hiring 77, 12 new
HQ: Bengaluru, India • IT Outsourcing & Dedicated Teams • IT Consulting & Digital Transformation • Est. 1986 • Hiring 77, 12 new
Sonata Software is an Indian IT services and technology company headquartered in Bengaluru that provides modernization engineering across cloud, data, AI, Microsoft Dynamics, test automation and managed services. Founded in 1986 and listed in India, it runs the Harmoni.AI enterprise platform and works with Fortune 500 clients in banking, retail, healthcare, manufacturing and technology, with a presence in the US, UK, Malaysia, Mexico, Australia, the DACH region and the Nordics. It hires data and cloud architects, MLOps and DevOps engineers, .NET and full-stack developers, identity and security specialists, SharePoint administrators and JD Edwards consultants.
No named decision-maker yet.
Careers mailbox is known
8 days ago Hiring now · 88%

Full Stack Java Developer & Team Lead

≈ $20k – $46k per year (Estimated) • Lead • 8+ years exp • Hybrid (Pune, India) • Full-Time • Staff
Java
JavaScript
SQL
C#
Java
Hibernate
Maven
C#
.NET
Databases
MySQL
Snowflake
Neo4j
MS SQL
Amazon DocumentDB
AI/ML
JAX
Frontend
Bootstrap
JQuery
DevOps
Rest API
Git
AWS
GitHub
Apache HTTP Server
+12
Apply
8 days ago Hiring now · 89%

Senior Test Automation Engineer

≈ $11k – $27k per year (Estimated) • Senior • 5+ years exp • Hybrid (Pune, India) • Full-Time • Senior
Java
SQL
JavaScript
Java
Hibernate
Databases
Snowflake
MySQL
Neo4j
MS SQL
Amazon DocumentDB
AI/ML
JAX
Frontend
Bootstrap
JQuery
Mobile
JUnit
DevOps
CI/CD
AWS
Rest API
GitHub Actions
Jenkins
Git
GitHub
Apache HTTP Server
QA
TestNG
Selenium
Playwright
Cucumber
JMeter
Postman
Rest-Assured
+20
Apply
1 month ago Long shot · 21%

Technical Team Lead

≈ $26k – $69k per year (Estimated) • Lead • 8+ years exp • In office (Pune, India) • Staff
Java
SQL
JavaScript
Java
Maven
Hibernate
Databases
MySQL
MS SQL
AI/ML
JAX
Frontend
Bootstrap
JQuery
DevOps
Rest API
Git
GitHub
+5
Apply

EnvisionWare

envisionware.com
envisionware.com • Web Portals • Hiring 3
Web Portals • Hiring 3
Enriching Public Library Service Inside and Out. Discover how libraries have transformed their services with EnvisionWare. Our customer spotlights highlight the tangible benefits of our collaborations, from enhanced self-service to improved library service.
No named decision-maker yet.
No careers mailbox known
Verified live · 7 hours ago Visa sponsor 8 days ago Hiring now · 79%

AI-Directed Software Engineer

≈ $124k – $234k per year (Estimated) • Middle • 3+ years exp • In office (United States, Canada) • Full-Time • Middle
Python
Java
TypeScript
SQL
C#
JavaScript
Java
Spring Boot
Jackson
Maven
Apache Tomcat
C#
.NET
Databases
PostgreSQL
ActiveMQ
AI/ML
Copilot
Cursor
Claude Code
Model Context Protocol
Function Calling
AI Agents
AWS Bedrock
LLM
RAG
AWS Bedrock AgentCore
Human-in-the-Loop
LLM Guardrails
Tool Use
JAX
Frontend
Angular
React.js
Vite
PrimeNG
DevOps
Terraform
Helm
CloudFormation
Kustomize
AWS
Docker
Kubernetes
AWS Lambda
Amazon S3
IAM
Amazon CloudWatch
Windows
Cybersecurity
Least Privilege
Analytics
ETL/ELT
QA
Swagger
+38
Apply

HCLTech

hcltech.com
hcltech.com • HQ: Noida, India • Systems Integrators • IT Consulting & Digital Transformation • IT Outsourcing & Dedicated Teams • 5000+ employees • Est. 1976 • Hiring 2302, 841 new
HQ: Noida, India • Systems Integrators • IT Consulting & Digital Transformation • IT Outsourcing & Dedicated Teams • 5000+ employees • Est. 1976 • Hiring 2302, 841 new
HCLTech is a major Indian multinational information technology (IT) services and consulting company headquartered in Noida, Uttar Pradesh. Spun off from the original HCL Group in 1991, it ranks as one of India's largest technology companies alongside firms like TCS, Infosys, and Wipro.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago 8 days ago Hiring now · 87%

Solution Architect - Google Cloud Platform (GCP)

≈ $130k – $253k per year (Estimated) • Staff+ • 12+ years exp • In office (United States) • Bachelor's Degree • Architect
CUDA Toolkit
Vertex AI
JAX
TensorFlow
PyTorch
LLM
CUDA
TPU
Machine Learning
DevOps
Terraform
GCP
GitHub Actions
OpenTelemetry
Prometheus
CI/CD
GitOps
ArgoCD
Jenkins
Kubernetes
Grafana
Platform Engineering
Service Mesh
Google GKE
FinOps
Incident Management
IAM
+18
Apply

Advanced Space

advancedspace.com
advancedspace.com • HQ: Westminster, United States • Space Services • Satellite Navigation • 11-50 employees • Est. 2011 • Hiring 19, 2 new
HQ: Westminster, United States • Space Services • Satellite Navigation • 11-50 employees • Est. 2011 • Hiring 19, 2 new
Advanced Space is a space technology company founded in 2011 that provides mission design, navigation and operations services for lunar and deep space missions. It developed and operated NASA's CAPSTONE mission, which launched in June 2022 and reached its near rectilinear halo orbit around the Moon in November 2022, and it builds autonomous navigation software for cislunar spacecraft. The company is headquartered in Westminster, Colorado, United States.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 8 days ago Hiring now · 86%

Machine Learning Engineer (2-5 yrs)

$100k – $134k per year • Junior • 2+ years exp • Hybrid (Westminster, United States) • Relocation • Bachelor's Degree • Junior
Python
AI/ML
Reinforcement Learning
JAX
AI Agents
TensorFlow
PyTorch
Machine Learning
Robotics
Reinforcement Learning
Sim-to-Real
Digital Twin
+2
Apply
Verified live · 1 hour ago 15 days ago Hiring now · 86%

Machine Learning Engineer (5-8 yrs)

$124k – $171k per year • Senior • 5+ years exp • Hybrid (Westminster, United States) • Relocation • Bachelor's Degree • Senior
Python
AI/ML
Reinforcement Learning
JAX
AI Agents
TensorFlow
PyTorch
Machine Learning
Robotics
Reinforcement Learning
Sim-to-Real
Digital Twin
+2
Apply

Graphcore

graphcore.ai
graphcore.ai • HQ: Bristol, United Kingdom • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2016 • Trust A • Hiring 163, 3 new
HQ: Bristol, United Kingdom • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2016 • Trust A • Hiring 163, 3 new
Graphcore is a British semiconductor company founded in Bristol in 2016 that designs the Intelligence Processing Unit, a processor architected specifically for machine learning rather than adapted from graphics. Its chips place large amounts of memory directly on the die and expose fine-grained parallelism, an approach aimed at sparse and irregular models that map poorly onto conventional accelerators. The company sells IPU systems and the Poplar software stack to research and enterprise customers, and has operated as a wholly owned subsidiary of SoftBank Group since its acquisition in 2024.
Simon Knowles · Co-Founder & Chief Technology Officer · decision-maker
Nigel Toon · Co-Founder & Chief Executive Officer · decision-maker
Careers mailbox is known
Verified live · 2 hours ago 8 days ago Hiring now · 99%

Senior Machine Learning Engineer (Large Systems)

≈ $104k – $193k per year (Estimated) • Senior • Hybrid (Cambridge, London, Bristol, United Kingdom) • Master's Degree • Senior
Python
C++
C++
PyTorch C++
AI/ML
JAX
PyTorch
Machine Learning
CUDA Toolkit
CUDA
Triton
InfiniBand
NVLink
DevOps
Kubernetes
HPC
+5
Apply

Cognition

cognition.ai
cognition.ai • HQ: San Francisco, United States • AI Development Tools • AI Agents • Code Intelligence • 501-1000 employees • Est. 2023 • Startup • Trust B • Hiring 97, 2 new
HQ: San Francisco, United States • AI Development Tools • AI Agents • Code Intelligence • 501-1000 employees • Est. 2023 • Startup • Trust B • Hiring 97, 2 new
Cognition is an applied artificial intelligence lab founded in 2023 that created Devin, an autonomous software engineering agent able to plan, write and debug code across a full repository. The company drew attention for demonstrating long-horizon coding work rather than single-file completion, and it sells Devin to engineering organisations. In 2025 it acquired the Windsurf coding assistant team and product, broadening its developer tooling.
David Cronin · Executive Officer, Director · decision-maker
Gerald Wesel · Executive Officer, Director · decision-maker
Dale Gallaher · Executive Officer, Director · decision-maker
Jeff Williams · Director
No careers mailbox known
Verified live · 3 hours ago 8 days ago Hiring now · 86%

Research Engineer, Safety & Alignment

≈ $169k – $371k per year (Estimated) • In office (San Francisco, United States) • PhD • Full-Time
Python
AI/ML
Cursor
JAX
AI Agents
PyTorch
Post-training
Devin
Scale AI
Human-in-the-Loop
Interpretability
Reward Modeling
Machine Learning
+4
Apply

Cosine

cosine.sh
cosine.sh • HQ: London, United Kingdom • AI Agents • Code Intelligence • 51-200 employees • Est. 2023 • Startup • Hiring 7, 5 new
HQ: London, United Kingdom • AI Agents • Code Intelligence • 51-200 employees • Est. 2023 • Startup • Hiring 7, 5 new
Cosine builds an autonomous software engineering agent that learns a codebase. Its Genie model was trained on real developer problem solving traces. The company targets complex bug fixing and feature work.
No named decision-maker yet.
No careers mailbox known
Verified live · 21 min ago 8 days ago Hiring now · 89%

ML Systems Engineer - Model Training and Infrastructure (SWE-focused LLMs)

$107k – $147k per year • Equity • Middle • 3+ years exp • In office (London, United Kingdom) • Full-Time • Middle
Python
SQL
Databases
Apache Iceberg
DuckDB
AI/ML
Fine-tuning
Reinforcement Learning
JAX
TensorFlow
PyTorch
Synthetic Data
OpenAI
SFT
Reward Modeling
Machine Learning
LLM
DevOps
GCP
Azure
AWS
Docker
Kubernetes
Cybersecurity
GDPR
Management
Slack
+14
Apply
liquid.tech • HQ: Johannesburg, South Africa • IT Consulting & Digital Transformation • Internet Service Providers • 201-500 employees • Est. 2005 • Hiring 6, 3 new
HQ: Johannesburg, South Africa • IT Consulting & Digital Transformation • Internet Service Providers • 201-500 employees • Est. 2005 • Hiring 6, 3 new
Liquid Intelligent Technologies is a pan-African technology company with major operations in Johannesburg. It runs one of the largest fibre networks on the continent and provides cloud and cybersecurity services. The company is part of the Cassava Technologies group.
No named decision-maker yet.
No careers mailbox known
Verified live · 2 hours ago 8 days ago Hiring now · 90%

Senior AI Solutions Architect (Technical Pre-Sales)

Staff+ • 8+ years exp • In office (South Africa) • Bachelor's Degree • Architect
Python
C++
C++
TensorFlow C++
PyTorch C++
Databases
Snowflake
Databricks
AI/ML
Fine-tuning
JAX
TensorFlow
PyTorch
LLM
RAG
InfiniBand
DevOps
Terraform
Ansible
SLURM
Azure
AWS
Docker
Kubernetes
HPC
+13
Apply

ByteDance

bytedance.com
bytedance.com • HQ: Beijing, China • Advertising • Digital Media • Recommendation Systems • Video & Photo Sharing • Consumer Internet Groups • 1001-5000 employees • Est. 2012 • Trust A • Hiring 6309, 49 new
HQ: Beijing, China • Advertising • Digital Media • Recommendation Systems • Video & Photo Sharing • Consumer Internet Groups • 1001-5000 employees • Est. 2012 • Trust A • Hiring 6309, 49 new
ByteDance is a Chinese internet technology company founded in Beijing in 2012 that operates some of the world's largest content and commerce platforms. Its portfolio includes the short-video apps TikTok and Douyin, the news aggregator Toutiao, the video editor CapCut, the workplace suite Lark and the Doubao family of AI assistants and models. Recommendation systems trained on user behaviour sit at the centre of every product, and the company has grown into one of the highest-revenue private technology businesses in the world.
Liang Rubo · Co-Founder & Chief Executive Officer · decision-maker
Shou Zi Chew · Chief Executive Officer, TikTok · decision-maker
Zhang Yiming · Co-Founder · founder
No careers mailbox known
Verified live · 3 hours ago 8 days ago Hiring now · 86%

AI算法工具链工程师(存算与AI) - 芯片研发

In office (Beijing, China) • Full-Time • Chinese: C1
Python
C++
C++
PyTorch C++
LLVM
AI/ML
llama.cpp
JAX
PyTorch
Apache TVM
Apply
Verified live · 3 hours ago Visa sponsor 2 months ago Hiring now · 71%

Research Scientist Graduate (Seed Model - Multimodal Interaction & World Model) - 2026 Start (PhD)

≈ $174k – $385k per year (Estimated) • Junior • In office (San Jose, United States) • PhD • Full-Time • Junior
Reinforcement Learning
Multimodal AI
World Models
JAX
Transformers
PyTorch
Mixture of Experts
TPU
Robotics
Reinforcement Learning
+1
Apply
Verified live · 3 hours ago Visa sponsor 2 months ago Hiring now · 70%

Student Researcher - (Seed Model - Infra - Compiler) - 2026 Start (PhD)

≈ $228k – $500k per year (Estimated) • In office (San Jose, United States) • PhD • Internship • Intern
DeepSpeed
vLLM
JAX
SGLang
PyTorch
LLM
Megatron-LM
MLIR
CUDA Toolkit
Reinforcement Learning
CUDA
Triton
FSDP
TPU
NCCL
XLA
+8
Apply

Meta

meta.com
meta.com • HQ: Menlo Park, United States • Digital Media • Foundation Models • Advertising • Virtual & Augmented Reality • Consumer Internet Groups • Messaging Apps • Social Networks • 5000+ employees • Est. 2004 • Trust B • Hiring 1008, 111 new
HQ: Menlo Park, United States • Digital Media • Foundation Models • Advertising • Virtual & Augmented Reality • Consumer Internet Groups • Messaging Apps • Social Networks • 5000+ employees • Est. 2004 • Trust B • Hiring 1008, 111 new
Meta Platforms is an American technology company founded in 2004 as Facebook and renamed in 2021 to signal its investment in immersive computing. It operates the largest family of social and messaging applications in the world, comprising Facebook, Instagram, WhatsApp, Messenger and Threads, and earns almost all of its revenue from advertising sold against that audience. Headquartered in Menlo Park, California, the company also develops the open Llama language models and the Reality Labs division behind Quest headsets and Ray-Ban smart glasses.
Mark Zuckerberg · Founder, Chairman & Chief Executive Officer · decision-maker
John Sines · Executive Officer · decision-maker
Jordan Schwartz · Executive Officer · decision-maker
Meron Gribetz · Executive Officer, Director · decision-maker
No careers mailbox known
Verified live · 2 hours ago 8 days ago Likely open · 54%

Software Engineer, Systems ML - Compilers / Kernels

≈ $236k – $450k per year (Estimated) • Senior • 6+ years exp • In office (Bellevue, United States) • Bachelor's Degree • Full-Time • Senior
CUDA Toolkit
JAX
TensorFlow
PyTorch
CUDA
ROCm
Context Engineering
MLIR
Apache TVM
XLA
Machine Learning
Analytics
A/B Testing
+4
Apply
Verified live · 2 hours ago 26 days ago Long shot · 20%

Research Scientist, Contextual AI and Multimodal Agents

≈ $270k – $552k per year (Estimated) • Staff+ • 12+ years exp • In office (Redmond, United States) • Bachelor's Degree • Full-Time • Principal
Python
AI/ML
Reinforcement Learning
Quantization
JAX
Multimodal AI
Knowledge Distillation
Function Calling
AI Agents
PyTorch
Post-training
Tool Use
Model Distillation
Machine Learning
Mobile
Core ML
+6
Apply
Verified live · 2 hours ago 1 month ago Long shot · 14%

Partner Engineering GenAI - US

≈ $236k – $450k per year (Estimated) • Senior • 5+ years exp • Hybrid (Menlo Park, United States) • Bachelor's Degree • Full-Time • Senior
Python
Go
Java
Rust
C++
C++
PyTorch C++
AI/ML
llama.cpp
Claude
vLLM
Fine-tuning
JAX
Multimodal AI
AI Agents
SGLang
Transformers
PyTorch
Mixture of Experts
Context Engineering
Edge AI
Speculative Decoding
Machine Learning
+13
Apply
Machines can read this selection · checked 12:05 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.