8 of 8 Google JAX Jobs

Explainable AI (XAI) SHAP/LIME interpretation, fairness flagging Data Storytelling Generating readable summaries from charts or model outputs Tools TensorFlow/PyTorch/JAX/HuggingFace We offer a pay range of $25-to- $100 per hour, with the exact rate determined after evaluating your experience, expertise ...

This temporary role requires an ML Quant Engineer with expertise within an Investment Bank. The position is based in London and involves developing and implementing machine learning models to support financial decision-making. Client Details ...

candidates who: Have deep familiarity with Python data ecosystem Understanding of Jupyter notebooks Exposure to machine learning libraries like PyTorch, XGBoost and JAX Understanding of crypto or traditional financial markets Strong API design and documentation skills What do you get in return? Up to 250k base (depending on experience ...

candidates who: Have deep familiarity with Python data ecosystem Understanding of Jupyter notebooks Exposure to machine learning libraries like PyTorch, XGBoost and JAX Understanding of crypto or traditional financial markets Strong API design and documentation skills What do you get in return? Up to £250k base (depending on experience ...

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

maintaining, or launching software products, and 1 year of experience with software design and architecture. Experience using machine learning frameworks (e.g., Tensorflow, PyTorch, or Jax). Experience with machine learning model pretraining, fine-tuning, evaluation, and methodologies. Preferred qualifications: Master's degree or PhD in Computer Science … related technical field. Experience publishing in high-quality machine learning journals. Experience with Python and C++. About The Job Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information ...

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

runs. You Might Be a Good Fit If You Have Strong engineering experience in large-scale distributed training or HPC systems. Deep familiarity with JAX internals, distributed training libraries, or custom kernels/fused ops. Experience with multi-node cluster orchestration (Slurm, Ray, Kubernetes, or similar). Comfort debugging … closely with infra, research, and deployment teams. Nice to Have Experience with training LLMs or other large transformer architectures. Contributions to ML frameworks (PyTorch, JAX, DeepSpeed, Megatron, xFormers, etc.). Familiarity with evaluation and serving frameworks (vLLM, TensorRT-LLM, custom KV caches). Experience with data pipeline optimization, sharded ...