craffel / jax-tutorial
A tutorial on JAX (https://github.com/google/jax/)
☆45Updated 5 years ago
Related projects ⓘ
Alternatives and complementary repositories for jax-tutorial
- Companion code for a tutorial on using Hydra.☆27Updated 3 years ago
- ☆30Updated 2 years ago
- Investigate the speed of adaptation of structural causal models☆16Updated 3 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆13Updated 4 years ago
- ☆42Updated 3 years ago
- Local Attention - Flax module for Jax☆20Updated 3 years ago
- Yet Another Neural Network Library 🤔☆25Updated 7 months ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆40Updated 4 years ago
- codes for TokenManipulationGAN☆7Updated 4 years ago
- An easy way to start a python programming environment using GitHub Codespaces.☆15Updated 4 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- Python implementation of projection losses.☆25Updated 5 years ago
- General Invertible Transformations for Flow-based Generative Models☆17Updated 3 years ago
- Code base for MomentumRNN.☆18Updated 4 years ago
- Analogous Safe-state Exploration (ASE) is an algorithm for provably safe and optimal exploration in MDPs with unknown, stochastic dynamic…☆11Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- Code accompanying the paper 'Manifold MCMC methods for Bayesian inference in a wide class of diffusion models'☆10Updated last year
- The Union of Intersections Framework in Python☆13Updated last year
- Repository of models in Pyro☆29Updated 3 months ago
- ☆45Updated 5 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 5 years ago
- notebooks of cool EBM visualizations☆16Updated 3 years ago
- The repository for the paper "When Do You Need Billions of Words of Pretraining Data?"☆20Updated 4 years ago
- GASP! Dataset - Generating Abstracts of Scientific Papers from Abstracts of Cited Papers☆9Updated 4 years ago
- ☆18Updated 2 years ago
- ☆36Updated 2 years ago
- Layerwise Batch Entropy Regularization☆22Updated 2 years ago
- ☆16Updated 2 years ago