Search Results - RepositoryStats

7

159

apache-2.0

5

OpTree: Optimized PyTree Utilities

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

Created 2022-09-23

297 commits to main branch, last one 6 days ago

4

57

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

2

42

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,104 commits to main branch, last one 2 months ago