15 of 15 Google JAX Jobs

learning, and/or large language models (LLMs) Experience in chip design and applied AI research Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience influencing organizational policy and securing major research funding ...

learning, and/or large language models (LLMs) Experience in chip design and applied AI research Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience influencing organizational policy and securing major research funding ...

ICLR. Hands-on experience applying deep learning models to complex, ideally temporal or financial, datasets. Expertise in Python and modern ML frameworks (TensorFlow, PyTorch, JAX, etc.). Prior experience in a commercial or industry setting post-PhD, ideally in a similarly data-driven domain. Why Apply Compensation and total ...

awareness of latency and cost tradeoffs MLOps competency — experiment tracking, model versioning, evaluation pipelines, ML CI Experience with Python and modern ML frameworks (PyTorch, JAX) Excellent verbal and written English communication skills Excellent analytical and problem-solving skills A desire to innovate and push current practice Desirable: Experience with ...

reinforcement learning Experience with agent frameworks, tool-use planning and workflow orchestration Systems-level fluency across data pipelines, training infrastructure and inference (PyTorch, JAX, TensorFlow) Strong MLOps discipline: CI/CD for models, offline/online evaluation, safety and guardrails The ability to influence at exec level and mentor ...

particularly in the field of deep learning very strong programming skills in Python experience with at least one deep learning framework (e.g., PyTorch or JAX) strong background in computer vision and probabilistic modeling knowledge of representation learning, self supervised learning, or transfer learning interest in digital signal processing, physics ...

knowledge of ML evaluation, validation strategy, and failure modes including error analysis and iteration planning ● Experience with model development using PyTorch/TensorFlow/JAX ● Ability to communicate clearly across technical + scientific stakeholders ● Strong experience leading or coordinating cross-functional teams, and thinking strategically across product and science ...

work in bwHPC-S5. Ideally, your profile is supplemented by: Experience with AI frameworks, workloads (e.g. Natural Language Processing), and libraries (e.g., PyTorch or JAX). Experience with parallel programming paradigms. Some background in the field of (green) transport (e.g., fleet optimisation, energy-efficient logistics, smart city planning). ...

3. A working knowledge and practical experience with Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL. Key Words: Machine Learning Engineer/Data Processing/Model Building/NMR/Deep Learning/Computational Biophysics/ ...

deep learning and modern foundation model architectures Experience with probabilistic modelling techniques such as Bayesian methods, Gaussian Processes, or variational inference Expertise in PyTorch, JAX, or similar frameworks Experience training large models on distributed infrastructure Strong software engineering and systems design capabilities Nice to Have Background in computational biology ...

generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics ...

excited by applying those skills to one of the most interesting data domains in AI. Areas of interest include: PyTorch/JAX internals CUDA, XLA or Triton optimisation Distributed training systems GPU cluster performance tuning Large-scale experiment management Training reliability and reproducibility Kubernetes and orchestration tooling Infrastructure supporting ...

Silicon Valley office | Reports to Founders Introduction: Join us at Fastino as we build the next generation of LLMs. Our team, boasting alumni from Google Research, Apple, Stanford, and Cambridge is on a mission to develop specialized, efficient AI. Fastino's GLiNER family of open source models has been … comparable generative AI architectures Optional - Comprehensive technical proficiency and practical experience with leading deep learning frameworks, including advanced competency in one of PyTorch, JAX, TensorFlow, or equivalent platforms for model development and optimization ...

World Wide Technology UK is looking for a hands-on Cloud Engineer with strong expertise in HPC and AI/ML performance workloads on Google Cloud Platform (GCP). The role focuses on benchmarking, optimizing, and validating performance across advanced accelerator platforms including NVIDIA GPUs, AMD GPUs, and Google … inference, scientific workloads) Provision and optimize GPU/TPU-based infrastructure on GCP (A3/A4, TPU pods) Analyze performance across frameworks (PyTorch, TensorFlow, JAX, CUDA, ROCm) Identify system bottlenecks (compute, memory, network, I/O) Build automation tools for benchmarking and reporting Collaborate with teams to align workloads ...

frontier AI company backed by NVIDIA, building large-scale open-weight foundation models alongside researchers and engineers from DeepMind, OpenAI, Meta, Anthropic, and Google Brain. ⚡ What You’ll Do Optimise GPU performance and training efficiency across 1,000+ GPU clusters Improve utilisation, throughput, and reliability across distributed training infrastructure … accelerate large-scale model training 🔧 What They’re Looking For Deep GPU infrastructure/distributed systems experience Strong knowledge of CUDA, NCCL, PyTorch, DeepSpeed, JAX, Megatron-LM, vLLM, etc. Experience operating large-scale GPU clusters (1,000+ GPUs) Kubernetes, Slurm, or similar orchestration expertise BONUS: Experience working on NVIDIA ...