8 of 8 Remote/Hybrid Permanent 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 ...

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

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

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