fpaboim / tinysparse
A fork of tinygrad made to work with sparse tensors. Sparse neural networks are here!
☆11Updated 3 years ago
Alternatives and similar repositories for tinysparse:
Users that are interested in tinysparse are comparing it to the libraries listed below
- Latent Large Language Models☆17Updated 8 months ago
- ☆49Updated last year
- RWKV model implementation☆37Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 11 months ago
- GPU accelerated client-side embeddings for vector search, RAG etc.☆66Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆60Updated 3 years ago
- train with kittens!☆57Updated 6 months ago
- ☆19Updated 3 weeks ago
- Jax like function transformation engine but micro, microjax☆30Updated 6 months ago
- ☆21Updated 5 months ago
- ☆53Updated last year
- ☆38Updated 9 months ago
- Genalog is an open source, cross-platform python package allowing generation of synthetic document images with custom degradations and te…☆42Updated last year
- ☆22Updated last year
- Automatically take good care of your preemptible TPUs☆36Updated last year
- Implementation of the Mamba SSM with hf_integration.☆56Updated 7 months ago
- This is a port of Mistral-7B model in JAX☆32Updated 9 months ago
- Collection of autoregressive model implementation☆85Updated 2 months ago
- FastFeedForward Networks☆20Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated 10 months ago
- ☆60Updated 3 years ago
- Implementation of Spectral State Space Models☆16Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆37Updated last year
- NanoGPT-speedrunning for the poor T4 enjoyers☆62Updated this week
- NLP with Rust for Python 🦀🐍☆62Updated 10 months ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆49Updated last year
- ☆9Updated 6 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated last month
- Make triton easier☆47Updated 10 months ago