mhlr / awesome-jax
List of awesome JAX resources
☆12Updated last year
Related projects ⓘ
Alternatives and complementary repositories for awesome-jax
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆14Updated 3 years ago
- High-performance tokenized language data-loader for Python C++ extension☆12Updated 3 months ago
- ☆33Updated 3 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆13Updated 4 years ago
- ☆30Updated 2 years ago
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 3 years ago
- ☆42Updated 4 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 3 years ago
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 2 years ago
- notebooks of cool EBM visualizations☆16Updated 3 years ago
- LaTeX source code for the slides☆22Updated 3 years ago
- JAX implementation of Graph Attention Networks☆13Updated 2 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆40Updated 4 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆23Updated 3 years ago
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 4 years ago
- A framework for implementing equivariant DL☆10Updated 3 years ago
- ☆13Updated last year
- The stand-alone training engine module for the ALOHA.eu project.☆15Updated 5 years ago
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Updated 2 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆39Updated 5 years ago
- Massively Parallel and Asynchronous Architecture for Logic-based AI☆41Updated last year
- 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 minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 4 years ago
- Universal Python binding for the LMDB 'Lightning' Database☆12Updated 7 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 5 years ago
- Reference implementation of algorithms for reinforcement learning and Markov decision processes.☆11Updated 3 years ago
- PyMC3-like Interface for Pyro☆37Updated 4 years ago