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
mediapipev1.0.1↑

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

2.1M/moApache 2.0★ 36.7K
87
#2
lerobotv0.6.1↑

🤗 LeRobot: State-of-the-art Machine Learning for Real-World Robotics in Pytorch

230.0K/moApache-2.0★ 26.8K
77
#3
torchaxv0.0.13↑

torchax is a library for running Jax and PyTorch together

146.6K/moApache-2.0★ 238
75
#4
grafeov0.5.42↑

A high-performance, embeddable graph database with Python bindings

28.3K/moApache-2.0★ 752
67
#5
churnkitv1.3.4a5↑

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

20.2K/moApache-2.0
59
#6
tpu-inferencev0.27.0↑
58.8K/moApache Software License
56
#7
contextpressv0.6.6↑

Deterministic context compression for LLM chat, RAG, and agent pipelines

5.5K/moApache-2.0
53
#8
jaxnnv0.2.0↑

Foundation models in JAX/Flax

167/moApache-2.0
44
#9
tensorflow-cpuv2.21.0↑

TensorFlow is an open source machine learning framework for everyone.

1.2M/moApache 2.0★ 197.3K
92
#10
korniav0.8.3↑

Open Source Differentiable Computer Vision Library for PyTorch

2.3M/moApache-2.0★ 11.3K
90
#11
tensorflow-intelv2.18.0↑

TensorFlow is an open source machine learning framework for everyone.

139.2K/moApache 2.0★ 197.2K
90
#12
unstructuredv0.27.1↑

A library that prepares raw documents for downstream ML tasks.

4.1M/moApache-2.0★ 15.3K
90
← Browse all categories