nx VS jax

Compare nx vs jax and see what are their differences.

nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir (by elixir-nx)

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (by google)
Our great sponsors
  • WorkOS - The modern identity platform for B2B SaaS
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • SaaSHub - Software Alternatives and Reviews
nx jax
36 82
2,442 27,509
1.9% 3.8%
9.4 10.0
10 days ago about 20 hours ago
Elixir Python
- Apache License 2.0
The number of mentions indicates the total number of mentions that we've tracked plus the number of user suggested alternatives.
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.

nx

Posts with mentions or reviews of nx. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2023-08-25.

jax

Posts with mentions or reviews of jax. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2024-03-22.

What are some alternatives?

When comparing nx and jax you can also consider the following projects:

Numba - NumPy aware dynamic Python compiler using LLVM

functorch - functorch is JAX-like composable function transforms for PyTorch.

julia - The Julia Programming Language

Pytorch - Tensors and Dynamic neural networks in Python with strong GPU acceleration

Cython - The most widely used Python to C compiler

jax-windows-builder - A community supported Windows build for jax.

mesh-transformer-jax - Model parallel transformers in JAX and Haiku

dex-lang - Research language for array processing in the Haskell/ML family

tensorflow - An Open Source Machine Learning Framework for Everyone

tinygrad - You like pytorch? You like micrograd? You love tinygrad! ❤️

Elixir - Elixir is a dynamic, functional language for building scalable and maintainable applications

brax - Massively parallel rigidbody physics simulation on accelerator hardware.