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:
- High-performance tokenized language data-loader for Python C++ extension☆13Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆45Updated 5 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Updated 5 years ago
- Evaluation Framework for Probabilistic Programming Languages☆103Updated last year
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- Numerically Solving Parametric Families of High-Dimensional Kolmogorov Partial Differential Equations via Deep Learning (NeurIPS 2020)☆22Updated 3 years ago
- Companion code for a tutorial on using Hydra.☆30Updated 4 years ago
- Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"☆14Updated 5 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- 💡 Learnergy is a Python library for energy-based machine learning models.☆67Updated 7 months ago
- Massively Parallel and Asynchronous Architecture for Logic-based AI☆43Updated 2 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 3 years ago
- Quantification of Uncertainty with Adversarial Models☆29Updated 2 years ago
- ☆42Updated 5 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- Deep relational learning through differentiable logic programming.☆111Updated 2 months ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆24Updated 2 years ago
- ☆108Updated 2 years ago
- Jax SSM Library☆48Updated 2 years ago
- Code for the 2-simplicial Transformer paper☆21Updated 5 years ago
- Variational inference for hierarchical dynamical systems☆48Updated last year
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆49Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆33Updated 4 years ago
- Library for learning and inference with Sum-product Networks utilizing TensorFlow 2.x and Keras☆48Updated last month
- PyMC3-like Interface for Pyro☆38Updated 5 years ago
- Functional machine learning for fun☆85Updated 4 years ago