berndprach / AOLLinks
Code for paper Almost-Orthogonal Layers for Efficient General-Purpose Lipschitz Networks
☆12Updated 3 years ago
Alternatives and similar repositories for AOL
Users that are interested in AOL are comparing it to the libraries listed below
Sorting:
- Neural Optimal Transport with Lagrangian Costs☆61Updated 8 months ago
- NF-Layers for constructing neural functionals.☆93Updated 2 years ago
- ☆39Updated 2 weeks ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆42Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- JAX exponential map normalising flows on sphere☆17Updated 5 years ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)☆43Updated 3 years ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Updated 3 years ago
- Graphically structured diffusion model.☆21Updated 2 years ago
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- Riemannian Optimization Using JAX☆54Updated 2 years ago
- PyTorch implementation of "Wasserstein-2 Generative Networks" (ICLR 2021)☆66Updated 3 years ago
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- ☆33Updated last year
- Convex potential flows☆85Updated 4 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆29Updated 4 years ago
- Modern Fixed Point Systems using Pytorch☆127Updated 2 years ago
- ☆55Updated last year
- Monotone operator equilibrium networks☆54Updated 5 years ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆29Updated 3 years ago
- Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"☆40Updated 2 years ago
- ☆73Updated last year
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆24Updated last month
- Euclidean Wasserstein-2 optimal transportation☆46Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆72Updated 10 months ago
- ☆36Updated 10 months ago
- A simple pytorch implementation of Langevin Monte Carlo algorithms.☆53Updated 5 years ago
- ☆22Updated 4 years ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆28Updated 3 months ago
- Transformers with doubly stochastic attention☆53Updated 3 years ago