Open Source Alternatives
Open source alternatives to JAX
Open source alternatives to JAX, ranked by GitHub stars and freshness.
6 open-source alternatives in the index, ranked by GitHub stars and freshness.
TensorFlow
Community
An Open Source Machine Learning Framework for Everyone
Best for: Teams building production ML systems that need cross-platform deployment and performance optimization
PaddlePaddle
Community
PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）
Best for: Teams building large-scale production ML systems who need distributed training and cross-platform deployment out of the box
Apache MXNet
Community
Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
Best for: Teams building distributed training pipelines or mobile ML applications who need multi-language flexibility
Oneflow
Community
OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.
Best for: Developers who need a high-performance, scalable deep learning framework and are comfortable with C++
MegEngine
Community
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架
Best for: Developers building custom deep learning models in C++ who need high performance and scalability
VectorFlow
Community
A minimalist neural network library optimized for sparse data and single machine environments.
Best for: Developers building lightweight neural network models for observability on a single machine