This repo provides testing capability for AXLearn.
The tests are stored in .github/workflows and can be customized easily.
The Docker images used to build the testing environment are also available in the cpu, gpu, tpu, and training folders.