gorodnitskiy / jax-cuda-dockerView on GitHub
This repo contains Dockerfile that can be used to easily run JAX with CUDA support in Docker without JAX and CUDA/cuDNN versions mismatching.
15Feb 1, 2023Updated 3 years ago

Alternatives and similar repositories for jax-cuda-docker

Users that are interested in jax-cuda-docker are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?