BLACK FRIDAY - Get 50% off new job postings now through Dec 31 using code BLACK50

JAX jobs

Open developer roles that use JAX, pulled straight from company job boards and refreshed every few hours.

405
open roles
380
posted this week
156
fully remote
144
pay $150k+
Cohere
Senior Member of Technical Staff, Multimodal AI
Cohere - San Francisco, CA +3 - Remote
4 days ago
Cohere
Member of Technical Staff, Modeling
Cohere - San Francisco, CA +3 - Remote
4 days ago
Cognition
Research Engineer, Safety & Alignment
Cognition - San Francisco, CA - On-site
4 days ago
Skydio
Autonomy Engineer Intern, Reinforcement Learning, 2027 (HQ)
Skydio - San Mateo, CA - On-site
$98k–146k4 days ago
OpenAI
Training Performance Engineer
OpenAI - San Francisco, CA - Hybrid
$295k–500k4 days ago
OpenAI
Research Engineer, Privacy
OpenAI - San Francisco, CA - On-site
Up to $500k4 days ago
Snowflake
AI Research Scientist, New Grad – Agents & Reinforcement Learning
Snowflake - Bellevue, WA - Hybrid
$176k–230k4 days ago
Snowflake
Post-Doctoral AI Researcher (2 year Fixed-Term)
Snowflake - Bellevue, WA - Hybrid
$160k–220k4 days ago
Harvey
Research Engineer, Infrastructure
Harvey - Remote - Remote
$230k–450k4 days ago
Harvey
Research Engineer, Infrastructure
Harvey - San Francisco, CA - On-site
$230k–450k4 days ago
Harvey
Research Engineer, Infrastructure
Harvey - New York, NY - On-site
$230k–450k4 days ago
Airwallex
Staff Engineer, Agentic AI
Airwallex - San Francisco, CA - Hybrid
$182k–310k4 days ago
Airwallex
Senior Software Engineer, Machine Learning Platform
Airwallex - Seattle, WA - Hybrid
$140k–245k4 days ago
Airwallex
Software Engineer, Agentic AI
Airwallex - San Francisco, CA - Hybrid
$120k–200k4 days ago
Bjak
AI Phone - Product Designer
Bjak - United States - Remote
4 days ago
Bjak
AI Phone - Product Lead
Bjak - United States - Remote
4 days ago
Bjak
AI Phone - AI Systems / Agent Architect
Bjak - United States - Remote
4 days ago
Bjak
Product Manager
Bjak - United States - Remote
4 days ago
Bjak
Lead Technical Product Manager
Bjak - United States - Remote
4 days ago
Bjak
Senior Technical Product Manager
Bjak - United States - Remote
4 days ago
Bjak
Chief Technology Officer
Bjak - United States - Remote
4 days ago
Bjak
Technical Product Lead - AI Stockbroking
Bjak - United States - Remote
4 days ago
Bjak
Technical Product Lead - AI Investing
Bjak - United States - Remote
4 days ago
Bjak
Technical Product Lead - AI Neobank
Bjak - United States - Remote
4 days ago
Bjak
Technical Product Lead - AI Finance
Bjak - United States - Remote
4 days ago