craffel / jax-tutorial
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
- An easy way to start a python programming environment using GitHub Codespaces.☆15Updated 4 years ago
- notebooks of cool EBM visualizations☆15Updated 4 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- Companion code for a tutorial on using Hydra.☆29Updated 3 years ago
- ☆30Updated 3 years ago
- ☆30Updated 3 years ago
- Code accompanying the paper 'Manifold MCMC methods for Bayesian inference in a wide class of diffusion models'☆10Updated last year
- Fast IdEntification of State-of-The-Art models using adaptive bandit algorithms☆14Updated 2 years ago
- codes for TokenManipulationGAN☆7Updated 4 years ago
- General Invertible Transformations for Flow-based Generative Models☆17Updated 4 years ago
- Experiment management with Hydra and MLflow☆13Updated 4 years ago
- Code for running the experiments in Deep Subjecthood: Higher Order Grammatical Features in Multilingual BERT☆17Updated last year
- Python implementation of projection losses.☆25Updated 5 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆9Updated 4 years ago
- ☆17Updated last month
- Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"☆31Updated 4 years ago
- Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing☆13Updated 2 years ago
- code for the paper "Cluster & Tune: Boost Cold Start Performance in Text Classification" for ACL2022☆28Updated 2 years ago
- Materials for "Transformers from the Ground Up" at PyData Jeddah on August 5, 2021☆18Updated 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
- ☆18Updated 3 years ago
- Code for "On the Expressiveness of Approximate Inference in Bayesian Neural Networks"☆13Updated 3 years ago
- Python package for graph statistics☆9Updated 4 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Updated 2 years ago
- ☆45Updated 5 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- Yet Another Neural Network Library 🤔☆26Updated 2 months ago
- Code for "MIM: Mutual Information Machine" paper.☆16Updated 2 years ago
- Deep Probabilistic Programming Course @ DIKU☆57Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago