8 projects
jax-cuda13-plugin
JAX Plugin for NVIDIA GPUs
jax-cuda13-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
maxtext
MaxText is a simple, performant and scalable Jax LLM!
xla-cuda13-pjrt
XLA PJRT Plugin
xla-cuda12-pjrt
XLA PJRT Plugin
seed-env
A CLI tool for environment setup that combines a seeder project with your specific requirements.
mosaic-gpu-cuda12
An empty wheel for MOSAIC GPU CUDA12 reservation
mosaic-gpu-cuda13
An empty wheel for MOSAIC GPU CUDA13 reservation
