26 to 50 of 102 Google JAX Jobs

C++ Experience working with Linux operating systems and the ability to learn new technologies quickly Exposure to machine learning frameworks such as PyTorch, JAX, Triton or TensorFlow, test automation, CI/CD, data analytics, hardware architectures or compilers would be useful. Experience with distributed workload management systems or machine ...

experience. Level (Scientist or Senior Scientist) will be set based on experience. Strong Python skills and experience building practical deep learning systems (e.g. PyTorch, JAX). Experience training, evaluating or applying ML models to complex real-world biological data. Understanding of how genetic variants affect protein structure and function ...

building machine-learning, generative-AI or LLM applications, gained through employment, industry or research internships, applied research or significant academic projects.* Experience with PyTorch, JAX, TensorFlow or a similar framework.* Familiarity with core agentic AI concepts such as tool calling, planning, memory and workflow orchestration.* An understanding of reinforcement ...

neural network training dynamics Strong programming skills in Python and hands‐on experience with at least one modern deep learning framework, such as PyTorch, JAX, or TensorFlow Strong communication skills with the ability to explain technical ideas clearly to both technical and non‐technical audiences Please note that ...

field. Experience with machine learning for speech recognition, machine translation, NLP or speech synthesis. Hands-on experience with deep learning toolkits such as PyTorch, JAX or TensorFlow. Preferred Qualifications Research or product experience in speech-to-speech or speech-to-text translation, especially end-to-end, streaming or simultaneous ...

Machine Learning, Computer Science, Maths, Data Science, or related field Software development skills and proficiency with a deep learning framework such as PyTorch or JAX An understanding of deep learning architectures, optimisation, evaluation and scaling, with the ability to design, run and explain ML experiments A collaborative approach ...

Machine Learning, Computer Science, Maths, Data Science, or related field Software development skills and proficiency with a deep learning framework such as PyTorch or JAX An understanding of deep learning architectures, optimisation, evaluation and scaling, with the ability to design, run and explain ML experiments A collaborative approach ...

members in diverse fields Strong programming abilities in Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL Expertise with deep learning approaches such as diffusion or flow matching Proficiency in modern software development practices: code testing, documentation ...

would be a strong fit? Engineers with solid foundations in machine learning, deep learning and Python , ideally with hands-on experience in PyTorch or JAX . You might be working in an AI research lab, an autonomous systems company, a high-performance engineering environment or an ambitious AI start ...

programming skills in Python, with experience in C++, MATLAB, or similar languages desirable. Experience with a deep learning framework such as PyTorch, TensorFlow, or JAX (desirable). Experience with engineering simulation tools relevant to CFD, hydrodynamics, or vessel performance, such as STAR-CCM+. Understanding of naval architecture, ship hydrodynamics ...

related approaches* - Comfortable writing production-quality code in Python and/or C++, familiar with a major framework such as PyTorch, TensorFlow, or JAX* - Evidence of taking a model from idea to something that worked in a real system, not just a benchmark leaderboard* - Curiosity and range matter more ...

record of shipping products and features inside a fast-moving environment Strong coding skills in Python and familiarity with ML frameworks like PyTorch or JAX Experience training, fine-tuning, and evaluating large language models Deep familiarity with large language models and web agents, how they work, and how they ...

. Experience limited to API usage or prompt engineering is not sufficient. Experience building large-scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks. Deep knowledge of large-model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking. Evidence of research ...

machine learning, speech processing, computer science, or a closely related field Strong experience training and evaluating deep learning models using frameworks such as PyTorch, JAX, or TensorFlow Several years of research or industry experience developing machine learning models (this is not a graduate or entry-level role) In addition ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

training techniques — architecture, tokenization, fine-tuning, and evaluation. Hands-on experience with Python and PyTorch; you write clean, experimental code and iterate fast. JAX is acceptable with strong Python proficiency. C++ and Rust are not required, CUDA and GPU knowledge is expected but hands-on experience is not required. ...

preferred) in Computer Science, EE, Acoustics, ML or a related technical field. + Strong proficiency in Python and deep learning frameworks (PyTorch or JAX). + Deep expertise in acoustic signal processing (spectrograms, audio transformers) and time-series analysis (continuous physiological data, sequence modelling). + Experience deploying ...

/ML methods to complex scientific datasets. Strong programming and scientific computing skills in Python (e.g. numpy, pandas, PyTorch and/or JAX). Experience analysing complex biological, imaging or spatial datasets, or strong evidence of rapidly adapting to new data domains. Excellent communication, organisational and collaborative working skills. ...

Experience with multi-omics integration, network inference, regulatory analysis, genetics/eQTL, proteomics or phenotype data Desirable Hands-on experience with scikit-learn, PyTorch, JAX, TensorFlow or equivalent ML/deep-learning frameworks Desirable Knowledge of mechanistic modelling, Digital Twins, perturbation modelling or hybrid mechanistic-AI systems Desirable Requirement ...

running scientific codes on high-performance computing facilities would be an advantage Experience in at least one deep learning framework such as PyTorch, Tensorflow, JAX is required Experience with machine learning workflows, including data preparation, training and inference is required Knowledge of data formats commonly used in meteorology, such ...

training RL agents and iterating on reward design, with strong intuitions for what works and why Proficiency in Python and ML frameworks (PyTorch or JAX) with the ability to implement and run experiments independently Track record of research output: publications, open-source contributions, or impactful technical projects Willingness ...

related Expertise in generative modelling, machine learning, music information retrieval or speech and audio processing Industry experience of coding in Python and PyTorch or JAX Excellent communication and presentation skills Preferred Qualifications Deep expertise in one of the following areas: audio generation and editing, post-training of diffusion models ...

team and culture The Frameworks team connects Graphcore hardware with the tools ML engineers and researchers rely on. We work across PyTorch, Triton, JAX and TensorFlow, focusing on usability and performance. Work happens through agile practices, close technical discussion and clear ownership. Engineers are trusted to speak up, solve ...

experience with other robotics simulators (Drake, MuJoCo, Newton, Isaac, RaiSim, Jiminy, Brax). PyTorch, our user-facing interface. Other JIT compilers such as Triton, JAX, or Numba. 3D tooling and interactive editors (Maya, Blender, Houdini). Serialization formats (USD, glTF, MJCF/URDF). C++, CUDA, or Metal. Developer ...

production, including model design, tooling, and infrastructure development. Ideal applicants have a strong ML background, publications, and experience with PyTorch/TensorFlow/JAX, as well as Python/C++. The role offers on-site work in London with visa sponsorship for international candidates. #J-18808-Ljbffr ...