simple-pytree
A dead simple Python package for creating custom JAX pytree objects
Installation
To install this package, run one of the following:
Usage Tracking
0.2.2
0.1.7
2 / 8 versions selected
Downloads (Last 6 months): 0
Description
A dead simple Python package for creating custom JAX pytree objects.
- Strives to be minimal, the implementation is just ~100 lines of code
- Has no dependencies other than JAX
- Its compatible with both dataclasses and regular classes
- It has no intention of supporting Neural Network use cases (e.g. partitioning)
About
Summary
A dead simple Python package for creating custom JAX pytree objects
Last Updated
Jun 20, 2024 at 19:47
License
MIT
Total Downloads
44
Supported Platforms
noarch
GitHub Repository
https://github.com/cgarciae/simple-pytree