1 to 25 of 69 Google JAX Jobs

models, deep learning, and scalable AI pipelines, with demonstrated involvement in LLM development, fine-tuning, or deployment. Technical Skills: Proficiency in Python and frameworks such as PyTorch , TensorFlow , or JAX . Strong familiarity with distributed computing and data engineering tools (e.g., SLURM , Apache Spark , Airflow ). Hands-on experience with LLM training, fine-tuning, and deployment (e.g., Hugging Face More ❯

models, deep learning, and scalable AI pipelines, with demonstrated involvement in LLM development, fine-tuning, or deployment. Technical Skills: Proficiency in Python and frameworks such as PyTorch , TensorFlow , or JAX . Strong familiarity with distributed computing and data engineering tools (e.g., SLURM , Apache Spark , Airflow ). Hands-on experience with LLM training, fine-tuning, and deployment (e.g., Hugging Face More ❯

models, deep learning, and scalable AI pipelines, with demonstrated involvement in LLM development, fine-tuning, or deployment. Technical Skills: Proficiency in Python and frameworks such as PyTorch , TensorFlow , or JAX . Strong familiarity with distributed computing and data engineering tools (e.g., SLURM , Apache Spark , Airflow ). Hands-on experience with LLM training, fine-tuning, and deployment (e.g., Hugging Face More ❯

Good foundational understanding and subsequent practical application of ML techniques, preferably NLPs, LLMs, Bayesian Optimisation and MCMCs Strong proficiency in Python and ML frameworks and tools (e.g. PyTorch, TensorFlow, JAX, Hugging Face, vLLM, MCP, prompt engineering) Excellent communication skills and ability to explain ML concepts to non-technical stakeholders Ability to work effectively in multi-disciplinary teams, collaborating across More ❯

Good foundational understanding and subsequent practical application of ML techniques, preferably NLPs, LLMs, Bayesian Optimisation and MCMCs Strong proficiency in Python and ML frameworks and tools (e.g. PyTorch, TensorFlow, JAX, Hugging Face, vLLM, MCP, prompt engineering) Excellent communication skills and ability to explain ML concepts to non-technical stakeholders Ability to work effectively in multi-disciplinary teams, collaborating across More ❯

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

in a leadership role. Proven experience delivering production-grade systems in Search, Recommender Systems , or Personalisation . Strong understanding of ML fundamentals and deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Hands-on experience with GenAI , including working with LLMs (e.g., OpenAI, Anthropic, HuggingFace models). Deep knowledge of MLOps, experimentation, and model evaluation techniques. Experience scaling ML platforms More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

algorithms, optimisation, and model evaluation 📈 Strong mathematical reasoning and analytical problem-solving skills 🗣 Excellent communication skills and a genuine passion for learning and innovation Nice to Have: 💾 Experience with JAX , Hydra , or Weights & Biases 📊 Exposure to GANs , diffusion models , or autoregressive architectures 📚 Knowledge of data privacy , machine unlearning , or secure ML methods 💻 Internship or academic project experience in AI More ❯

across technical teams and stakeholders Effective problem-solver who takes initiative in complex production settings Experience with scientific computing, deep learning, big data, or health IT ontologies (e.g., PyTorch, JAX, Spark, HL7, FHIR) (desirable) Familiarity with cloud infrastructure (Azure/AWS), infrastructure as code, Kubernetes, Linux, Docker, data pipelines, and MLOps tools (desirable) Passion for biomedical topics and startup More ❯

across technical teams and stakeholders Effective problem-solver who takes initiative in complex production settings Experience with scientific computing, deep learning, big data, or health IT ontologies (e.g., PyTorch, JAX, Spark, HL7, FHIR) (desirable) Familiarity with cloud infrastructure (Azure/AWS), infrastructure as code, Kubernetes, Linux, Docker, data pipelines, and MLOps tools (desirable) Passion for biomedical topics and startup More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this More ❯

Reinforcement Learning (policy optimisation, reward modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs/TPUs. If this More ❯