jax-ml / jaxLinks
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
☆32,533Updated this week
Alternatives and similar repositories for jax
Users that are interested in jax are comparing it to the libraries listed below
Sorting:
- Flax is a neural network library for JAX that is designed for flexibility.☆6,624Updated this week
- Efficiently computes derivatives of NumPy code.☆7,291Updated last week
- NumPy & SciPy for GPU☆10,283Updated this week
- A hyperparameter optimization framework☆12,136Updated this week
- Open standard for machine learning interoperability☆19,121Updated this week
- Deep Learning for humans☆63,115Updated last week
- JAX-based neural network library☆3,044Updated last week
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆90,884Updated this week
- A scikit-learn compatible neural network library that wraps PyTorch☆6,052Updated last week
- NumPy aware dynamic Python compiler using LLVM☆10,482Updated this week
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.☆29,644Updated this week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,512Updated last month
- Deep universal probabilistic programming with Python and PyTorch☆8,798Updated last month
- AutoML library for deep learning☆9,239Updated 6 months ago
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆31,543Updated last week
- Distributed Asynchronous Hyperparameter Optimization in Python☆7,432Updated 3 weeks ago
- Unsupervised text tokenizer for Neural Network-based text generation.☆10,994Updated 2 months ago
- 💫 Industrial-strength Natural Language Processing (NLP) in Python☆31,784Updated 3 weeks ago
- The fastai deep learning library☆27,064Updated 2 weeks ago
- Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.☆16,218Updated 2 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,991Updated last month
- Trax — Deep Learning with Clear Code and Speed☆8,223Updated 2 months ago
- scikit-learn: machine learning in Python☆62,389Updated this week
- Open source platform for the machine learning lifecycle☆20,923Updated this week
- TensorFlow's Visualization Toolkit☆6,905Updated 2 weeks ago
- TensorFlow-based neural network library☆9,860Updated last week
- 💥 Fast State-of-the-Art Tokenizers optimized for Research and Production☆9,817Updated this week
- 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal model…☆145,689Updated this week
- Hydra is a framework for elegantly configuring complex applications☆9,407Updated last month
- Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.☆37,558Updated this week