GitHub repository: ASEM000/PyTreeClass
Primary language: Python
Description: Visualize, create, and operate on JAX pytrees in the most intuitive way possible.