GitHub repository: kscalelabs/xax
Primary language: Python
Description: JAX experimentation library.