google / tree-math

Mathematical operations for JAX pytrees
188Updated 5 months ago

Related projects

Alternatives and complementary repositories for tree-math