jax-ml / 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,996Updated 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,079Updated this week
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆82,248Updated this week
- The fastai deep learning library☆26,122Updated this week
- Google Research☆33,830Updated this week
- A library for efficient similarity search and clustering of dense vectors.☆30,511Updated this week
- Open standard for machine learning interoperability☆17,647Updated 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
- Flax is a neural network library for JAX that is designed for flexibility.☆5,955Updated this week
- Development repository for the Triton language and compiler☆12,744Updated this week
- NumPy & SciPy for GPU☆8,124Updated last week
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆30,180Updated last week
- Visualizer for neural network, deep learning and machine learning models☆27,558Updated this week
- Deep Learning for humans☆61,612Updated this week
- 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.☆132,113Updated this week
- Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.☆14,158Updated 2 weeks ago
- The fundamental package for scientific computing with Python.☆27,559Updated this week
- Open source platform for the machine learning lifecycle☆18,352Updated this week
- scikit-learn: machine learning in Python☆59,448Updated this week
- A hyperparameter optimization framework☆10,550Updated this week
- Streamlit — A faster way to build and share data apps.☆34,580Updated this week
- A toolkit for developing and comparing reinforcement learning algorithms.☆34,502Updated 4 months ago
- A game theoretic approach to explain the output of any machine learning model.☆22,506Updated this week
- An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model c…☆13,990Updated 2 months ago
- Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including …☆23,193Updated last month
- TensorFlow code and pre-trained models for BERT☆37,839Updated last month
- cuDF - GPU DataFrame Library☆8,259Updated this week
- Trax — Deep Learning with Clear Code and Speed☆8,044Updated last week
- 💫 Industrial-strength Natural Language Processing (NLP) in Python☆29,678Updated this week
- Models and examples built with TensorFlow☆76,964Updated this week