Jax Jobs
23 open roles, updated daily
Filters3
Apply for the GPU Kernel Engineer role at Sciforium in San Francisco, CA. Salary: $190K–$250K with equity. Work on custom GPU kernels for AI systems.
Apply for the GPU Kernel Engineer role at Sciforium in San Francisco, CA. Salary: $190K–$250K with equity. Work on custom GPU kernels for AI systems.
Apply for the RL Systems Infra role at Thinking Machines in San Francisco. Distributed training, RL workloads, $425K-$475K salary, visa support.
Apply for the RL Systems Infra role at Thinking Machines in San Francisco. Distributed training, RL workloads, $425K-$475K salary, visa support.
Apply for the Senior Software Engineer role at Planet in Ljubljana. Hybrid position focusing on IAM services, Java, and cloud platforms.
Apply for the Senior Software Engineer role at Planet in Ljubljana. Hybrid position focusing on IAM services, Java, and cloud platforms.
Apply for the Senior HPC & GPU Infrastructure Engineer role at Sciforium in San Francisco, CA. Manage high-density GPU clusters with a $150K-$220K salary.
Apply for the Senior HPC & GPU Infrastructure Engineer role at Sciforium in San Francisco, CA. Manage high-density GPU clusters with a $150K-$220K salary.
Apply for the Senior Software Engineer, Vehicle Interaction role at Waymo in Mountain View, CA. Base salary $213,000 — $263,000 USD.
Apply for the Senior Software Engineer, Vehicle Interaction role at Waymo in Mountain View, CA. Base salary $213,000 — $263,000 USD.
Apply for the Senior Computer Vision Engineer role at Anduril Industries in Irvine, CA. Own the ML stack, edge compute models, and defense technology.
Apply for the Senior Computer Vision Engineer role at Anduril Industries in Irvine, CA. Own the ML stack, edge compute models, and defense technology.
Apply for the Research Engineer, Agentic EDA role at NormalComputing in New York City. Build multi-agent LLM systems. Salary: $200K-$400K + Equity.
Apply for the Research Engineer, Agentic EDA role at NormalComputing in New York City. Build multi-agent LLM systems. Salary: $200K-$400K + Equity.
Join Ai2 in Seattle as a Research Scientist for a 3-year project developing an AI-powered climate model. Base salary $167,030 - $260,570.
Join Ai2 in Seattle as a Research Scientist for a 3-year project developing an AI-powered climate model. Base salary $167,030 - $260,570.
Join Writer as an AI Research Scientist in San Francisco or New York. Drive LLM research, agentic workflows, and production deployments.
Join Writer as an AI Research Scientist in San Francisco or New York. Drive LLM research, agentic workflows, and production deployments.
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.
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.
Join Waymo as a Research Scientist in World Model Post-Training. Conduct applied research in Mountain View, San Francisco, or New York.
Join Waymo as a Research Scientist in World Model Post-Training. Conduct applied research in Mountain View, San Francisco, or New York.
Join Discord as a Staff ML Engineer on the Consumer Revenue team. Build large-scale recommendation systems with a base salary of $272k-$340k.
Join Discord as a Staff ML Engineer on the Consumer Revenue team. Build large-scale recommendation systems with a base salary of $272k-$340k.
Apply for the Senior Research Scientist, Magnets role at SandboxAQ in the United States. Competitive salary, hybrid work, and AI materials discovery.
Apply for the Senior Research Scientist, Magnets role at SandboxAQ in the United States. Competitive salary, hybrid work, and AI materials discovery.
Join Waymo in Mountain View or San Francisco as a Machine Learning Engineer. Build autonomous driving tech with PyTorch or JAX. Salary: $175K-$215K.
Join Waymo in Mountain View or San Francisco as a Machine Learning Engineer. Build autonomous driving tech with PyTorch or JAX. Salary: $175K-$215K.
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.
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 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.