cgarciae / treex

A Pytree Module system for Deep Learning in JAX

Home Page:https://cgarciae.github.io/treex/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cgarciae/treex Stargazers