cgarciae / treex

A Pytree Module system for Deep Learning in JAX
214Updated last year

Related projects

Alternatives and complementary repositories for treex