The-AI-Summer / JAX-examplesLinks
A set of of fundamental operations and deep learning models using JAX
☆13Updated 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:
- Cyclemoid implementation for PyTorch☆89Updated 3 years ago
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Updated 3 years ago
- A case study of efficient training of large language models using commodity hardware.☆69Updated 2 years ago
- Fastai community entry to 2020 Reproducibility Challenge☆18Updated 2 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- A multi-label text classifier to predict the subject areas of arXiv papers from their abstract bodies.☆17Updated 3 years ago
- ☆16Updated 2 years ago
- ☆24Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Updated last year
- Implementation of Feedback Transformer in Pytorch☆107Updated 4 years ago
- deep learning with pytorch lightning☆1Updated 7 months ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- NLP Examples using the 🤗 libraries☆41Updated 4 years ago
- TorchServe+Streamlit for easily serving your HuggingFace NER models☆33Updated 2 years ago
- Recurrent neural networks: building a custom LSTM/GRU cell in PyTorch☆28Updated 4 years ago
- ☆14Updated 2 years ago
- In-place debugger for the fastai library and pytorch☆28Updated 3 years ago
- Distributed skorch on Ray Train☆57Updated 2 years ago
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- Implementation of N-Grammer in Flax☆17Updated 2 years ago
- Another attempt at a long-context / efficient transformer by me☆38Updated 3 years ago
- Recursive Leasting Squares (RLS) with Neural Network for fast learning☆52Updated last year
- Examples of using Neptune to keep track of your experiments (maintenance only).☆26Updated 3 years ago
- Implementation of Metaformer, but in an autoregressive manner☆25Updated 2 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆50Updated 3 years ago
- ML/DL Math and Method notes☆61Updated last year
- ☆9Updated 2 years ago
- Basic guidance on how to contribute to Papers with Code☆23Updated 3 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆53Updated 2 years ago
- PyTorch implementation of GLOM☆22Updated 3 years ago