brantondemoss / GrokkingComplexity
Code for
☆25Updated 2 months ago
Alternatives and similar repositories for GrokkingComplexity:
Users that are interested in GrokkingComplexity are comparing it to the libraries listed below
- ☆31Updated 9 months ago
- Triton Implementation of HyperAttention Algorithm☆46Updated last year
- Implementation of Spectral State Space Models☆16Updated 11 months ago
- Jax like function transformation engine but micro, microjax☆30Updated 3 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 2 weeks ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆27Updated 4 years ago
- A repository for research on medium sized language models.☆76Updated 8 months ago
- ☆21Updated 3 months ago
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]☆18Updated last month
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 months ago
- ☆31Updated 10 months ago
- ☆51Updated 5 months ago
- ☆53Updated last year
- Evaluation of neuro-symbolic engines☆34Updated 6 months ago
- Efficient Scaling laws and collaborative pretraining.☆14Updated 3 weeks ago
- ☆38Updated 6 months ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆36Updated last year
- Using FlexAttention to compute attention with different masking patterns☆40Updated 5 months ago
- ☆42Updated last year