google / jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
☆29,930Updated this week
Related projects: ⓘ
- Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.☆27,963Updated this week
- Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries f…☆33,048Updated this week
- A library for efficient similarity search and clustering of dense vectors.☆30,511Updated this week
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆82,188Updated this week
- Open source platform for the machine learning lifecycle☆18,340Updated this week
- Google Research☆33,807Updated this week
- State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enter…☆13,233Updated last month
- Open standard for machine learning interoperability☆17,638Updated this week
- Visualizer for neural network, deep learning and machine learning models☆27,558Updated this week
- Flax is a neural network library for JAX that is designed for flexibility.☆5,950Updated this week
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆13,985Updated 2 months ago
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,158Updated 2 weeks ago
- NumPy & SciPy for GPU☆8,124Updated this week
- Development repository for the Triton language and compiler☆12,698Updated this week
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆30,165Updated last week
- A hyperparameter optimization framework☆10,527Updated this week
- Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including …☆23,170Updated last month
- TensorFlow code and pre-trained models for BERT☆37,839Updated last month
- Streamlit — A faster way to build and share data apps.☆34,580Updated this week
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆19,845Updated last month
- The fastai deep learning library☆26,108Updated 3 weeks ago
- Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on sing…☆26,107Updated this week
- 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.☆132,113Updated this week
- cuDF - GPU DataFrame Library☆8,259Updated this week
- Deep Learning for humans☆61,612Updated this week
- 💫 Industrial-strength Natural Language Processing (NLP) in Python☆29,654Updated this week
- A toolkit for developing and comparing reinforcement learning algorithms.☆34,479Updated 4 months ago
- 🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools☆19,012Updated this week
- Trax — Deep Learning with Clear Code and Speed☆8,044Updated last week
- A game theoretic approach to explain the output of any machine learning model.☆22,506Updated this week