#

function-transforms

Here is 1 public repository matching this topic...

SandorSzabolcs / einjax

Enable efficient large-scale sparse tensor computations on multi-GPU/TPU clusters using JAX with automated tensor-relational decomposition.

open-source functional-programming python-library automatic-differentiation scientific-computing neural-networks higher-order-functions vectorization gradients einstein-notation linalg jax ml-framework numerical-computing function-transforms just-in-time-compilation

Updated Mar 20, 2026
Python

Improve this page

Add a description, image, and links to the function-transforms topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the function-transforms topic, visit your repo's landing page and select "manage topics."