JAX implementation of configurable LLM distillation training
☆24Nov 15, 2025Updated 8 months ago
Alternatives and similar repositories for llm-distillation-jax
Users that are interested in llm-distillation-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆15Dec 4, 2025Updated 8 months ago
- Code release for "Understanding Bias in Large-Scale Visual Datasets"☆25Dec 4, 2024Updated last year
- Official repository for Parallax (Parameterized Local Linear Attention)☆68Jul 30, 2026Updated last week
- ☆20Nov 21, 2017Updated 8 years ago
- Implementation of our ACL2023 paper: Unifying Cross-Lingual and Cross-Modal Modeling Towards Weakly Supervised Multilingual Vision-Langua…☆19Jul 5, 2023Updated 3 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Lab Cookbook☆41Updated this week
- Joint Distillation for Fast Likelihood Evaluation and Sampling in Flow-based Models☆22Mar 5, 2026Updated 5 months ago
- (Personal project) Pruning algorithm for DNNs using "lottery ticket" pruning☆10Dec 8, 2022Updated 3 years ago
- Multilingual acoustic word embedding approaches applied and evaluated on GlobalPhone data.☆11Nov 3, 2020Updated 5 years ago
- This is a baseline for image restoration.☆14Mar 6, 2025Updated last year
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- FAQ for University of CaliforniaSanta Cruz 2019 Incoming Grads☆12Apr 4, 2019Updated 7 years ago
- Squeeze verbose LLM agent tool output down to only the relevant lines☆22Apr 27, 2026Updated 3 months ago
- NLP stuff with quantum computing☆17Nov 9, 2020Updated 5 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆12Jul 12, 2021Updated 5 years ago
- Fused KL divergence from hidden states for knowledge distillation