/gradient-base-container

Base Gradient Container

Primary LanguageDockerfile

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.