GitHub repository: trymirai/lalamo
Primary language: Python
Description: JAX infrastructure for model optimisation