10 of 10 Google JAX Jobs

NumPy). Hands-on experience in building and maintaining automated data pipelines for large-scale data processing. Familiarity with machine learning frameworks (e.g., PyTorch, JAX, scikit-learn) as applied to data quality and augmentation tasks. Expertise in working with healthcare data, including familiarity with the OMOP Common Data Model ...

advancements into scalable, real-world solutions. Ideally, you'd have: Practical experience working with LLMs, with proficiency in frameworks like PyTorch, JAX, or TensorFlow. You should also be skilled at rapidly interpreting research literature and turning new ideas into working prototypes. A track record of published research ...

areas: computer vision, multimodal learning, 3D Graphics, or large-scale representation learning. Experience developing and training deep learning models using modern frameworks (PyTorch, TensorFlow, JAX). A strong research track record, evidenced by multiple publications and presentations in top-tier, peer-reviewed venues. Proficiency in one or more programming ...

Waymo is an autonomous driving technology company with the mission to be the most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver-The World's Most Experienced Driver -to improve access to mobility while saving ...

Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

AMLZ Recruiting is the executive search brand of AMLZ GmbH, headquartered in Wiesbaden, Germany. We specialize in placing exceptional professionals and executives with leading companies worldwide. Backed by a global network and deep industry expertise ...

AMLZ Recruiting is the executive search brand of AMLZ GmbH, headquartered in Wiesbaden, Germany. We specialize in placing exceptional professionals and executives with leading companies worldwide. Backed by a global network and deep industry expertise ...

*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning ...

knowledge with their teammates. xAI does not have recruiters. Every application is reviewed directly by a technical member of the team. Tech Stack Python JAX and XLA Rust Location The role is based in the Bay Area San Francisco and Palo Alto . Candidates are expected to be located … that can accurately understand the universe and create new knowledge. Ideal Experiences Building distributed training systems for AI models in HPC clusters. Working with JAX/XLA for multi-host training. Digging into third-party source code for debugging and customization. Interview Process After submitting your application, the team ...