JAX Jobs
67 open roles, updated daily
Filters2
Lead the Fine-tuning Science team at Thinking Machines in San Francisco. Shape frontier customization techniques. Apply on their careers page.
Lead the Fine-tuning Science team at Thinking Machines in San Francisco. Shape frontier customization techniques. Apply on their careers page.
Apply for the Research, Tinker, RL Systems role at Thinking Machines in San Francisco or remote. Earn $350k-$475k with visa sponsorship.
Apply for the Research, Tinker, RL Systems role at Thinking Machines in San Francisco or remote. Earn $350k-$475k with visa sponsorship.
Apply for the 2027 Summer Intern, PhD, Machine Learning, Computer Vision role at Waymo in Mountain View, California.
Apply for the 2027 Summer Intern, PhD, Machine Learning, Computer Vision role at Waymo in Mountain View, California.
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.
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.
Apply for the ML Infrastructure Engineer role at Preference-Model in San Francisco. Build LLM post-training systems. Salary: $200K–$350K + equity.
Apply for the ML Infrastructure Engineer role at Preference-Model in San Francisco. Build LLM post-training systems. Salary: $200K–$350K + equity.
Join Preference-Model in San Francisco as a Technical Staff Research Engineer. Earn $200K-$350K plus equity, with visa sponsorship available.
Join Preference-Model in San Francisco as a Technical Staff Research Engineer. Earn $200K-$350K plus equity, with visa sponsorship available.
Apply for the Member of Technical Staff - Machine Learning Capabilities role at Preference-Model in San Francisco. Competitive salary and benefits.
Apply for the Member of Technical Staff - Machine Learning Capabilities role at Preference-Model in San Francisco. Competitive salary and benefits.
Join Discord's Safety ML team in the SF Bay Area. Build ML systems for content understanding and user safety. Base salary $220,000 to $275,000.
Join Discord's Safety ML team in the SF Bay Area. Build ML systems for content understanding and user safety. Base salary $220,000 to $275,000.
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.
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.
Join Waymo as a Senior ML Engineer in Predictive Planning. Build autonomous driving models with Python, C++, and deep learning. Salary $213k-$263k.
Join Waymo as a Senior ML Engineer in Predictive Planning. Build autonomous driving models with Python, C++, and deep learning. Salary $213k-$263k.
Apply for the 2027 Summer Intern PhD Vehicle Intent and Prediction role at Waymo in Mountain View, CA. Hourly pay of $85 USD.
Apply for the 2027 Summer Intern PhD Vehicle Intent and Prediction role at Waymo in Mountain View, CA. Hourly pay of $85 USD.
Join Anthropic in San Francisco as a Staff+ Software Engineer. Build ML inference infrastructure for safety systems. Salary: $320k-$485k.
Join Anthropic in San Francisco as a Staff+ Software Engineer. Build ML inference infrastructure for safety systems. Salary: $320k-$485k.
Join True Anomaly in Denver or Long Beach as a Machine Learning Engineer. Build autonomous space tech with a competitive salary and robust benefits.
Join True Anomaly in Denver or Long Beach as a Machine Learning Engineer. Build autonomous space tech with a competitive salary and robust benefits.
Apply for the Research Software Engineer, Post Training role at Thinking Machines in San Francisco. Salary: $350k-$475k. Remote options available.
Apply for the Research Software Engineer, Post Training role at Thinking Machines in San Francisco. Salary: $350k-$475k. Remote options available.
Apply for the Thermo ML Resident role at Extropic in Boston. Flexible part-time or full-time machine learning residency paying $75K–$200K.
Apply for the Thermo ML Resident role at Extropic in Boston. Flexible part-time or full-time machine learning residency paying $75K–$200K.
Join Extropic as a Thermo ML Resident in San Francisco. Flexible part-time or full-time role with $75K-$200K salary.
Join Extropic as a Thermo ML Resident in San Francisco. Flexible part-time or full-time role with $75K-$200K salary.
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.
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.
Apply for the Member of Technical Staff, Research PhD New Grad role at Fireworks AI in San Mateo or remotely. Earn $200k-$300k yearly.
Apply for the Member of Technical Staff, Research PhD New Grad role at Fireworks AI in San Mateo or remotely. Earn $200k-$300k yearly.
Join Cohere in Canada as a Senior Software Engineer for GPU Infrastructure. Build and scale ML superclusters with Python, Go, and Kubernetes.
Join Cohere in Canada as a Senior Software Engineer for GPU Infrastructure. Build and scale ML superclusters with Python, Go, and Kubernetes.