zanussbaum / surfgrad
webgpu autograd library
☆22Updated 4 months ago
Alternatives and similar repositories for surfgrad:
Users that are interested in surfgrad are comparing it to the libraries listed below
- look how they massacred my boy☆63Updated 6 months ago
- ☆38Updated 9 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆68Updated 2 months ago
- GPU accelerated client-side embeddings for vector search, RAG etc.☆66Updated last year
- This repository has code for fine-tuning LLMs with GRPO specifically for Rust Programming using cargo as feedback☆80Updated last month
- PyTorch implementation of models from the Zamba2 series.☆179Updated 3 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆96Updated last month
- ☆27Updated 9 months ago
- Pytorch script hot swap: Change code without unloading your LLM from VRAM☆112Updated last week
- ☆89Updated 6 months ago
- Lego for GRPO☆27Updated 3 weeks ago
- Latent Large Language Models☆18Updated 8 months ago
- ☆27Updated 2 weeks ago
- ☆129Updated 8 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆39Updated 2 months ago
- Inference Llama 2 in one file of zero-dependency, zero-unsafe Rust☆38Updated last year
- ☆97Updated 6 months ago
- Cerule - A Tiny Mighty Vision Model☆67Updated 7 months ago
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆64Updated 5 months ago
- Repository to create traveling waves integrate special information through time☆50Updated last month
- smolLM with Entropix sampler on pytorch☆151Updated 5 months ago
- DeMo: Decoupled Momentum Optimization☆186Updated 4 months ago
- Modify Entropy Based Sampling to work with Mac Silicon via MLX☆50Updated 5 months ago
- ANE accelerated embedding models!☆16Updated 4 months ago
- ☆26Updated 4 months ago
- A simple MLX implementation for pretraining LLMs on Apple Silicon.☆28Updated 3 months ago
- ☆33Updated 9 months ago
- ☆30Updated 6 months ago
- An easy-to-understand framework for LLM samplers that rewind and revise generated tokens☆139Updated 2 months ago
- ☆71Updated this week