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:
- A framework for implementing equivariant DL☆10Updated 4 years ago
- JAX implementation of Graph Attention Networks☆13Updated 3 years ago
- Repository of models in Pyro☆29Updated 10 months ago
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 3 years ago
- ☆33Updated 4 years ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆22Updated last week
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 5 years ago
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")☆40Updated 5 years ago
- LaTeX source code for the slides☆23Updated 3 years ago
- Graph Convolutional Networks in JAX☆32Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- ☆30Updated 3 years ago
- Code for the 2-simplicial Transformer paper☆18Updated 5 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- You should use PySR to find scaling laws. Here's an example.☆33Updated last year
- Random feature latent variable models in Python☆22Updated last year
- Document parameters using comments☆10Updated 3 years ago
- Code to reproduce the results of 👇☆18Updated 2 years ago
- First-Order Probabilistic Programming Language☆28Updated 6 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Updated 3 years ago
- Companion code for a tutorial on using Hydra.☆29Updated 4 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 4 years ago
- Jax SSM Library☆49Updated 2 years ago
- Code to estimate models in epidemiology☆9Updated 2 years ago
- Probabilistic programming system for fast and exact symbolic inference☆80Updated 11 months ago
- Distributed Bayesian Optimization☆23Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 5 years ago