craffel / jax-tutorialLinks
A tutorial on JAX (https://github.com/google/jax/)
☆46Updated 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:
- Yet Another Neural Network Library 🤔☆27Updated 3 months ago
- Code repo for "Transformer on a Diet" paper☆31Updated 5 years ago
- ☆30Updated 3 years ago
- Implementation of "Analysing Mathematical Reasoning Abilities of Neural Models"☆30Updated 2 years ago
- Tools for working with Long Short-Term Memory (LSTM) networks and sequences in Pytorch☆36Updated 4 years ago
- Adversarial examples to the new ConvNeXt architecture☆20Updated 3 years ago
- codes for TokenManipulationGAN☆7Updated 5 years ago
- Code accompanying the paper 'Manifold MCMC methods for Bayesian inference in a wide class of diffusion models'☆10Updated last month
- Investigate the speed of adaptation of structural causal models☆15Updated 4 years ago
- Implementation of the ALI-G algorithm (PyTorch, Tensorflow)☆22Updated 4 years ago
- Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"☆13Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- General Invertible Transformations for Flow-based Generative Models☆18Updated 4 years ago
- Deep Probabilistic Programming Course @ DIKU☆57Updated 4 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- Curated materials for different machine learning related summer schools☆19Updated 4 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆27Updated 4 years ago
- code for the paper "Cluster & Tune: Boost Cold Start Performance in Text Classification" for ACL2022☆28Updated 3 years ago
- Test pytorch code with minimal computational overhead☆25Updated 2 years ago
- A deep learning library based on Pytorch focussed on low resource language research and robustness☆70Updated 3 years ago
- Python implementation of projection losses.☆27Updated 5 years ago
- Companion code for a tutorial on using Hydra.☆29Updated 4 years ago
- An easy way to start a python programming environment using GitHub Codespaces.☆16Updated 4 years ago
- Universal Python binding for the LMDB 'Lightning' Database☆13Updated 7 years ago
- ☆31Updated 3 weeks ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 8 months ago
- ☆41Updated 4 years ago
- Using / reproducing DAC from the paper "Disentangled Attribution Curves for Interpreting Random Forests and Boosted Trees"☆28Updated 4 years ago
- Slides, Jupyter Notebooks and scripts for the Deep Learning: Do-It-Yourself! lectures at ENS☆21Updated 4 years ago
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retain☆34Updated 4 years ago