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:
- ☆30Updated 3 years ago
- A deep learning library based on Pytorch focussed on low resource language research and robustness☆70Updated 3 years ago
- Code repo for "Transformer on a Diet" paper☆31Updated 5 years ago
- Implementation of "Analysing Mathematical Reasoning Abilities of Neural Models"☆30Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 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
- Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing☆13Updated 3 years ago
- Yet Another Neural Network Library 🤔☆27Updated 4 months ago
- Python implementation of projection losses.☆27Updated 5 years ago
- ☆41Updated 4 years ago
- Distributed Bayesian Optimization☆23Updated 5 years ago
- The repository for the paper "When Do You Need Billions of Words of Pretraining Data?"☆21Updated 4 years ago
- Companion code for a tutorial on using Hydra.☆30Updated 4 years ago
- ☆11Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Implementations of normalizing flows using python and tensorflow☆24Updated 8 months ago
- ☆30Updated 3 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Updated 5 years ago
- DIY setup for virtual teaching on ubuntu☆39Updated 4 years ago
- AdaCat☆49Updated 3 years ago
- Deep Probabilistic Programming Course @ DIKU☆58Updated 4 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- Stats 479 Project☆22Updated 6 years ago
- General Invertible Transformations for Flow-based Generative Models☆18Updated 4 years ago
- A repository containing the code for the Bistable Recurrent Cell☆47Updated 4 years ago
- Some small scale experiments for my blog posts 📝☆79Updated 3 years ago
- ☆37Updated 3 years ago
- This repository provides the code examples for the corresponding blog posts. In case you have questions, feel free to contact me directly…☆51Updated 4 years ago