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:
- Fast IdEntification of State-of-The-Art models using adaptive bandit algorithms☆14Updated 2 years ago
- Python implementation of projection losses.☆27Updated 5 years ago
- Companion code for a tutorial on using Hydra.☆29Updated 4 years ago
- ☆42Updated 4 years ago
- ☆18Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- ☆30Updated 3 years ago
- Yet Another Neural Network Library 🤔☆27Updated 2 months ago
- Distributed Bayesian Optimization☆23Updated 4 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
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated 2 years ago
- some tutorials for blog: simonjisu.github.io☆23Updated 4 years ago
- Generate BERT vocabularies and pretraining examples from Wikipedias☆17Updated 5 years ago
- The stand-alone training engine module for the ALOHA.eu project.☆15Updated 5 years ago
- Document parameters using comments☆10Updated 3 years ago
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 3 years ago
- PyTorch implementation of GLOM☆22Updated 3 years ago
- Local Attention - Flax module for Jax☆22Updated 4 years ago
- Code base for MomentumRNN.☆19Updated 4 years ago
- Code for running the experiments in Deep Subjecthood: Higher Order Grammatical Features in Multilingual BERT☆17Updated last year
- ☆45Updated 5 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 8 months ago
- Standalone pre-training recipe with JAX+Flax☆32Updated 2 years ago
- ☆11Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- The repository for the paper "When Do You Need Billions of Words of Pretraining Data?"☆21Updated 4 years ago
- A set of methods for finding an appropriate number of topics in a text collection☆16Updated 2 months ago
- ☆37Updated 3 years ago
- codes for TokenManipulationGAN☆7Updated 5 years ago