2 of 2 Google JAX Jobs

Dialogflow, Gemini) ML Development: Experience with Machine Learning model development and deployment. AIML Frameworks: Experience with frameworks for deep learning (e.g., PyTorch, TensorFlow, Jax, Ray, etc.), AI accelerators (e.g. TPUs, GPUs), model architectures (e.g., encoders, decoders, transformers), and using machine learning APIs. Must Have- Troubleshooting Code Experience: 2 years ...

Python development experience for high-performance, data-intensive applications Experience with Machine Learning/Deep Learning Strong hands-on experience with PyTorch, TensorFlow, or JAX Experience with Pandas and statistical model evaluation Production ML model deployment, monitoring, and maintenance Experience with LLMs, Recommender Systems, or Operations Research Experience working ...