statedict2pytree

Converts torch models into PyTrees for Equinox


Install
pip install statedict2pytree==0.4.0