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:
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- ☆30Updated 3 years ago
- Fast IdEntification of State-of-The-Art models using adaptive bandit algorithms☆14Updated 2 years ago
- Companion code for a tutorial on using Hydra.☆29Updated 4 years ago
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 3 years ago
- ☆18Updated 3 years ago
- ☆30Updated 3 years ago
- Analogous Safe-state Exploration (ASE) is an algorithm for provably safe and optimal exploration in MDPs with unknown, stochastic dynamic…☆11Updated 4 years ago
- Yet Another Neural Network Library 🤔☆27Updated 2 months ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- notebooks of cool EBM visualizations☆15Updated 4 years ago
- General Invertible Transformations for Flow-based Generative Models☆18Updated 4 years ago
- ☆42Updated 4 years ago
- Repository of models in Pyro☆29Updated 10 months ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆26Updated 4 years ago
- Python implementation of projection losses.☆26Updated 5 years ago
- Code for running the experiments in Deep Subjecthood: Higher Order Grammatical Features in Multilingual BERT☆17Updated last year
- AdaCat☆49Updated 2 years ago
- PyTorch implementation of GLOM☆22Updated 3 years ago
- codes for TokenManipulationGAN☆7Updated 5 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- Code repo for "Transformer on a Diet" paper☆31Updated 4 years ago
- ☆45Updated 5 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Updated 5 years ago
- Probabilistic Solution of Differential Equations☆13Updated 2 years ago
- ☆37Updated 3 years ago
- Measuring if attention is explanation with ROAR☆22Updated 2 years ago
- Train transformer-based models.☆28Updated 3 weeks ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆49Updated 3 years ago