GitHub repository: karlotness/powerpax
Primary language: Python
Description: Small collection of useful utilities for JAX