google / tree-math

Mathematical operations for JAX pytrees
199Updated 4 months ago

Alternatives and similar repositories for tree-math:

Users that are interested in tree-math are comparing it to the libraries listed below