Alternatives and similar repositories for jax

Users that are interested in jax are comparing it to the libraries listed below

• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,977Updated this week
• pytorch / pytorch
  Tensors and Dynamic neural networks in Python with strong GPU acceleration
  ☆95,988Updated this week
• cupy / cupy
  NumPy & SciPy for GPU
  ☆10,666Updated last week
• HIPS / autograd
  Efficiently computes derivatives of NumPy code.
  ☆7,418Updated last week
• mlflow / mlflow
  The open source developer platform to build AI agents and models with confidence. Enhance your AI applications with end-to-end tracking, …
  ☆23,329Updated this week
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
  ☆30,586Updated this week
• optuna / optuna
  A hyperparameter optimization framework
  ☆13,217Updated this week
• horovod / horovod
  Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
  ☆14,641Updated 2 weeks ago
• ray-project / ray
  Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
  ☆40,344Updated this week
• onnx / onnx
  Open standard for machine learning interoperability
  ☆20,031Updated this week
• google-deepmind / deepmind-research
  This repository contains implementations and illustrative code to accompany DeepMind publications
  ☆14,552Updated last week
• google-deepmind / dm-haiku
  JAX-based neural network library
  ☆3,140Updated last week
• numba / numba
  NumPy aware dynamic Python compiler using LLVM
  ☆10,794Updated this week
• keras-team / keras
  Deep Learning for humans
  ☆63,631Updated this week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆9,312Updated 3 weeks ago
• rapidsai / cudf
  cuDF - GPU DataFrame Library
  ☆9,375Updated this week
• facebookresearch / hydra
  Hydra is a framework for elegantly configuring complex applications
  ☆10,036Updated last week
• google / trax
  Trax — Deep Learning with Clear Code and Speed
  ☆8,294Updated 2 months ago
• NVIDIA / DeepLearningExamples
  State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…
  ☆14,630Updated last year
• pyro-ppl / pyro
  Deep universal probabilistic programming with Python and PyTorch
  ☆8,933Updated 5 months ago
• lutzroeder / netron
  Visualizer for neural network, deep learning and machine learning models
  ☆32,003Updated this week
• dask / dask
  Parallel computing with task scheduling
  ☆13,648Updated last week
• pytorch / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆23,604Updated 3 months ago
• google-research / google-research
  Google Research
  ☆36,886Updated this week
• gradio-app / gradio
  Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!
  ☆40,937Updated this week
• triton-lang / triton
  Development repository for the Triton language and compiler
  ☆17,861Updated this week
• microsoft / nni
  An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…
  ☆14,312Updated last year
• explosion / spaCy
  💫 Industrial-strength Natural Language Processing (NLP) in Python
  ☆32,955Updated 3 weeks ago
• spotify / annoy
  Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk
  ☆14,092Updated last month
• google / sentencepiece
  Unsupervised text tokenizer for Neural Network-based text generation.
  ☆11,508Updated this week