catswe / LinearKANLinks
LinearKAN: A very fast implementation of Kolmogorov-Arnold Networks
☆18Updated 2 months ago
Alternatives and similar repositories for LinearKAN
Users that are interested in LinearKAN are comparing it to the libraries listed below
Sorting:
- all the materials for cs140e winter 2026☆33Updated this week
- ☆16Updated last year
- Implementation of Diffusion Transformer (DiT) in JAX☆305Updated last year
- ☆238Updated last year
- Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.☆26Updated 11 months ago
- ☆289Updated last year
- ☆135Updated last month
- Dion optimizer algorithm☆420Updated 2 weeks ago
- ☆540Updated 5 months ago
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆157Updated 2 months ago
- 🧱 Modula software package☆322Updated 5 months ago
- A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.☆457Updated 10 months ago
- A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.☆114Updated last month
- ☆27Updated last year
- Simple MPI implementation for prototyping or learning☆300Updated 5 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆349Updated 2 months ago
- For optimization algorithm research and development.☆558Updated 2 weeks ago
- Minimal yet performant LLM examples in pure JAX☆236Updated 2 weeks ago
- ☆234Updated last year
- Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"☆86Updated 4 months ago
- UNet diffusion model in pure CUDA☆662Updated last year
- Tensor library with autograd using only Rust's standard library☆71Updated last year
- making the official triton tutorials actually comprehensible☆104Updated 5 months ago
- Accelerated First Order Parallel Associative Scan☆196Updated 3 weeks ago
- Efficient optimizers☆281Updated last month
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆53Updated last year
- Learnings and programs related to CUDA☆431Updated 7 months ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆827Updated this week
- Easy no-frills Jax implementations of common abstractions for simple diffusion models.☆10Updated 9 months ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆197Updated 8 months ago