piwheels

statedict2pytree

Converts torch models into PyTrees for Equinox

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install statedict2pytree

Dependencies

• pydantic
• jaxtyping
• jaxlib
• beartype
• tqdm
• equinox
• jax

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Files
1.0.1	2025-08-06
statedict2pytree-1.0.1-py3-none-any.whl (5 KB)
1.0.0	2025-05-31
statedict2pytree-1.0.0-py3-none-any.whl (5 KB)
0.6.0	2024-07-07
statedict2pytree-0.6.0-py3-none-any.whl (259 KB)
0.5.4	2024-06-30
statedict2pytree-0.5.4-py3-none-any.whl (259 KB)
0.5.3	2024-06-22
statedict2pytree-0.5.3-py3-none-any.whl (259 KB)
0.5.2	2024-06-16
statedict2pytree-0.5.2-py3-none-any.whl (254 KB)
0.5.0	2024-05-18
statedict2pytree-0.5.0-py3-none-any.whl (15 KB)
0.4.0	2024-05-17
statedict2pytree-0.4.0-py3-none-any.whl (15 KB)
0.3.0	2024-05-16
statedict2pytree-0.3.0-py3-none-any.whl (15 KB)
0.2.0	2024-05-12
statedict2pytree-0.2.0-py3-none-any.whl (15 KB)
0.1.2	2024-05-12
statedict2pytree-0.1.2-py3-none-any.whl (14 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue