1 of 1 Google JAX Jobs in Camden

Staff Machine Learning, AI Infrastructure, Autonomy

Location
Camden Town, England, United Kingdom
tools, frameworks, and architectural patterns Qualifications PhD in CS/CE/EE, or equivalent, in industry experience Deep knowledge of PyTorch, TrensorFlow or Jax Knowledge of model training framework In-depth knowledge of transformer architecture and ways to accelerate the training and inference of transformer models Experience ...