A small library for creating and manipulating custom JAX Pytree classes
Home Page:https://cgarciae.github.io/treeo
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool