PyPI StatsPyPI Stats
DiscoverCompareTrendingDashboard
PyPI Stats — Download analytics for Python packages
APIGitHubPrivacyTerms

Sign in to subscribe to our weekly trending newsletter.

Download data sourced from BigQuery (Google). Counts may include CI/CD and mirror traffic.
Inspired by and built upon the work of pypistats.org
Discover/flax/Alternatives

Best Alternatives to flax

Flax: A neural network library for JAX designed for flexibility

12 similar packages · ranked by shared classifiers & health score · View flax →

#1
mediapipev0.10.32↑

MediaPipe is the simplest way for researchers and developers to build world-class ML solutions and applications for mobile, edge, cloud and the web.

2.2M/moApache 2.0★ 35.1K
0
#2
tpu-inferencev0.13.3↑
178.5K/mo
0
#3
churnkitv1.0.1a3↑

Structured ML framework for customer churn prediction -- from exploration notebooks to production pipelines, locally or on Databricks.

73.5K/moApache-2.0
0
#4
torchaxv0.0.11↑

torchax is a library for running Jax and PyTorch together

221.9K/moApache-2.0★ 218
0
#5
nltkv3.9.3↑

Natural Language Toolkit

43.6M/moApache License, Version 2.0★ 14.6K
96
#6
korniav0.8.2↑

Open Source Differentiable Computer Vision Library for PyTorch

2.5M/moApache-2.0★ 11.2K
93
#7
assemblyaiv0.54.1↑

AssemblyAI Python SDK

1.3M/moMIT License★ 205
87
#8
optimum-onnxv0.1.0↑

Optimum ONNX is an interface between the Hugging Face libraries and ONNX / ONNX Runtime

331.1K/moApache-2.0★ 146
83
#9
intervaltreev3.2.1↑

Editable interval tree data structure for Python 2 and 3

4.1M/mo★ 687
81
#10
neptune-apiv0.26.0↑

A client library for accessing Neptune API

1.8M/moApache-2.0★ 3
81
#11
docarrayv0.41.0↑

The data structure for multimodal data

101.3K/moApache 2.0★ 3.1K
81
#12
tensorflow-iov0.37.1↑

TensorFlow IO

280.9K/mo★ 735
80
← Browse all categories