jax

Here are 4 public repositories matching this topic...

A step-by-step guide to setting up Nvidia GPUs with CUDA support running on Docker (and Compose) containers on NixOS host

python deep-learning docker-compose nixos tensorflow pytorch nvidia triton nvidia-smi nvidia-docker nvidia-cuda jax

This repo contains Dockerfile that can be used to easily run JAX with CUDA support in Docker without JAX and CUDA/cuDNN versions mismatching.

docker cuda jax

docker pytorch uv jax devcontainers

TPU Container Integration for Jupyter Notebook & TensorFlow

python docker jupyter notebook reproducible-research tensorflow container jupyter-notebook oci tpu jax podman

Add a description, image, and links to the jax topic page so that developers can more easily learn about it.

To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."