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:
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Updated 5 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Updated 5 years ago
- ☆42Updated 5 years ago
- 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☆43Updated 2 years ago
- Evaluation Framework for Probabilistic Programming Languages☆103Updated last year
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- Quantification of Uncertainty with Adversarial Models☆29Updated 2 years ago
- ☆33Updated 4 years ago
- A framework for implementing equivariant DL☆10Updated 4 years ago
- 💡 Learnergy is a Python library for energy-based machine learning models.☆68Updated 7 months ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆22Updated 3 years ago
- Variational inference for hierarchical dynamical systems☆48Updated last year
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- ☆192Updated 3 months ago
- Repository of models in Pyro☆30Updated last year
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 4 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆49Updated last year
- A simple OpenAI Gym environment for Neural Architecture Search (NAS)☆30Updated 5 years ago
- Deep relational learning through differentiable logic programming.☆112Updated last month
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 years ago
- A Gentle Principled Introduction to Deep Reinforcement Learning☆19Updated 6 months ago
- ☆108Updated 2 years ago