google / tree-math

Mathematical operations for JAX pytrees
189Updated 6 months ago

Related projects

Alternatives and complementary repositories for tree-math