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+
Page 10
Filter in search →
Sr. Machine Learning Engineer, Autonomy
42dot - Sunnyvale, CA - On-site
$210k–300k4 days ago
Sr. Evaluation, Validation & Simulation Engineer
42dot - Sunnyvale, CA - On-site
$210k–300k4 days ago
Staff VLA Engineer
42dot - Sunnyvale, CA - On-site
$250k–325k4 days ago
Sr. Vehicle Operations & Integration Engineer
42dot - Sunnyvale, CA - On-site
4 days ago
Member of Technical Staff - Post-training (Data Infrastructure)
Cohere - United States; Canada - Remote
4 days ago
Member of Technical Staff - Post-Training (Reinforcement Learning & Generalization)
Cohere - New York, NY +1 - Remote
4 days ago
Technical Program Manager, AI Delivery
Cohere - United States; Canada - Remote
$140k–220k4 days ago
Member of Technical Staff, North Modelling (Evals)
Cohere - Toronto, ON - Remote
4 days ago
Software Engineer, Integrations
Cohere - London, ON +3 - Remote
4 days ago
Engineering Manager, FDE Agentic Platform
Cohere - Toronto, ON +1 - Remote
4 days ago
Member of Technical Staff, Agentic Environments
Cohere - New York, NY +2 - Remote
4 days ago
Product Manager, Integrations
Cohere - Toronto, ON +2 - Remote
$160k–285k4 days ago
Machine Learning Intern/Co-op (Winter 2027)
Cohere - Canada; Europe; United States; United Kingdom - Remote
4 days ago
Data Annotation Specialist, Engineering (civil, mechanical, electrical, chemical)
Cohere - Calgary, AB +3 - Remote
4 days ago
Research Internship (Winter 2027)
Cohere - San Francisco, CA +2 - Hybrid
4 days ago
Senior Member of Technical Staff, Safety and Security for Agents
Cohere - New York, NY +2 - Hybrid
4 days ago
Product Manager, Agent Harness & Modelling
Cohere - Toronto, ON +2 - Hybrid
$160k–320k4 days ago
Staff Research Engineer, Model Efficiency
Cohere - New York, NY +3 - Hybrid
4 days ago
Senior Security Engineer
Cohere - Toronto, ON +3 - Hybrid
4 days ago
Senior Research Scientist, Model Evaluation
Cohere - Toronto, ON +3 - Remote
4 days ago
Member of Technical Staff, Integration/RL Team (Research Engineer)
Cohere - New York, NY +2 - Remote
4 days ago
Member of Technical Staff, Data Analysis and Evaluation
Cohere - San Francisco, CA +3 - Remote
4 days ago
Member of Technical Staff, Post-Training
Cohere - San Francisco, CA +3 - Remote
4 days ago
Member of Technical Staff, Training Infra Engineer
Cohere - San Francisco, CA +3 - Remote
4 days ago
Member of Technical Staff, Training Performance Engineer
Cohere - New York, NY +2 - Remote
4 days ago