craffel / jax-tutorialLinks
A tutorial on JAX (https://github.com/google/jax/)
☆47Updated 6 years ago
Alternatives and similar repositories for jax-tutorial
Users that are interested in jax-tutorial are comparing it to the libraries listed below
Sorting:
- A deep learning library based on Pytorch focussed on low resource language research and robustness☆70Updated 4 years ago
- Implementation of "Analysing Mathematical Reasoning Abilities of Neural Models"☆30Updated 2 years ago
- ☆30Updated 4 years ago
- ☆30Updated 3 years ago
- Code repo for "Transformer on a Diet" paper☆31Updated 5 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Distributed Bayesian Optimization☆23Updated 5 years ago
- Tools for working with Long Short-Term Memory (LSTM) networks and sequences in Pytorch☆36Updated 4 years ago
- Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing☆14Updated 3 years ago
- A queue service for quickly developing scripts that use all your GPUs efficiently☆88Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- An easy way to start a python programming environment using GitHub Codespaces.☆15Updated 5 years ago
- Yet Another Neural Network Library 🤔☆27Updated 4 months ago
- NLP Examples using the 🤗 libraries☆40Updated 4 years ago
- The stand-alone training engine module for the ALOHA.eu project.☆15Updated 6 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆27Updated 4 years ago
- The repository for the paper "When Do You Need Billions of Words of Pretraining Data?"☆21Updated 5 years ago
- Composable kernels for scikit-learn implemented in JAX.☆45Updated 5 years ago
- Agents that build knowledge graphs and explore textual worlds by asking questions☆79Updated 2 years ago
- ☆40Updated 4 years ago
- ☆31Updated last month
- Local Attention - Flax module for Jax☆22Updated 4 years ago
- Companion code for a tutorial on using Hydra.☆30Updated 4 years ago
- ☆68Updated 9 months ago
- A generative modelling toolkit for PyTorch.☆70Updated 4 years ago
- ☆45Updated 6 years ago
- Python implementation of projection losses.☆27Updated 6 years ago
- Some small scale experiments for my blog posts 📝☆80Updated 3 years ago