Simple Pytree | Anaconda.org

ryanvolz / packages / simple-pytree 0.2.2

0

A dead simple Python package for creating custom JAX pytree objects

License: MIT
Home: https://github.com/cgarciae/simple-pytree
Development: https://github.com/cgarciae/simple-pytree
33 total downloads
Last upload: 1 year and 6 days ago

Installers

noarch v0.2.2

conda install

To install this package run one of the following:
conda install ryanvolz::simple-pytree

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)

© 2025 Anaconda, Inc. All Rights Reserved. (v4.2.0) Legal | Privacy Policy