Install Bazel:
sudo apt install apt-transport-https curl gnupg curl -fsSL https://bazel.build/bazel-release.pub.gpg | gpg --dearmor > bazel.gpg sudo mv bazel.gpg /etc/apt/trusted.gpg.d/ echo "deb [arch=amd64] https://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list sudo apt update sudo apt install bazel sudo apt-get install python3-venv python3 -m venv env source env/bin/activate pip3 install -r requirements.txt python3 submission_runner.py --workload=mnist --submission_path=workloads/mnist/submission.py Note that the current MNIST example uses tf.data loaders and the Flax library for Jax models, but these may not be required dependencies for other implementations of the spec (e.g. PyTorch).