2 of 2 Google JAX Jobs in Somerset

e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

. Hands-on experience training and deploying learned policies for complex behaviors in robots or simulated characters. Fluency with modern ML toolchains (PyTorch/JAX, RL or behavior-cloning libraries, agentic frameworks, MCP, distributed training workflows). Solid software-engineering fundamentals - tested, documented, maintainable Python & C++. Demonstrated ability to More ❯