Alternatives to JAX

Alternatives, ranked

1. 1
   TR

   Transformers

   Python library with model definitions and pretrained weights for text, vision and audio transformers.

   • Higher score
   • Runs on more platforms
   • No shared features
   Updated 10 days ago

   4 93.5 = Compare
2. 2
   PY

   PyTorch

   Python deep learning framework with tensors, automatic differentiation and GPU acceleration.

   • Released more recently
   • Higher score
   • Runs on more platforms
   • No shared features
   Updated 1 day ago

   17 91.8 = Compare
3. 3
   SL

   scikit-learn

   Python library for classical machine learning: classification, regression, clustering and preprocessing.

   • Higher score
   • Runs on more platforms
   • No shared features
   Updated 9 days ago

   33 90.2 = Compare
4. 4
   TE

   TensorFlow

   Machine learning platform for building, training and deploying models on servers, browsers and devices.

   • Runs on more platforms
   • No shared features
   Stale: 198 days without a release

   152 69.2 = Compare