NVIDIA/JAX-Toolbox

**Brief Summary (2‑3 sentences)** NVIDIA JAX‑Toolbox is an open‑source Python library that provides utilities and reference implementations for working with JAX