Search Results - RepositoryStats

9

178

apache-2.0

5

OpTree: Optimized PyTree Utilities

tree optree pytree tree-map tree-flatten tree-operation tree-structure

Created 2022-09-23

350 commits to main branch, last one 6 days ago

4

56

mit

3

A small library for creating and manipulating custom JAX Pytree classes

jax pytree

This repository has been archived (exclude archived)

Created 2021-09-19

144 commits to master branch, last one 2 years ago

2

45

apache-2.0

1

Visualize, create, and operate on pytrees in the most intuitive way possible.

jax data pytree pytorch pipelines tensorflow dataclasses deep-learning machine-learning

Created 2022-07-11

1,106 commits to main branch, last one 3 months ago