GitHub repository: Astera-org/jax_loop_utils
Primary language: Python
Created: 2024-11-13T23:08:26Z
Description: Utilities for writing ML training loops in JAX