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:
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- High-performance tokenized language data-loader for Python C++ extension☆13Updated last year
- Massively Parallel and Asynchronous Architecture for Logic-based AI☆42Updated 2 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- ☆33Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- ☆42Updated 5 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 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
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆49Updated last year
- 💡 Learnergy is a Python library for energy-based machine learning models.☆66Updated 5 months ago
- Quantification of Uncertainty with Adversarial Models☆30Updated 2 years ago
- notebooks of cool EBM visualizations☆15Updated 4 years ago
- Evaluation Framework for Probabilistic Programming Languages☆101Updated last year
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- A simple OpenAI Gym environment for Neural Architecture Search (NAS)☆30Updated 5 years ago
- ☆12Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Jax SSM Library☆49Updated 2 years ago
- Companion code for a tutorial on using Hydra.☆30Updated 4 years ago
- Code for the 2-simplicial Transformer paper☆21Updated 5 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Accelerated Stochastic Power Iteration with Momentum☆9Updated 7 years ago
- ☆108Updated 2 years ago
- Parallel random matrix tools and complexity for deep learning☆33Updated last year
- Variational inference for hierarchical dynamical systems☆48Updated last year
- A framework for implementing equivariant DL☆10Updated 4 years ago