Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)
IMPORTANT: This repository is being deprecated. Please migrate or onboard your ML tests to our new repository here.
A set of tools and examples to run machine learning tests on ML hardware
accelerators (TPUs or GPUs) using Google Cloud Platform.
This is not an officially supported Google product.
In this mode, your tests and/or models run on an automated schedule in GKE.
Results are collected by the “Metrics Handler” and written to BigQuery.
deployments
directory to set up a Kubernetes Cluster.images
directory to set up the Docker image that your tests will run.templates
directory for a JSonnet template library to generate test config files.Are you interested in using ML Testing Accelerators? E-mail users@googlegroups.com"">ml-testing-accelerators-users@googlegroups.com and tell us about your use-case. We’re happy to help you get started.