captain-pool / hydra-example
☆20Updated 2 years ago
Alternatives and similar repositories for hydra-example:
Users that are interested in hydra-example are comparing it to the libraries listed below
- Official code for Coupled Oscillatory RNN (ICLR 2021, Oral)☆44Updated 3 years ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- ☆36Updated 2 years ago
- Some small scale experiments for my blog posts 📝☆79Updated 2 years ago
- Collection of snippets for PyTorch users☆25Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆9Updated 4 years ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- ☆18Updated 3 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆62Updated 4 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Code for the paper "Attention Approximates Sparse Distributed Memory"☆22Updated 3 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- ☆38Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆82Updated last year
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆27Updated 2 years ago
- ☆24Updated last year
- ☆72Updated 2 years ago
- Codes for the paper "A mathematical perspective on Transformers".☆35Updated 6 months ago
- Probabilistic Solution of Differential Equations☆13Updated 2 years ago
- A basic implementation of the paper Eigengame : PCA as a Nash Equilibrium☆21Updated 3 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- A metrics library for the JAX ecosystem☆39Updated last year
- notebooks of cool EBM visualizations☆15Updated 3 years ago
- Scalable Computation of Hessian Diagonals☆12Updated 7 months ago
- ☆191Updated 8 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆41Updated last year
- PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆34Updated 3 years ago
- Official implementation of the paper "Interventions, Where and How? Experimental Design for Causal Models at Scale", NeurIPS 2022.☆19Updated 2 years ago
- codebase for "A Theory of the Inductive Bias and Generalization of Kernel Regression and Wide Neural Networks"☆50Updated last year
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year