GitHub repository: natbutter/jax-contained
Description: Jax, torch, tensorflow on Ubuntu 16.04 and Cuda 10.2