2 of 2 Permanent Google JAX Jobs in the South West

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

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