Open source alternatives to JAX

Open source alternatives to JAX, ranked by GitHub stars and freshness.

5 open-source alternatives in the index, ranked by GitHub stars and freshness.

Community

An Open Source Machine Learning Framework for Everyone

Alternative to Pytorch, Jax

Best for: Teams building production ML systems that need cross-platform deployment and performance optimization

Community

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Alternative to Tensorflow, Jax, Paddlepaddle +1 more

Best for: ML researchers and engineers building custom neural network architectures with GPU training needs

Community

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）

Alternative to Tensorflow, Pytorch, Caffe +2 more

Best for: Teams building large-scale production ML systems who need distributed training and cross-platform deployment out of the box

Community

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.

Alternative to Tensorflow, Pytorch, Jax +1 more

Best for: Developers who need a high-performance, scalable deep learning framework and are comfortable with C++

Community

MindSpore is a new open source deep learning training/inference framework that could be used for mobile, edge and cloud scenarios.

Alternative to Tensorflow, Pytorch, Keras +2 more

Best for: Developers seeking an open source deep learning framework for cross-platform deployment from edge to cloud