GitHub repository: pwolle/flarejax
Primary language: Python
Description: Pytree modules classes with easy manipulation and serialization