HazyResearch / bwlerLinks
Official repo for BWLer: Barycentric Weight Layer
☆28Updated 2 months ago
Alternatives and similar repositories for bwler
Users that are interested in bwler are comparing it to the libraries listed below
Sorting:
- Code for the paper "Function-Space Learning Rates"☆23Updated 6 months ago
- ☆32Updated last year
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆25Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆83Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- ☆34Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆103Updated 11 months ago
- ☆28Updated last year
- Foundation Model for Physics Simulations☆110Updated 5 months ago
- Jax like function transformation engine but micro, microjax☆34Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 11 months ago
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated last year
- ☆190Updated 3 weeks ago
- ☆33Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated last year
- Supporting code for the blog post on modular manifolds.☆105Updated 2 months ago
- We study toy models of skill learning.☆31Updated 11 months ago
- Deep Networks Grok All the Time and Here is Why☆38Updated last year
- Implementation of Spectral State Space Models☆16Updated last year
- ☆33Updated 11 months ago
- Patched Attention for Nonlinear Dynamics☆162Updated this week
- ☆105Updated 4 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆68Updated last month
- The 2D discrete wavelet transform for JAX☆43Updated 2 years ago
- ☆82Updated last year
- A simple example of VAEs with KANs☆11Updated last year
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆30Updated last year
- Code for Lie Symmetries SSL paper☆24Updated last year
- Schedule free optimiser implemented in JAX using Optimistix☆15Updated last year
- ☆153Updated last month