Gradient base Docker image
This repo contains the Dockerfiles used to build the Gradient base Machine Learning Docker image which includes PyTorch, TensorFlow, JAX, Transformers, and other popular ML/AI libraries. The Docker image is hosted on Dockerhub.