23 of 23 Google JAX Jobs

production-ready solutions. Strong proficiency in Python and core AI/data science libraries, such as: PyTorch, TensorFlow, ScikitLearn, Keras, NumPy, Pandas PyTorch Geometric, JAX, XGBoost, OpenCV Transformers, LangChain, Pinecone Hands-on experience with cloud computing systems for training, optimising, and deploying models. Strong communication skills and the ability ...

with a demonstrable track record of models made it from the lab into production environments . Deep Technical Mastery: Expert-level proficiency in PyTorch, JAX, or TensorFlow , with a focus on building custom layers, loss functions, and optimization loops. Published Excellence: A strong record of high-quality publications ...

machine learning theory, including optimization, generalisation, and various model architectures. Programming : Strong python skills and experience with deep learning libraries (TensorFlow/PyTorch/JAX). Communication : Ability to clearly explain complex ML concepts and research findings to both technical and non-technical audiences. Education : Master's Degree ...

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

Developer will work with a variety of technologies including the Python scientific stack (NumPy, Pandas, SciPy, etc.), frameworks such as OpenCV, PyTorch, TensorFlow and JAX, data pipeline tools such as Airflow or Prefect and data processing tools such as Spark or Databricks. The Company The Python Developer will ...

Developer will work with a variety of technologies including the Python scientific stack (NumPy, Pandas, SciPy, etc.), frameworks such as OpenCV, PyTorch, TensorFlow and JAX, data pipeline tools such as Airflow or Prefect and data processing tools such as Spark or Databricks. The Company The Python Developer will ...

Developer will work with a variety of technologies including the Python scientific stack (NumPy, Pandas, SciPy, etc.), frameworks such as OpenCV, PyTorch, TensorFlow and JAX, data pipeline tools such as Airflow or Prefect and data processing tools such as Spark or Databricks. The Company The Python Developer will ...

Developer will work with a variety of technologies including the Python scientific stack (NumPy, Pandas, SciPy, etc.), frameworks such as OpenCV, PyTorch, TensorFlow and JAX, data pipeline tools such as Airflow or Prefect and data processing tools such as Spark or Databricks. The Company The Python Developer will ...

Developer will work with a variety of technologies including the Python scientific stack (NumPy, Pandas, SciPy, etc.), frameworks such as OpenCV, PyTorch, TensorFlow and JAX, data pipeline tools such as Airflow or Prefect and data processing tools such as Spark or Databricks. The Company The Python Developer will ...

Developer will work with a variety of technologies including the Python scientific stack (NumPy, Pandas, SciPy, etc.), frameworks such as OpenCV, PyTorch, TensorFlow and JAX, data pipeline tools such as Airflow or Prefect and data processing tools such as Spark or Databricks. The Company The Python Developer will ...

Mathematics, Physics, Computer Science, or a related field Strong hands-on experience building LLM-based systems with modern ML tooling (PyTorch, Ray, JAX) Experience designing agentic systems (planning, memory, tool use, multi-step reasoning) Strong understanding of transformers and foundation model behavior Comfortable owning whatever needs doing, from research ...

Mathematics, Physics, Computer Science, or a related field Strong hands-on experience building, training, and scaling foundation models with modern ML tooling (PyTorch, Ray, JAX) Strong background in graph neural networks and transformer architectures Demonstrated ability to turn research ideas into robust, scalable implementations Comfortable owning whatever needs doing ...

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

production systems Hands-on post-training experience (RLHF, DPO, or similar) and deployed LLMs in production Strong Python and modern ML tooling (PyTorch/JAX, training infrastructure, evaluation pipelines) Track record shipping research from prototype to production impact Comfortable owning work end-to-end: architecture to data pipelines ...

skills required for this ML Engineer position are: Building ML systems Computer Vision Building processing pipelines for image data AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract ML Engineer role, outside IR35, please do apply. ...

skills required for this ML Engineer position are: Building ML systems Computer Vision Building processing pipelines for image data AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract ML Engineer role, outside IR35, ple click apply for full job details ...

skills required for this ML Engineer position are: Building ML systems Computer Vision Building processing pipelines for image data AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract ML Engineer role, outside IR35, please do apply. ...

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

basis, outside IR35, for a major pharmaceutical company. The key skills required for this MLOps Engineer position are: MLOps Image data AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract MLOps Engineer role, outside IR35, please do apply. ...

engineering, fine-tuning, RAG (Retrieval Augmented Generation) Multimodal AI (text, image, audio, video) Agentic frameworks and orchestration AI/ML Engineering Python, PyTorch, TensorFlow, JAX NLP, deep learning, model optimisation Evaluation frameworks and guardrails Data & Platforms Vector databases (Pinecone, Weaviate, FAISS, Milvus) Knowledge graphs and semantic search Data engineering …/CD for ML, model monitoring, versioning Kubernetes, Docker, serverless architectures MLflow, Kubeflow, SageMaker or similar Cloud & Architecture Azure AI, AWS AI/Bedrock, Google Vertex AI API-led architectures and microservices Security, governance, and responsible AI Frameworks & Tooling LangChain, LlamaIndex, Semantic Kernel or equivalent Integration with enterprise systems ...

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

intuitive OS and interfaces for orchestrating factories, monitoring machines, controlling production, etc. all with AI. For context, we are a small dynamic team from Google X, Microsoft, Cambridge, MIT, Imperial, BBC R&D to name a few. We work a lot and are all close friends – come join … machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down ...

intuitive OS and interfaces for orchestrating factories, monitoring machines, controlling production, etc. all with AI. For context, we are a small dynamic team from Google X, Microsoft, Cambridge, MIT, Imperial, BBC R&D to name a few. We work a lot and are all close friends – come join … machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down ...