tree-math

Mathematical operations for JAX pytrees

181 stars

11 watching

7 forks

Language: Python

last commit: 5 months ago

Linked from 1 awesome list

jaxtree-math

Backlinks from these awesome lists:

n2cholas/awesome-jax