Pinned Loading

1. flax flax Public

   Forked from google/flax

   Flax is a neural network library for JAX that is designed for flexibility.

   Jupyter Notebook

2. google-research google-research Public

   Forked from google-research/google-research

   Google Research

   Jupyter Notebook

3. mesh-transformer-jax mesh-transformer-jax Public

   Forked from kingoflolz/mesh-transformer-jax

   Model parallel transformers in JAX and Haiku

   Python

4. nvidia-container-toolkit nvidia-container-toolkit Public

   Forked from NVIDIA/nvidia-container-toolkit

   Build and run containers leveraging NVIDIA GPUs

   Go

5. open-gpu-kernel-modules open-gpu-kernel-modules Public

   Forked from NVIDIA/open-gpu-kernel-modules

   NVIDIA Linux open GPU kernel module source

   C