GitHub repository: Daniel-Dodd/mytree
Primary language: Python
Description: "Module pytrees" that cleanly handle parameter trainability and transformations for JAX models.