20 of 20 Google JAX Jobs

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

Implement best practices for data governance, security, and MLOps on GCP Proficiency with Python and common AI/ML frameworks such as TensorFlow, PyTorch, JAX, scikit learn, and Hugging Face libraries Knowledge of MLOps and LLMOps practices-including CI/CD for models, model registry/versioning, feature stores ...

Implement best practices for data governance, security, and MLOps on GCP. Proficiency with Python and common AI/ML frameworks such as TensorFlow, PyTorch, JAX, scikit learn, and Hugging Face libraries. Knowledge of MLOps and LLMOps practicesincluding CI/CD for models, model registry/versioning, feature stores, orchestration ...

ICLR. Hands-on experience applying deep learning models to complex, ideally temporal or financial, datasets. Expertise in Python and modern ML frameworks (TensorFlow, PyTorch, JAX, etc.). Prior experience in a commercial or industry setting post-PhD, ideally in a similarly data-driven domain. Why Apply Compensation and total ...

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

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

workflows Backtesting frameworks Model monitoring and evaluation What We're Looking For Must Have Strong Python skills (NumPy, pandas, scikit-learn, PyTorch or JAX) Solid experience in time-series modelling Strong problem-solving and analytical thinking Good to Have Forecasting methods (ARIMA, Prophet, etc.) Causal inference/modelling Experience ...

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

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

outside IR35, for a major pharmaceutical company. The key skills required for this MLOps Engineer position are: MLOps Image data Terraform 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. ...

outside IR35, for a major pharmaceutical company. The key skills required for this MLOps Engineer position are: MLOps Image data Terraform 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 click apply ...

researchers and contribute to a culture of technical excellence. Must-Have Skills: Machine Learning, Deep Learning, NLP, Computer Vision, Generative AI, Python, TensorFlow, PyTorch, JAX Other Skills: Machine Learning, Deep Learning, NLP, Computer Vision, Generative AI, Python, TensorFlow, PyTorch, JAX Apply Here Click here to apply ...

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