einops VS extending-jax

Compare einops vs extending-jax and see what are their differences.

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others) (by arogozhnikov)

extending-jax

Extending JAX with custom C++ and CUDA code (by dfm)
Scout Monitoring - Free Django app performance insights with Scout Monitoring
Get Scout setup in minutes, and let us sweat the small stuff. A couple lines in settings.py is all you need to start monitoring your apps. Sign up for our free tier today.
www.scoutapm.com
featured
InfluxDB - Power Real-Time Data Analytics at Scale
Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
www.influxdata.com
featured
einops extending-jax
20 2
8,075 357
- -
7.5 3.5
17 days ago 8 months ago
Python Python
MIT License MIT 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.

einops

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

extending-jax

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

What are some alternatives?

When comparing einops and extending-jax you can also consider the following projects:

opt_einsum - ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

mpi4jax - Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

kymatio - Wavelet scattering transforms in Python with GPU acceleration

thinc - 🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

d2l-en - Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

equinox - Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

data-science-ipython-notebooks - Data science Python notebooks: Deep learning (TensorFlow, Theano, Caffe, Keras), scikit-learn, Kaggle, big data (Spark, Hadoop MapReduce, HDFS), matplotlib, pandas, NumPy, SciPy, Python essentials, AWS, and various command lines.

trax - Trax — Deep Learning with Clear Code and Speed

horovod - Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

elegy - A High Level API for Deep Learning in JAX

3d-ken-burns - an implementation of 3D Ken Burns Effect from a Single Image using PyTorch

diffrax - Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Scout Monitoring - Free Django app performance insights with Scout Monitoring
Get Scout setup in minutes, and let us sweat the small stuff. A couple lines in settings.py is all you need to start monitoring your apps. Sign up for our free tier today.
www.scoutapm.com
featured
InfluxDB - Power Real-Time Data Analytics at Scale
Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
www.influxdata.com
featured