ASEM000/pytreeclass
Visualize, create, and operate on pytrees in the most intuitive way possible.
This tool helps machine learning engineers and researchers manage complex JAX data structures, called 'PyTrees', more easily. It simplifies creating, visualizing, and manipulating these data structures, making it intuitive to inspect and modify them. You input JAX PyTrees, and it outputs organized visual representations (like diagrams and summaries) and modified PyTree structures.
No commits in the last 6 months. Available on PyPI.
Use this if you are a machine learning engineer or researcher working with JAX and need a more intuitive way to define, inspect, and update your model parameters or complex nested data structures.
Not ideal if you are not using JAX for your machine learning or scientific computing tasks.
Stars
46
Forks
2
Language
Python
License
Apache-2.0
Category
Last pushed
Jan 11, 2025
Commits (30d)
0
Dependencies
1
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ASEM000/pytreeclass"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
LAMDA-NJU/Deep-Forest
An Efficient, Scalable and Optimized Python Framework for Deep Forest (2021.2.1)
grf-labs/grf
Generalized Random Forests
cerlymarco/linear-tree
A python library to build Model Trees with Linear Models at the leaves.
benedekrozemberczki/awesome-decision-tree-papers
A collection of research papers on decision, classification and regression trees with implementations.
ysraell/random-forest-mc
Random Forest with Dynamic Tree Selection Monte Carlo Based (RF-TSMC).