Search Results - RepositoryStats

6

136

apache-2.0

5

OpTree: Optimized PyTree Utilities

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

Created 2022-09-23

272 commits to main branch, last one a day ago

4

59

mit

4

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 about a year ago

1

41

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,103 commits to main branch, last one about a year ago