Latest AI JobsBrowse Jobs

Jax Jobs

10 open roles, updated daily

Filters3
10 jobs foundClear all
T
Machine Learning Engineer, Applied
tracelabs
13d 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 9d ago
PythonPyTorchJAX
$185K – $245KView Job
Writer logo
AI research scientist
Writer
28d ago

Join Writer as an AI Research Scientist in San Francisco or New York. Drive LLM research, agentic workflows, and production deployments.

San Francisco, CAEngineering, product & designFull-timeSenior · 7+ yrsRemoteOpen · checked 9d ago
PythonPyTorchJAX
$199K – $325KView Job
Writer logo
AI engineer
Writer
28d ago

Join Writer as an AI Engineer in San Francisco, New York, or Seattle. Build scalable enterprise LLMs and AI agents with a $198K-$259K salary range.

San Francisco, CAEngineering, product & designFull-timeSenior · 5+ yrsRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$198K – $259KView Job
Discord logo
Staff Software Engineer, Machine Learning (Consumer Revenue)
Discord
29d ago

Join Discord as a Staff ML Engineer on the Consumer Revenue team. Build large-scale recommendation systems with a base salary of $272k-$340k.

San Francisco Bay Area or Remote (U.S.)Product EngineeringFull-timeSenior · 8+ yrsRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$272K to $340KView Job
sandboxaq logo
Senior Research Scientist, Magnets
sandboxaq
29d ago

Apply for the Senior Research Scientist, Magnets role at SandboxAQ in the United States. Competitive salary, hybrid work, and AI materials discovery.

United StatesAI Simulation Full-timeSenior · 3+ yrsRemoteOpen · checked 9d ago
PythonPyTorchJAX
$134.4K – $252KView Job
Niantic Spatial logo
Senior Computer Vision Engineer (Localization)
Niantic Spatial
33d ago

Apply for the Senior Computer Vision Engineer role at Niantic Spatial in San Francisco or remote. Build physical AI mapping and VPS tech.

San Francisco, CAEngineeringFull-timeRemoteOpen · checked 9d ago
PythonPyTorchJAX
$189K – $255KView Job
Instacart logo
Senior Machine Learning Engineer, Ads Response Prediction
Instacart
35d ago

Join Instacart as a Senior ML Engineer on the Ads Response Prediction team. Remote in Canada (ON, AB, BC, NS). Base pay $180k-$190k CAD.

Canada - Remote (ON, AB, BC, or NS Only)Machine LearningFull-timeSenior · 3+ yrsRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$180K — $190K CADView Job
inductive-bio logo
Member of Technical Staff - Machine Learning
inductive-bio
39d ago

Apply for the Member of Technical Staff - Machine Learning role at inductive-bio in New York City, San Francisco, or Boston. Salary $180K–$250K + equity.

New York City, San Francisco, or BostonMachine LearningFull-timeSenior · 4+ yrsRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$180K – $250KView Job
Fireworks AI logo
Member of Technical Staff, Research (2026 PhD New Grad)
Fireworks AI
42d ago

Apply for the Member of Technical Staff, Research PhD New Grad role at Fireworks AI in San Mateo or remotely. Earn $200k-$300k yearly.

San MateoEngineeringFull-timeRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
$200K – $300KView Job
Cohere logo
Senior Software Engineer, GPU Infrastructure (HPC)
Cohere
47d ago

Join Cohere in Canada as a Senior Software Engineer for GPU Infrastructure. Build and scale ML superclusters with Python, Go, and Kubernetes.

CanadaModelingFull-timeSeniorRemoteOpen · checked 9d ago
PythonPyTorchTensorFlow
CA$285K – CA$340KView Job