The-AI-Summer / JAX-examplesLinks
A set of of fundamental operations and deep learning models using JAX
☆14Updated 4 years ago
Alternatives and similar repositories for JAX-examples
Users that are interested in JAX-examples are comparing it to the libraries listed below
Sorting:
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- ML/DL Math and Method notes☆64Updated last year
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- Scripts to convert datasets from various sources to Hugging Face Datasets.☆57Updated 3 years ago
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Updated 3 years ago
- Examples of using PyTorch hooks, as covered in my YouTube tutorial video.☆38Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Updated last year
- Reads arXiv papers using Text-to-Speech☆62Updated 2 years ago
- This project shows how to derive the total number of training tokens from a large text dataset from 🤗 datasets with Apache Beam and Data…☆27Updated 3 years ago
- Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow ✅, Pytorch 🔜, and Jax 🔜)☆53Updated last year
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Updated 3 years ago
- A Streamlit app to add structured tags to a dataset card☆22Updated 3 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆58Updated 3 years ago
- Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.☆45Updated 4 years ago
- A multi-label text classifier to predict the subject areas of arXiv papers from their abstract bodies.☆17Updated 4 years ago
- Fastai community entry to 2020 Reproducibility Challenge☆17Updated 3 years ago
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Updated 4 years ago
- ☆75Updated 2 years ago
- ☆16Updated 2 years ago
- ☆36Updated 3 years ago
- NLP Examples using the 🤗 libraries☆40Updated 4 years ago
- AI Summer's complete catalog of articles☆112Updated 3 years ago
- Plugin for deploying MLflow models to TorchServe☆110Updated 2 years ago
- This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.☆30Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆159Updated 3 years ago
- Personal implementation of the Transformer paper.☆22Updated 2 years ago
- Unofficial PyTorch implementation of Fastformer based on paper "Fastformer: Additive Attention Can Be All You Need"."☆133Updated 4 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 4 years ago
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.☆12Updated 2 years ago