mhlr / awesome-jaxLinks
List of awesome JAX resources
☆13Updated 2 years ago
Alternatives and similar repositories for awesome-jax
Users that are interested in awesome-jax are comparing it to the libraries listed below
Sorting:
- Companion code for a tutorial on using Hydra.☆29Updated 4 years ago
- ☆42Updated 5 years ago
- ☆26Updated 2 years ago
- High-performance tokenized language data-loader for Python C++ extension☆13Updated 10 months ago
- LaTeX source code for the slides☆23Updated 3 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"☆31Updated 4 years ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆16Updated 3 years ago
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 5 years ago
- Code to estimate models in epidemiology☆9Updated 2 years ago
- Numerically Solving Parametric Families of High-Dimensional Kolmogorov Partial Differential Equations via Deep Learning (NeurIPS 2020)☆22Updated 2 years ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆22Updated 2 years ago
- ☆30Updated 3 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Updated 5 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Updated 5 years ago
- ☆33Updated 4 years ago
- Code for the 2-simplicial Transformer paper☆18Updated 5 years ago
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆50Updated last year
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago