HazyResearch / bwlerLinks
Official repo for BWLer: Barycentric Weight Layer
☆23Updated last month
Alternatives and similar repositories for bwler
Users that are interested in bwler are comparing it to the libraries listed below
Sorting:
- Jax like function transformation engine but micro, microjax☆33Updated 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
- ☆32Updated last year
- ☆34Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- Code for the paper "Function-Space Learning Rates"☆23Updated 5 months ago
- A State-Space Model with Rational Transfer Function Representation.☆82Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 9 months ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- A simple example of VAEs with KANs☆11Updated last year
- Foundation Model for Physics Simulations☆107Updated 3 months ago
- ☆68Updated 11 months ago
- ☆33Updated last year
- ☆28Updated last year
- Fork of Flame repo for training of some new stuff in development☆18Updated 3 weeks ago
- We study toy models of skill learning.☆31Updated 9 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆103Updated 10 months ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 9 months ago
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆45Updated last year
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆25Updated last year
- Diffusion models in PyTorch☆112Updated this week
- ☆34Updated 11 months ago
- TorchFSM: Fourier Spectral Method with PyTorch☆50Updated last week
- ☆60Updated last year
- Patched Attention for Nonlinear Dynamics☆159Updated last month
- Explorations into the recently proposed Taylor Series Linear Attention☆99Updated last year
- JAX implementation of Kolmogorov Arnold Networks (KANs).☆10Updated last year
- Einstein Fields official implementation in JAX.☆61Updated 2 months ago
- Code for☆27Updated 10 months ago