Latest AI JobsBrowse Jobs

JAX Jobs

6 open roles, updated daily

Filters3
Work mode
Remote only
Job type
All typesFull-timeInternshipContractPart-time
Experience
Any levelInternshipEntryMidSeniorLeadExecutive
Posted
Any timeLast 24 hoursLast 7 daysLast 30 days
Salary
Salary listed
Popular skills
PythonPyTorchLLMMachine LearningComputer VisionNLPKubernetesAWSRustC++
6 jobs foundClear all
A
Machine Learning Engineer (2-5 yrs)
Advanced Space
10d ago

Apply for the Machine Learning Engineer role at Advanced Space in Westminster, CO. Hybrid position, salary $100K-$134K. Space exploration missions.

Westminster, COMachine LearningFull-timeMid · 2+ yrsOpen · checked 7d ago
PythonPyTorchTensorFlow
$100K -$134KView Job
T
Machine Learning Engineer, Applied
tracelabs
12d ago

Apply for the Applied Machine Learning Engineer role at tracelabs in the United States. $185K–$245K salary plus equity.

United StatesEngineeringFull-timeMid · 2+ yrsRemoteOpen · checked 7d ago
PythonPyTorchJAX
$185K – $245KView Job
A
Machine Learning Engineer (5-8 yrs)
Advanced Space
17d ago

Develop ML solutions for lunar and Mars missions. 5-8 years experience required in autonomy, navigation, and mission planning.

Westminster, COMachine LearningFull-timeMid · 5+ yrsOpen · checked 7d ago
PythonPyTorchTensorFlow
$124K -$171KView Job
T
Research Engineer, Climate Modeling
The Allen Institute for Artificial Intelligence
24d ago

Join Ai2 in Seattle as a Research Engineer for a 3-year climate modeling project. Work with JAX, Google Cloud, and ML models. $146k-$220k.

Seattle, WAApplied ScienceFull-timeMid · 2+ yrsOpen · checked 7d ago
PythonJAXMachine Learning
$146,880 - $220,320View Job
True Anomaly logo
Machine Learning Engineer (II-III), Space Edge Deployment
True Anomaly
40d ago

Join True Anomaly in Denver or Long Beach as a Machine Learning Engineer. Build autonomous space tech with a competitive salary and robust benefits.

Denver, Colorado or Long Beach, CaliforniaAutonomy EngineeringFull-timeMidOpen · checked 7d ago
PythonPyTorchTensorFlow
$125K - $220KView Job
W
Research Scientist, Map Scalability
Waymo
41d ago

Apply for the Research Scientist, Map Scalability role at Waymo in Mountain View, San Francisco, or New York City. Salary $213,000 - $263,000 USD.

Mountain View, CA, U.S.A ; San Francisco, CA, U.S.A ; New York City, NY, U.S.AAI Foundations (7SJ)Full-timeMid · 1+ yrsOpen · checked 7d ago
PythonPyTorchTensorFlow
$213K — $263K USDView Job