HazyResearch / bwlerLinks
Official repo for BWLer: Barycentric Weight Layer
☆28Updated 3 months ago
Alternatives and similar repositories for bwler
Users that are interested in bwler are comparing it to the libraries listed below
Sorting:
- Foundation Model for Physics Simulations☆110Updated 6 months ago
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆25Updated last year
- ☆35Updated last year
- Code for the paper "Function-Space Learning Rates"☆23Updated 7 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆83Updated last year
- Generative cellular automaton-like learning environments for RL.☆20Updated 11 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 11 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆103Updated last year
- Deep Networks Grok All the Time and Here is Why☆38Updated last year
- ☆209Updated last month
- Fork of Flame repo for training of some new stuff in development☆19Updated this week
- Code for Lie Symmetries SSL paper☆24Updated last year
- Jax like function transformation engine but micro, microjax☆34Updated last year
- ☆32Updated last year
- Implementation and explorations into Blackbox Gradient Sensing (BGS), an evolutionary strategies approach proposed in a Google Deepmind p…☆20Updated 5 months ago
- ☆27Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆100Updated last year
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆30Updated last year
- We study toy models of skill learning.☆31Updated 11 months ago
- Kolmogorov-Arnold Networks with various basis functions like B-Splines, Fourier, Chebyshev, Wavelets etc☆35Updated last year
- Induce brain-like topographic structure in your neural networks☆71Updated 5 months ago
- FID computation in Jax/Flax.☆29Updated last year
- Experimental scripts for researching data adaptive learning rate scheduling.☆22Updated 2 years ago
- ☆33Updated last year
- [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"☆22Updated 2 months ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆97Updated 5 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- A simple example of VAEs with KANs☆11Updated last year
- Einstein Fields official implementation in JAX.☆67Updated 4 months ago