Filter By


Save Search

MORE FILTERS

LESS FILTERS

  • Job Types

386 Jobs in JAX framework

___
Full Time

3 Oct 2025

Summary

Our team crafts developer-first tooling that ensures developers get the best out of Arm technology on day one. We strive to support our outstanding developer ecosystem by provisioning it with...

Key Skills

Aartificial intelligence,Design,JAX framework,Machine learning techniques,Python Programming,PyTorch

___
Full Time

18 Sep 2025

Summary

We are currently focused on three key research areas:

Observability Foundation Models – Building state-of-the-art models for advanced forecasting, anomaly detection, and multi-modal...

Key Skills

Aartificial intelligence,C++,DataDog,JAX framework,Machine learning techniques,Python Programming,PyTorch,Rust,Site Reliability Engineering (SRE)

___
Full Time

17 Sep 2025

Summary

We are currently focused on three key research areas:

Observability Foundation Models – Building state-of-the-art models for advanced forecasting, anomaly detection, and multi-modal...

Key Skills

Aartificial intelligence,C++,DataDog,JAX framework,Python Programming,PyTorch,Rust

___
Full Time

29 Aug 2025

Summary

We are seeking an exceptional and creative Senior/Staff Machine learning Scientist to lead and innovate within our core AI research team. You will pioneer novel deep learning systems to tackle...

Key Skills

Aartificial intelligence,Effective communication skills,JAX framework,Machine learning techniques,NeurIPS,PyTorch

___
Full Time

24 Jul 2025

Summary

As an ML Engineer, you will drive the development of large-scale training and inference pipelines for foundation models, enabling agentic system to adapt to Apple’s diverse ecosystem. You will...

Key Skills

JAX framework,Machine learning techniques,Optimization,PyTorch,TensorFlow

___
Full Time

24 Jul 2025

Summary

As a senior machine learning engineer on the team, you will play a key role in our development. You will have responsibilities ranging from model design, implementation, training, model iteration,...

Key Skills

Aartificial intelligence,Generative AI,JAX framework,Large Language Models - LLMs,Machine learning techniques,NeurIPS,PyTorch,TensorFlow

___
Full Time

20 Jul 2025

Summary

As a Senior ML Engineer, you bring deep expertise in building robust production-grade machine learning systems and infrastructure. You’ll lead the design, development and maintenance of core...

Key Skills

Aartificial intelligence,Continuous Integration & Continuous Delivery - CI/CD,Google Cloud Platform (GCP),JAX framework,Kubeflow,Machine learning techniques,MLflow,PyTorch,TensorFlow

___
Full Time

28 Jun 2025

Summary

General description: A machine learning/artificial intelligence specialist with experience in LLM and agentic workflow for industrial engineering, system engineering and/or chemical...

Key Skills

Aartificial intelligence,C Programming,C++,JAX framework,Large Language Models - LLMs,Machine learning techniques,Python Programming,PyTorch

___
Full Time

28 Jun 2025

Summary

How you will make 10x impact:

  • Collaborate with X project teams to experiment and build novel technology for AI for mental health
  • Rapidly develop prototypes of machine learning...

Key Skills

Aartificial intelligence,AWS,Azure,Google Cloud Platform (GCP),JAX framework,Machine learning techniques,PyTorch,TensorFlow

___
Full Time

18 Jun 2025

Summary

About the role

As a Staff Infrastructure Engineer on our team you will work end to end, identifying and addressing key infra blockers on the path to scientific AGI. Strong candidates should...

Key Skills

Aartificial intelligence,Apache Beam,AWS,Design,Docker,Google Cloud Platform (GCP),Infrastructure as code,JAX framework,Kubernetes-K8s,Machine learning techniques,PyTorch,SPARK Programming,Teamwork