zanussbaum / surfgrad
webgpu autograd library
☆19Updated 3 weeks ago
Alternatives and similar repositories for surfgrad:
Users that are interested in surfgrad are comparing it to the libraries listed below
- ☆37Updated 5 months ago
- ☆27Updated 6 months ago
- realtime latent world model inference demo☆38Updated 2 months ago
- Focused on fast experimentation and simplicity☆64Updated 3 weeks ago
- DeMo: Decoupled Momentum Optimization☆170Updated last month
- Cerule - A Tiny Mighty Vision Model☆67Updated 4 months ago
- look how they massacred my boy☆63Updated 3 months ago
- Training Models Daily☆17Updated last year
- A synthetic story narration dataset to study small audio LMs.☆31Updated 11 months ago
- Fast approximate inference on a single GPU with sparsity aware offloading☆38Updated last year
- Latent Large Language Models☆17Updated 4 months ago
- ☆49Updated 4 months ago
- working implimention of deepseek MLA☆23Updated last week
- Aana SDK is a powerful framework for building AI enabled multimodal applications.☆36Updated this week
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆60Updated 2 months ago
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆14Updated 2 weeks ago
- Collection of autoregressive model implementation☆76Updated last week
- ☆25Updated last month
- ☆22Updated 2 months ago
- ☆96Updated 3 months ago
- A streamlined implementation of Grounding DINO and SAM for advanced image segmentation. This lightweight solution simplifies the integrat…☆63Updated 3 months ago
- BH hackathon☆14Updated 9 months ago
- implementation of https://arxiv.org/pdf/2312.09299☆20Updated 6 months ago
- alternative way to calculating self attention☆18Updated 7 months ago
- ☆30Updated 3 months ago
- A collection of optimizers for MLX☆25Updated this week
- ☆20Updated 2 months ago
- Jax like function transformation engine but micro, microjax☆30Updated 2 months ago
- Video+code lecture on building nanoGPT from scratch☆65Updated 7 months ago
- Gpu benchmark☆50Updated 3 months ago