Open Source Alternatives to Jax

1 alternative1.4k total stars75 total forks1 language

What is Jax?

Jax is a high-performance numerical computing library for Python that provides automatic differentiation and accelerated linear algebra operations.. There is currently 1 open-source alternative to Jax, with a combined total of 1.4k GitHub stars. The most common language among these projects is Go.

Showing 1 open-source alternative

GoMLXgomlx/gomlx

An accelerated machine learning framework for Go, offering a PyTorch/Jax/TensorFlow-like experience with support for CPUs, GPUs, TPUs, and WASM.

1.4k75Go

3 days ago

Community-curated · Updated weekly · 100% open source

Found a gem we're missing?

Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.

Submit a project Star on GitHub