jax VS functorch

Compare jax vs functorch and see what are their differences.

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more (by google)

functorch

functorch is JAX-like composable function transforms for PyTorch. (by pytorch)
Our great sponsors
  • WorkOS - The modern identity platform for B2B SaaS
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • SaaSHub - Software Alternatives and Reviews
jax functorch
82 11
27,509 1,366
3.4% 1.0%
10.0 0.6
about 15 hours ago 5 days ago
Python Jupyter Notebook
Apache License 2.0 BSD 3-clause "New" or "Revised" License
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.

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.

functorch

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

What are some alternatives?

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

Numba - NumPy aware dynamic Python compiler using LLVM

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! ❤️

brax - Massively parallel rigidbody physics simulation on accelerator hardware.

nn - 🧑‍🏫 60 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

stan - Stan development repository. The master branch contains the current release. The develop branch contains the latest stable development. See the Developer Process Wiki for details.