GitHub repository: Drenderer/klax
Primary language: Python
Created: 2025-03-05T12:19:35Z
Description: Collection of semi-common machine learning architectures and tools build on JAX + Equinox