HazyResearch / bwlerLinks
Official repo for BWLer: Barycentric Weight Layer
☆23Updated 2 weeks ago
Alternatives and similar repositories for bwler
Users that are interested in bwler are comparing it to the libraries listed below
Sorting:
- ☆32Updated last year
- Jax like function transformation engine but micro, microjax☆33Updated 11 months ago
- ☆28Updated last year
- Code for the paper "Function-Space Learning Rates"☆23Updated 4 months ago
- We study toy models of skill learning.☆31Updated 8 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- ☆33Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆82Updated last year
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆25Updated 11 months ago
- A simple example of VAEs with KANs☆12Updated last year
- ☆34Updated last year
- Fork of Flame repo for training of some new stuff in development☆18Updated this week
- Foundation Model for Physics Simulations☆104Updated 3 months ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆102Updated 9 months ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆96Updated 2 months ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Patched Attention for Nonlinear Dynamics☆157Updated last week
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 8 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆99Updated last year
- Diffusion models in PyTorch☆111Updated 3 weeks ago
- Generative cellular automaton-like learning environments for RL.☆19Updated 8 months ago
- A system for automating selection and optimization of pre-trained models from the TAO Model Zoo☆25Updated last year
- ☆58Updated last year
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆45Updated last year
- ☆67Updated 10 months ago
- 📄Small Batch Size Training for Language Models☆63Updated last week
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆18Updated 2 months ago
- ☆102Updated 2 months ago
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆85Updated last month