12 of 12 Remote/Hybrid Google JAX Jobs

track record of building and maintaining production‐quality systems. Expert‐level Python experience with numerical, statistical, or ML libraries (e.g. NumPy, pandas, PyTorch, JAX, TensorFlow). Deep understanding of statistical modelling, optimisation, or machine‐learning techniques. Experience building and operating backend systems and data pipelines in a cloud environment ...

quality, well-tested and reproducible technical work you have delivered yourself. Strong Python skills, including proficiency with modern deep learning frameworks (e.g., PyTorch, TensorFlow, JAX, Hugging Face), and working knowledge of SQL for querying data. Hands-on experience applying machine learning and NLP, including large language models, to real ...

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 ...

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 ...

with evaluating and improving the generalizability and robustness of ML systems. Proficiency in programming languages such as Python and ML frameworks (e.g., PyTorch, TensorFlow, JAX). Excellent communication skills to collaborate effectively with cross‐functional teams and present findings. One or more papers at top‐tier venues (such ...

YSDA, or international equivalents). Strong foundation in probability, statistics, linear algebra, and algorithms. Tech Stack & Hard Skills: Python (NumPy, Numba). PyTorch or JAX (for the ML track) OR advanced backtesting/quantitative modeling frameworks (for the Quant track). Proven track record of developing profitable ML models ...

creative teams to turn ML capabilities into compelling experiences What We’re Looking For: Strong knowledge of Python Experience writing models in PyTorch or JAX A good understanding of how large language models work Comfortable working along the whole ML pipeline: from data preparation to model design, training ...

significant experience designing and building production-grade machine learning systems at scale. You have deep experience with modern ML frameworks such as PyTorch, TensorFlow, JAX, or similar. You have experience with—or a strong interest in—multimodal machine learning across areas such as text, audio, image, or video. ...

ensuring models stay honest against field conditions and remain deployable in real-time systems. Experience with time-series or biosignals ML using PyTorch or JAX is highly valued, with hybrid work across Manchester and Los Angeles. #J-18808-Ljbffr ...

design and scale infrastructure that powers the future of AI-driven engineering? Apply now. Responsibilities Design and scale training environments using PyTorch Distributed, JAX, or NVIDIA NeMo across multi-node/multi-GPU clusters. Manage a mixed-compute strategy spanning public clouds (AWS/Azure) and sovereign industrial clouds … that validates physical consistency laws. Skills/Must have Orchestration: Expert-level Kubernetes (AKS/EKS) is essential. ML Frameworks: Strong proficiency in PyTorch, JAX, or NVIDIA NeMo. HPC/Data: Solid experience with Python, Go, and distributed data tools (Dask/Spark). Background: Experience in AI research ...

approaches, and lessons learned You May Be a Good Fit If You Have hands-on experience training large language models, or deep expertise with JAX, TPU, PyTorch, or large-scale distributed systems Genuinely enjoy both research and engineering work—you'd describe your ideal split as roughly 50/… Care about the societal impacts of AI and responsible scaling Strong Candidates May Also Have Previous experience training LLM’s or working extensively with JAX/TPU, PyTorch, or other ML frameworks at scale Contributed to open-source LLM frameworks (e.g., open_lm, llm-foundry, mesh-transformer-jax ...

methods and fidelity levels. Develop and implement solvers and numerical methods, potentially in modern differentiable or GPU-native frameworks such as JAX or NVIDIA Warp. Work with multi-fidelity data: understand tradeoff between fidelity and CFD simulators and how the two can be combined. Use commercial and open-source … ideally in turbomachinery, aerospace, defence or the chemical and process industries (CPI). Experience developing numerical solvers or PDE solvers, for example in JAX, Warp or a comparable framework. Hands‐on experience with at least one commercial CFD solver, such as Simcenter STAR-CCM+ or Ansys Fluent. Comfort working ...