search

cgarciae / treeo

A small library for creating and manipulating custom JAX Pytree classes
https://cgarciae.github.io/treeo
MIT License
58 stars 4 forks source link

update poetry #25

Closed cgarciae closed 1 year ago