issues
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
Compact
#1
Closed
cgarciae
closed
3 years ago
cgarciae
commented
3 years ago
Changes
Removes
opaque_is_equal
, same functionality available through
opaque
.
Adds
compact
decorator that enable the definition of
Tree
subnodes at runtime.
Adds the
Compact
mixin that adds the
first_run
property and the
get_field
method.
Changes
opaque_is_equal
, same functionality available throughopaque
.compact
decorator that enable the definition ofTree
subnodes at runtime.Compact
mixin that adds thefirst_run
property and theget_field
method.