zanussbaum / surfgradLinks
webgpu autograd library
☆33Updated 8 months ago
Alternatives and similar repositories for surfgrad
Users that are interested in surfgrad are comparing it to the libraries listed below
Sorting:
- MLX port for xjdr's entropix sampler (mimics jax implementation)☆61Updated last year
- look how they massacred my boy☆63Updated last year
- Repository to create traveling waves integrate special information through time☆56Updated 5 months ago
- A tree-based prefix cache library that allows rapid creation of looms: hierarchal branching pathways of LLM generations.☆77Updated 11 months ago
- Pytorch script hot swap: Change code without unloading your LLM from VRAM☆125Updated 9 months ago
- ☆40Updated last year
- DeMo: Decoupled Momentum Optimization☆198Updated last year
- Official CLI and Python SDK for Prime Intellect - access GPU compute, remote sandboxes, RL environments, and distributed training infrast…☆143Updated this week
- implementation of https://arxiv.org/pdf/2312.09299☆21Updated last year
- Lego for GRPO☆30Updated 8 months ago
- A graph visualization of attention☆57Updated 8 months ago
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆109Updated 10 months ago
- Samples of good AI generated CUDA kernels☆99Updated 8 months ago
- ☆34Updated 10 months ago
- TensorRT-LLM server with Structured Outputs (JSON) built with Rust☆65Updated 9 months ago
- This repository has code for fine-tuning LLMs with GRPO specifically for Rust Programming using cargo as feedback☆114Updated 10 months ago
- Optimizing Causal LMs through GRPO with weighted reward functions and automated hyperparameter tuning using Optuna☆59Updated 3 months ago
- Plotting (entropy, varentropy) for small LMs☆99Updated 8 months ago
- Simple high-throughput inference library☆155Updated 8 months ago
- SIMD quantization kernels☆94Updated 4 months ago
- Pivotal Token Search☆144Updated last month
- ☆50Updated 3 months ago
- ☆27Updated last year
- PyTorch implementation of models from the Zamba2 series.☆186Updated last year
- Approximating the joint distribution of language models via MCTS☆22Updated last year
- RWKV-7: Surpassing GPT☆104Updated last year
- ☆90Updated last year
- EXO Gym is an open-source Python toolkit that facilitates distributed AI research.☆94Updated last month
- Turing machines, Rule 110, and A::B reversal using Claude 3 Opus.☆58Updated last year
- Efficient non-uniform quantization with GPTQ for GGUF☆58Updated 4 months ago