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

9 Mothers

9mothers.com
9mothers.com • HQ: Austin, United States • Robotics • Artificial Intelligence • Defense AI • Drones & UAVs • Startup • Hiring 6, 1 new
HQ: Austin, United States • Robotics • Artificial Intelligence • Defense AI • Drones & UAVs • Startup • Hiring 6, 1 new
Headquartered in Austin, Texas, 9 Mothers is an aerospace and defense technology company specializing in AI-powered autonomous counter-drone systems. The enterprise develops advanced counter-small Unmanned Aerial Systems (C-sUAS) and close-range point defense platforms designed to detect, track, and neutralize fast-moving aerial threats. By combining high-speed motion engineering, passive acoustic sensing, and in-house computer vision models, the organization delivers agile, cost-effective defensive solutions built for modern tactical environments.
No named decision-maker yet.
Careers mailbox is known
Open 184 daysTop 25% pay 6 months ago Long shot · 6%

Perception Engineer

$110k – $300k per year • Senior • 6+ years exp • In office (Austin, United States) • Relocation • Full-Time • Senior
Rust
C++
Python
AI/ML
Quantization
JAX
TensorRT
Synthetic Data
Computer Vision
PyTorch
Machine Learning
Multimodal AI
Robotics
Sensor Fusion
Sim-to-Real
+5
Apply

Forus

forus.com
forus.com • HQ: New York, United States • Care Coordination & Population Health • Patient Access & Booking • Medical AI • 501-1000 employees • Est. 2013 • Trust C • Hiring 53, 1 new
HQ: New York, United States • Care Coordination & Population Health • Patient Access & Booking • Medical AI • 501-1000 employees • Est. 2013 • Trust C • Hiring 53, 1 new
Forus is a New York-based health technology company building an AI-powered network that connects doctors, pharmacies, payers and biopharma companies so that patients can start prescribed medicines faster. Its platform automates insurance coverage, financial assistance and supply chain navigation at no cost to doctors and patients, supports patients in more than 85 percent of US ZIP codes, works with 9 of the 15 largest biopharma companies and is backed by more than $300 million in funding. Forus hires software, security and platform engineers, provider growth sales staff, clinical partnership leads, compliance and legal specialists, and recruiting and product roles in New York.
Jeff Schulte · Director · decision-maker
David Ramirez · Executive Officer, Director · decision-maker
Shinichiro Inoue · Executive Officer, Director · decision-maker
Nick Shalek · Director
No careers mailbox known
Likely inactiveTop 25% pay Visa sponsor 6 months ago Long shot · 3%

Research Engineer

$170k – $300k per year • In office (New York, United States) • Full-Time
Fine-tuning
Reinforcement Learning
JAX
Multimodal AI
NLP
Mamba
Few-Shot Learning
Interpretability
TensorFlow
PyTorch
LLM
AI Agents
+4
Apply

SpaceX

spacex.com
spacex.com • HQ: Starbase, United States • Satellite Manufacturing • Satellite Communications • Launch Services • 5000+ employees • Est. 2002 • Trust A • Hiring 2535, 106 new
HQ: Starbase, United States • Satellite Manufacturing • Satellite Communications • Launch Services • 5000+ employees • Est. 2002 • Trust A • Hiring 2535, 106 new
SpaceX is an American aerospace manufacturer and launch services provider founded by Elon Musk in 2002 with the goal of making life multiplanetary. It developed the reusable Falcon 9, now by far the most flown orbital rocket in the world, the Dragon capsule that carries NASA astronauts to the International Space Station, and Starship, the fully reusable super-heavy vehicle intended for lunar and Mars missions. The company also operates Starlink, a satellite constellation of thousands of spacecraft providing broadband to consumers, aircraft, ships and militaries, which has become its largest source of revenue.
Elon Musk · Founder, Chief Executive Officer & Chief Engineer · decision-maker
Gwynne Shotwell · President & Chief Operating Officer · decision-maker
Tom Mueller · Founding Employee & former Propulsion Chief
Careers mailbox is known
Top 25% payVerified live · 7 min ago 6 months ago Long shot · 19%

Sr. Software Engineer, Computer Vision

$160k – $225k per year • Equity • Senior • 5+ years exp • In office (Hawthorne, United States) • Bachelor's Degree • Nationals only • Senior
Python
C++
Databases
RabbitMQ
Apache Kafka
PostgreSQL
AI/ML
Airflow
Kubeflow
Ray
Computer Vision
Machine Learning
OpenCV
MLFlow
Fine-tuning
JAX
Edge AI
PyTorch
LLM
AI Agents
DevOps
CI/CD
Prometheus
Grafana
Docker
Linux
Kubernetes
Robotics
Sensor Fusion
Analytics
A/B Testing
+18
Apply

Onto Innovation

ontoinnovation.com
ontoinnovation.com • HQ: Wilmington, United States • Test & Measurement Instruments • Semiconductor Equipment • 1001-5000 employees • Est. 2019 • NYSE: ONTO • Trust B • Hiring 134, 12 new
HQ: Wilmington, United States • Test & Measurement Instruments • Semiconductor Equipment • 1001-5000 employees • Est. 2019 • NYSE: ONTO • Trust B • Hiring 134, 12 new
Onto Innovation is a semiconductor equipment and software company headquartered in Wilmington, Massachusetts, supplying process control tools for chip and advanced packaging manufacturing. Formed in 2019 through the merger of Rudolph Technologies and Nanometrics, it provides 3D and optical critical dimension metrology, wafer and package macro defect inspection such as the Dragonfly systems, factory analytics software and lithography for advanced packaging, with service teams near fabs in Taiwan, Korea, Japan, Singapore and the United States. It hires field service, applications, mechanical, electrical and software engineers, QA and manufacturing staff, and program managers.
No named decision-maker yet.
Careers mailbox is known
Open 187 daysVerified live · 1 day ago 6 months ago Long shot · 7%

Senior R&D Scientist

≈ $49k – $109k per year (Estimated) • Equity • Senior • 4+ years exp • In office (Singapore) • Master's Degree • Full-Time • Senior
MATLAB
Python
AI/ML
Reinforcement Learning
Scikit-learn
JAX
TensorFlow
PyTorch
Machine Learning
Chips/EDA
Photolithography
+1
Apply

Logical Intelligence

logicalintelligence.com
logicalintelligence.com • HQ: San Francisco, United States • Code Intelligence • Industrial AI • Artificial Intelligence • Est. 2022 • Startup • Trust B • Hiring 3
HQ: San Francisco, United States • Code Intelligence • Industrial AI • Artificial Intelligence • Est. 2022 • Startup • Trust B • Hiring 3
Logical Intelligence engineers deterministic AI for critical systems. We deliver mathematically verified intelligence beyond LLMs for autonomy, finance, & defense. Explore Math AI & EBMs.
No named decision-maker yet.
No careers mailbox known
Open 187 daysVerified live · 2 hours ago 6 months ago Long shot · 5%

AI Engineer - Reinforcement Learning

≈ $155k – $300k per year (Estimated) • Middle • 3+ years exp • In office (San Francisco, United States) • Middle
Go
C++
Python
Rust
C++
TensorFlow C++
AI/ML
Fine-tuning
JAX
TensorFlow
PyTorch
LLM
Machine Learning
DevOps
Kubernetes
Web3
Smart Contracts
+5
Apply
Open 261 daysVerified live · 2 hours ago 8 months ago Long shot · 5%

AI Engineer in Synthetic Data

≈ $151k – $292k per year (Estimated) • Middle • 3+ years exp • In office (San Francisco, United States) • Middle
C++
Python
Rust
Go
C++
TensorFlow C++
AI/ML
JAX
Data Augmentation
Synthetic Data
TensorFlow
PyTorch
Machine Learning
DevOps
GCP
Kubernetes
Azure
AWS
Web3
Smart Contracts
+8
Apply
Open 261 daysVerified live · 2 hours ago 8 months ago Long shot · 6%

AI Researcher - Frontiers & Reasoning

≈ $169k – $323k per year (Estimated) • Senior • 5+ years exp • In office (San Francisco, United States) • PhD • Senior
Go
C++
Python
Rust
C++
TensorFlow C++
AI/ML
JAX
TensorFlow
PyTorch
Machine Learning
Web3
Smart Contracts
+2
Apply
Report

The Voleon Group

voleon.com
voleon.com • HQ: Berkeley, United States • Machine Learning • Financial AI • Hedge Funds & Alternative Investments • 501-1000 employees • Est. 2007 • Trust A • Hiring 56, 1 new
HQ: Berkeley, United States • Machine Learning • Financial AI • Hedge Funds & Alternative Investments • 501-1000 employees • Est. 2007 • Trust A • Hiring 56, 1 new
The Voleon Group is a premier quantitative investment management firm headquartered in Berkeley, California, that specializes in machine learning-driven trading strategies. Founded in 2007 by scientists Michael Kharitonov and Jon McAuliffe, the hedge fund combines advanced data science, statistical modeling, and computational AI to forecast financial markets and execute systematic algorithmic trades.
No named decision-maker yet.
Careers mailbox is known
Top 25% payVerified live · 29 min ago 6 months ago Fading · 27%

Senior Cluster Site Reliability Engineer

$205k – $235k per year • Senior • 5+ years exp • Hybrid (Berkeley, United States) • Bachelor's Degree • Full-Time • Senior
Ruby
Python
Python
Dask
AI/ML
MLFlow
Kubeflow
Machine Learning
DeepSpeed
JAX
Modin
Ray
InfiniBand
TensorFlow
Spark
PyTorch
DevOps
Loki
OpenTelemetry
SLURM
Configuration Management
HPC
Amazon S3
Prometheus
Grafana
Ansible
Terraform
GCP
AWS
Podman
IAM
Docker
Kubernetes
Cybersecurity
Zero Trust
+23
Apply

Vultr

vultr.com
vultr.com • HQ: West Palm Beach, United States • 501-1000 employees • Est. 2014 • Startup • Trust B • Hiring 43, 1 new
HQ: West Palm Beach, United States • 501-1000 employees • Est. 2014 • Startup • Trust B • Hiring 43, 1 new
Vultr provides cloud compute, bare metal and graphics processing infrastructure. Its global footprint spans dozens of data centre locations at competitive prices. Developers and artificial intelligence teams use it as an independent cloud alternative.
No named decision-maker yet.
No careers mailbox known
Open 191 daysVerified live · 3 hours ago 6 months ago Long shot · 9%

Strategic Technical Account Manager GPU

≈ $74k – $171k per year (Estimated) • Junior • 2+ years exp • Remote (United States) • Full-Time • Junior
DeepSpeed
Fine-tuning
Embeddings
JAX
Ray
CUDA
Hugging Face
Megatron-LM
InfiniBand
NVLink
ROCm
TensorFlow
PyTorch
RAG
LLM
DevOps
SLURM
Vultr
HPC
Linux
Kubernetes
+12
Apply

Zencore

zencore.dev
zencore.dev • HQ: San Francisco, United States • Cloud Consulting & Migration • 51-200 employees • Est. 2013 • Trust C • Hiring 18
HQ: San Francisco, United States • Cloud Consulting & Migration • 51-200 employees • Est. 2013 • Trust C • Hiring 18
Zencore is a cloud consulting firm focused on Google Cloud that helps companies plan and carry out cloud migrations, data platform builds, infrastructure modernization and managed operations. Its engineers and architects work with clients in the United States, the United Kingdom and Latin America. The company is headquartered in San Francisco, California, United States.
No named decision-maker yet.
Careers mailbox is known
Likely inactiveVerified live · 6 min ago 6 months ago Long shot · 8%

Principal Architect, AI/ML

≈ $135k – $247k per year (Estimated) • Lead • Remote (United States) • Master's Degree • Full-Time • Architect
LangGraph
LoRA
vLLM
Vertex AI
Fine-tuning
Quantization
JAX
Langfuse
LangSmith
TensorRT-LLM
Llama
Mistral
Gemini
Google ADK
TPU
LLM Guardrails
XLA
Agentic Workflows
Vertex AI Agent Builder
LangChain
PyTorch
Claude
LLM
Machine Learning
AI Agents
TensorRT
PEFT
DevOps
Google GKE
GCP
Azure
AWS
Kubernetes
Cybersecurity
GDPR
+25
Apply

webAI

webai.com
webai.com • HQ: Austin, United States • Machine Learning • AI Development Tools • AI Compute & Inference • Edge AI • Est. 2019 • Startup • Trust B • Hiring 20, 2 new
HQ: Austin, United States • Machine Learning • AI Development Tools • AI Compute & Inference • Edge AI • Est. 2019 • Startup • Trust B • Hiring 20, 2 new
webAI is an artificial intelligence platform company headquartered in Austin, Texas, and founded in 2019 by David Stout, Tyler Mauer, and Ethan Baird. The company builds software that trains and runs large models on hardware the customer already owns, including its Navigator model building tool, webFrame for spreading models across local clusters, and a device native assistant. It targets organizations in aviation, healthcare, manufacturing, financial services, and the public sector that need AI without sending data to a public cloud.
No named decision-maker yet.
Careers mailbox is known
Likely inactiveVerified live · 2 hours ago Visa sponsor 6 months ago Long shot · 8%

Director of Applied AI

≈ $172k – $323k per year (Estimated) • Head+ • 8+ years exp • Hybrid (Austin, United States) • Bachelor's Degree • Full-Time • Architect
Python
AI/ML
Fine-tuning
JAX
Multimodal AI
Edge AI
TensorFlow
PyTorch
Machine Learning
Apply
Likely inactiveVerified live · 2 hours ago Visa sponsor 6 months ago Long shot · 8%

Senior Forward Deployed AI Engineer

≈ $140k – $267k per year (Estimated) • Senior • Hybrid (Austin, Washington, United States) • Master's Degree • Full-Time • Senior
JAX
Function Calling
ONNX
Agentic Workflows
TensorFlow
PyTorch
Machine Learning
AI Agents
Apply
Likely inactiveVerified live · 2 hours ago Visa sponsor 10 months ago Long shot · 7%

AI Research Scientist

≈ $131k – $255k per year (Estimated) • Middle • 4+ years exp • Hybrid (Austin, United States) • Bachelor's Degree • Full-Time • Middle
Python
AI/ML
Fine-tuning
Quantization
JAX
Multimodal AI
Knowledge Distillation
Diffusion Models
Edge AI
Model Distillation
TensorFlow
PyTorch
Machine Learning
Reinforcement Learning
Federated Learning
DevOps
Platform Engineering
+7
Apply

Goodfire

goodfire.com
goodfire.com • HQ: San Francisco, United States • Explainable AI (XAI) • AI Safety & Alignment • AI Research Labs • 201-500 employees • Est. 2023 • Startup • Trust C • Hiring 25
HQ: San Francisco, United States • Explainable AI (XAI) • AI Safety & Alignment • AI Research Labs • 201-500 employees • Est. 2023 • Startup • Trust C • Hiring 25
Goodfire is an artificial intelligence research company focused on mechanistic interpretability, building tools and methods to understand and edit the internal representations of neural networks. Founded in 2023, it develops the Ember interpretability platform, publishes research on model internals and partners with AI developers and scientific organisations to make models more reliable and safe. The company is headquartered in San Francisco, California, United States.
Eric Ho · Executive Officer, Director · decision-maker
Yan-David Erlich · Director
Daniel Balsam · Director
Tom McGrath · Director
Careers mailbox is known
Likely inactiveTop 25% pay 6 months ago Long shot · 3%

Forward Deployed Engineer

$200k – $325k per year • In office (San Francisco, United States)
Python
AI/ML
JAX
Interpretability
OpenAI
PyTorch
Fine-tuning
Apply
Likely inactiveTop 25% pay 9 months ago Long shot · 7%

Head of Engineering, Platform

$300k – $450k per year • Head+ • 10+ years exp • In office (San Francisco, United States) • Architect
JAX
Interpretability
OpenAI
PyTorch
Machine Learning
Apply

H Company

hcompany.ai
hcompany.ai • HQ: Paris, France • Foundation Models • AI Research Labs • AI Agents • 201-500 employees • Est. 2023 • Startup • Trust A • Hiring 33, 1 new
HQ: Paris, France • Foundation Models • AI Research Labs • AI Agents • 201-500 employees • Est. 2023 • Startup • Trust A • Hiring 33, 1 new
H Company is an artificial intelligence research company headquartered in Paris, France, and founded in 2023 by Charles Kantor with a founding team drawn largely from Google DeepMind. The company develops agentic AI systems, including Runner H, an agent that carries out multi-step tasks in a browser on the user's behalf. It raised one of the largest seed rounds in European history and positions itself as a French alternative in the agent tooling market.
No named decision-maker yet.
Careers mailbox is known
Open 193 daysVerified live · 4 hours ago 6 months ago Long shot · 12%

Research Engineer, Model Inference & Serving - Paris

≈ $57k – $123k per year (Estimated) • Hybrid (Paris, France) • Full-Time
Rust
C++
Python
AI/ML
vLLM
JAX
Multimodal AI
SGLang
PyTorch
AI Agents
llama.cpp
Quantization
TensorRT
TensorRT-LLM
MLX ML
CUDA
Triton
Edge AI
ONNX Runtime
Computer Use
Speculative Decoding
KV Cache
DevOps
Kubernetes
+14
Apply
Open 193 daysVerified live · 4 hours ago 6 months ago Long shot · 15%

Research Engineer / Scientist, Post-training & Reinforcement Learning - London

Hybrid (London, United Kingdom) • Master's Degree • Full-Time
Rust
Python
AI/ML
RLHF
Reinforcement Learning
JAX
Multimodal AI
Knowledge Distillation
Function Calling
NLP
Mixture of Experts
Synthetic Data
DPO
SFT
Computer Use
Model Distillation
Reward Modeling
TensorFlow
PyTorch
AI Agents
+11
Apply
Open 197 daysVerified live · 4 hours ago 6 months ago Long shot · 12%

Research Engineer, Model Inference & Serving - London

Hybrid (London, United Kingdom) • Full-Time
Rust
C++
Python
AI/ML
vLLM
JAX
Multimodal AI
SGLang
PyTorch
AI Agents
llama.cpp
Quantization
TensorRT
TensorRT-LLM
MLX ML
CUDA
Triton
Edge AI
ONNX Runtime
Computer Use
Speculative Decoding
KV Cache
DevOps
Kubernetes
+14
Apply

Perplexity AI

perplexity.ai
perplexity.ai • HQ: San Francisco, United States • Foundation Models • LLM & Generative AI • Search Engines • AI Search • 1001-5000 employees • Est. 2022 • Trust B • Hiring 131, 10 new
HQ: San Francisco, United States • Foundation Models • LLM & Generative AI • Search Engines • AI Search • 1001-5000 employees • Est. 2022 • Trust B • Hiring 131, 10 new
Perplexity AI is an American company that builds an answer engine combining live web search with large language models to return sourced, conversational responses instead of a list of links. Its products span a consumer assistant on web and mobile, the Comet browser, enterprise search over internal documents and the Sonar developer API that exposes the same grounded retrieval stack. Founded in 2022 in San Francisco by former researchers and engineers from OpenAI, Meta and Databricks, the company is backed by NVIDIA, IVP, New Enterprise Associates and SoftBank.
Johnny Ho · Co-Founder & Chief Strategy Officer · decision-maker
Denis Yarats · Co-Founder & Chief Technology Officer · decision-maker
Aravind Srinivas · Co-Founder & Chief Executive Officer · decision-maker
Aravind Lakshminarayanan · Director, Executive Officer · decision-maker
No careers mailbox known
Open 194 daysTop 25% pay Visa sponsor 6 months ago Long shot · 13%

Member of Technical Staff (AI Inference Engineer)

$220k – $485k per year • Staff+ • 3+ years exp • In office (San Francisco, New York, Palo Alto, United States) • Full-Time • Staff
Rust
Python
AI/ML
Multimodal AI
Perplexity
CUDA
Triton
CUTLASS
KV Cache
Quantization
JAX
NCCL
InfiniBand
NVLink
Speculative Decoding
TensorFlow
PyTorch
LLM
Frontend
Sass
DevOps
API Gateway
Kubernetes
+12
Apply
Open 199 daysVerified live · 4 hours ago Visa sponsor 6 months ago Long shot · 3%

Internship - Search Machine Learning Engineer

Intern • In office (London, United Kingdom) • Internship • Intern
Python
Rust
AI/ML
JAX
Perplexity
Recommender Systems
TensorFlow
PyTorch
RAG
LLM
Machine Learning
NLP
Analytics
A/B Testing
+4
Apply

Manifold Bio

manifold.bio
manifold.bio • HQ: Boston, United States • Drug Discovery • 11-50 employees • Startup • Trust B • Hiring 8, 1 new
HQ: Boston, United States • Drug Discovery • 11-50 employees • Startup • Trust B • Hiring 8, 1 new
Manifold Bio has built the first high-throughput in vivo (HTV) discovery engine—combining massively multiplexed in vivo screening and AI-powered design to create targeted biologics.
No named decision-maker yet.
Careers mailbox is known
Open 194 daysTop 25% pay 6 months ago Long shot · 4%

AI/ML Research Engineer

$140k – $225k per year • Equity • Junior • 2+ years exp • In office (Boston, San Francisco, United States) • Relocation • Bachelor's Degree • Junior
Python
AI/ML
Scikit-learn
JAX
Pandas
NumPy
PyTorch
Machine Learning
DevOps
Git
Docker
GCP
Kubernetes
AWS
Analytics
ETL/ELT
+5
Apply

GENIEE

geniee.co.jp
geniee.co.jp • HQ: Tokyo, Japan • Advertising • Est. 2010 • Trust A • Hiring 44, 2 new
HQ: Tokyo, Japan • Advertising • Est. 2010 • Trust A • Hiring 44, 2 new
GENIEE is a Tokyo advertising technology company founded in 2010 that builds supply side and demand side advertising platforms. Its systems help publishers in Japan and Southeast Asia maximise revenue from programmatic advertising, and it has acquired sales and marketing software businesses. The company operates across several Asian markets from its Tokyo base.
No named decision-maker yet.
No careers mailbox known
Verified live · 3 hours ago 6 months ago Likely open · 48%

【JAPAN AI】AI Evaluation Scientist / English

Equity • Middle • 3+ years exp • Hybrid (Tokyo, Japan) • Master's Degree • Full-Time • Middle • Japanese: C1 • English
Python
TypeScript
JavaScript
Databases
Google BigQuery
AI/ML
RLHF
JAX
Function Calling
NLP
Hallucination
Synthetic Data
Anthropic
DPO
Devin
Scale AI
Red Teaming
LLM Evaluation
Multi-Agent Systems
Reward Modeling
TensorFlow
Cursor
ChatGPT
Spark
PyTorch
Claude
LLM
Machine Learning
AI Agents
Weights & Biases
LangSmith
Pandas
Claude Code
Frontend
Next.js
React.js
DevOps
CI/CD
GitHub Actions
GitHub
Terraform
Docker
GCP
Kubernetes
Analytics
A/B Testing
Management
Linear
Confluence
QA
Pytest
+36
Apply
Verified live · 3 hours ago 6 months ago Likely open · 46%

【JAPAN AI】AI Evaluation Scientist / Japanese

In office (Tokyo, Japan) • Full-Time • Japanese: C1
TypeScript
Python
JavaScript
Databases
Google BigQuery
AI/ML
Weights & Biases
RLHF
JAX
LangSmith
Pandas
Anthropic
Google AI Studio
Devin
Scale AI
Cursor
ChatGPT
Claude Code
Spark
PyTorch
LLM
AI Agents
Frontend
Next.js
React.js
DevOps
GitHub Actions
GitHub
Terraform
Docker
GCP
Kubernetes
CI/CD
Management
Linear
Confluence
QA
Pytest
+24
Apply
Verified live · 3 hours ago 7 months ago Fading · 42%

【JAPAN AI】Research Engineer, LLM modeling / English

Equity • Hybrid (Tokyo, Japan) • Master's Degree • Full-Time • English • Japanese: C1
Python
TypeScript
JavaScript
AI/ML
Fine-tuning
RLHF
Reinforcement Learning
Quantization
JAX
Multimodal AI
Knowledge Distillation
Function Calling
NLP
Transformers
Synthetic Data
DPO
Google AI Studio
Multi-Agent Systems
Model Distillation
Cursor
ChatGPT
Prompt Engineering
PyTorch
Claude
LLM
Machine Learning
AI Agents
Weights & Biases
vLLM
Devin
Claude Code
Frontend
Next.js
React.js
DevOps
GitHub
Docker
GCP
Kubernetes
Management
Linear
Confluence
+30
Apply

Cornelis Networks

cornelis.com
cornelis.com • HQ: Wayne, United States • Information Technology • Artificial Intelligence • High-Performance Computing • 201-500 employees • Est. 2019 • Startup • Hiring 18, 3 new
HQ: Wayne, United States • Information Technology • Artificial Intelligence • High-Performance Computing • 201-500 employees • Est. 2019 • Startup • Hiring 18, 3 new
Cornelis Networks is a technology company headquartered in Wayne, Pennsylvania, United States, that designs high-performance networking solutions for artificial intelligence and high-performance computing clusters. The firm specializes in developing open active compute fabrics, host adapters, switches, and specialized software to maximize throughput and eliminate communication bottlenecks between accelerators. Originally spun out of Intel to advance the Omni-Path architecture, it provides enterprise data centers and research supercomputing facilities with scalable, ultra-low-latency interconnects for demanding computational workloads.
Lisa Spelman · Executive Officer, Director · decision-maker
Gunnar Gunnarsson · Executive Officer · decision-maker
Vladimir Tamarkin · Executive Officer · decision-maker
Philip Murphy Jr. · Executive Officer, Director · decision-maker
Careers mailbox is known
Open 198 daysVerified live · 4 hours ago 6 months ago Long shot · 9%

Senior Software Engineer – AI Middleware

Senior • 8+ years exp • In office (Austin, United States) • Full-Time • Senior
C++
C++
TensorFlow C++
AI/ML
DeepSpeed
JAX
CUDA
Megatron-LM
NCCL
ROCm
XLA
TensorFlow
PyTorch
DevOps
HPC
Linux
+5
Apply

NeoCognition

neocognition.io
neocognition.io • HQ: Palo Alto, United States • LLM & Generative AI • AI Agents • 51-200 employees • Startup • Trust B • Hiring 4
HQ: Palo Alto, United States • LLM & Generative AI • AI Agents • 51-200 employees • Startup • Trust B • Hiring 4
NeoCognition is an AI research lab focused on making LLM agents reliable, grounded and accessible to users, developers and enterprises. Its technical staff work on agent research and platform engineering for specialized intelligence. The company is headquartered in Palo Alto, California, United States.
No named decision-maker yet.
Careers mailbox is known
Open 198 daysVerified live · 1 hour ago Visa sponsor 6 months ago Long shot · 7%

Member of Technical Staff [Research]

≈ $195k – $392k per year (Estimated) • Staff+ • In office (Palo Alto, United States) • Full-Time • Staff
Python
AI/ML
Fine-tuning
RLHF
JAX
Function Calling
Post-training
Computer Use
Multi-Agent Systems
TensorFlow
PyTorch
LLM
Machine Learning
AI Agents
Llama
Mistral
+7
Apply

X2Robot

x2robot.com
x2robot.com • HQ: Shenzhen, China • Artificial Intelligence • 201-500 employees • Est. 2023 • Startup • Trust C • Hiring 156, 2 new
HQ: Shenzhen, China • Artificial Intelligence • 201-500 employees • Est. 2023 • Startup • Trust C • Hiring 156, 2 new
TwinDEX. A pair of co-designed, three-finger, nine-DoF dexterous manipulation interface: one wearable for data collection, one for robot deployment.
No named decision-maker yet.
Careers mailbox is known
Likely inactiveVerified live · 1 hour ago 6 months ago Long shot · 3%

具身算法工程师(system 0 方向)

In office (Shenzhen, China) • Full-Time • Chinese: C1
JAX
PyTorch
Robotics
MuJoCo
Isaac Lab
Sim-to-Real
Isaac Sim
Apply
Likely inactiveVerified live · 1 hour ago 6 months ago Long shot · 3%

具身算法工程师(灵巧操作)

In office (Shenzhen, China) • Full-Time • Chinese: C1
JAX
PyTorch
Robotics
MuJoCo
Isaac Lab
Isaac Sim
Apply
Likely inactiveVerified live · 1 hour ago 1 year ago Long shot · 3%

强化学习算法工程师(运控方向)

In office (Shenzhen, China) • Full-Time • Chinese: C1
JAX
TensorFlow
Robotics
Diffusion Policy
OpenVLA
Apply

Mindbeam

mindbeam.ai
mindbeam.ai • HQ: Melbourne, Australia • AI Development Tools • Machine Learning • Hiring 1
HQ: Melbourne, Australia • AI Development Tools • Machine Learning • Hiring 1
Litespark is a language model framework which utilises advanced algorithms to speedup training and inference workloads for generative AI applications.
No named decision-maker yet.
No careers mailbox known
Open 200 daysVerified live · 3 hours ago 6 months ago Long shot · 9%

Solutions Architect - Contractor

≈ $90k – $214k per year (Estimated) • Staff+ • 2+ years exp • Hybrid (Melbourne, Australia) • Bachelor's Degree • Contractor • Architect
JAX
TensorFlow
PyTorch
DevOps
Amazon EKS
Amazon EC2
Docker
Kubernetes
AWS
Apply
Report
gmppu.com • HQ: Concord, United States • Automotive Components • Motorsports • 501-1000 employees • Trust B • Hiring 57, 1 new
HQ: Concord, United States • Automotive Components • Motorsports • 501-1000 employees • Trust B • Hiring 57, 1 new
GM Performance Power Units is a Formula 1 power unit developer based in Concord, North Carolina, building hybrid and electric powertrains, sustainable fuel and high-efficiency combustion engine technology for the Cadillac Formula 1 team. The company is a partnership between General Motors and TWG Motorsports and is setting up its own team, processes and test facilities to design and build a race-winning F1 power unit. It hires mechanical design, additive manufacturing, controls and test cell engineers, CNC and facilities technicians, IT and HR staff, and a large group of summer interns and graduate placements.
No named decision-maker yet.
No careers mailbox known
Open 202 daysVerified live · 1 hour ago 6 months ago Long shot · 15%

ERS ML Design Optimization Engineer

≈ $119k – $231k per year (Estimated) • Middle • 3+ years exp • In office (Concord, United States) • Bachelor's Degree • Middle
MATLAB
AI/ML
JAX
Time Series
TensorFlow
PyTorch
DevOps
HPC
Apply

Axiom

axiommath.ai
axiommath.ai • HQ: San Francisco, United States • AI Development Tools • Machine Learning • Est. 2025 • Startup • Hiring 2
HQ: San Francisco, United States • AI Development Tools • Machine Learning • Est. 2025 • Startup • Hiring 2
Axiom is a research company building an artificial intelligence system for mathematics that can prove theorems and generate new results rather than only assist a human mathematician. Its work combines formal verification, proof search and reinforcement learning over symbolic reasoning, with the goal of a self improving mathematical reasoner. The laboratory was founded by researchers who see mathematics as the cleanest testbed for verifiable machine reasoning; its long term target is mathematics that machines can verify end to end without human checking.
No named decision-maker yet.
Careers mailbox is known
Open 204 daysVerified live · 2 hours ago 6 months ago Long shot · 9%

Member of Technical Staff

≈ $195k – $391k per year (Estimated) • Staff+ • In office (Palo Alto, Portland, United States, London, United Kingdom) • Full-Time • Staff
Python
AI/ML
Reinforcement Learning
JAX
TensorFlow
PyTorch
Machine Learning
Apply
Machines can read this selection · checked 04:04 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.