4 of 4 Google JAX Jobs

forests, and ensemble methods Experience with generative models (GANs, VAEs, Diffusion models) Programming & Tools : Expert-level Python programming Proficiency with ML frameworks (PyTorch, TensorFlow, JAX) Experience with scikit-learn, XGBoost, LightGBM Strong software engineering practices and clean code principles Data & Computing : Experience with big data technologies (Spark, Dask ...

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 Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible ...

/verbal communication; stakeholder-oriented thinking Team-working, independent planning/execution, problem-solving and data analysis/reporting skills PyTorch/TensorFlow/Jax/Transformers; problem scoping with customers; managing research data/experiments; AI tooling awareness Commitment to EDI and organisational values; eligible for SC clearance ...

*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning ...