A JAX Research Toolkit for Visualizing, Manipulating, and Understanding Gemma Models with Multi-modal Support based on Penzai.
☆98Jan 13, 2026Updated 9 months ago
Alternatives and similar repositories for gemma_penzai
Users that are interested in gemma_penzai are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Training and evaluating with OpenReward☆33Apr 28, 2026Updated 5 months ago
- ☆13Aug 30, 2025Updated last year
- ☆28Dec 19, 2025Updated 10 months ago
- Programmable chat templates for LLM training and inference.☆163Updated this week
- ☆27Jan 22, 2026Updated 9 months 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.
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- Code for "Variational Reasoning for Language Models"☆63Sep 29, 2025Updated last year
- Analysis and investigating the confounding effect of accents in end-to-end Automatic Speech Recognition models.☆15Jun 27, 2020Updated 6 years ago
- Official implementation of Continual Harness (arxiv.org/abs/2605.09998) on ARC-AGI-3☆40Jul 3, 2026Updated 3 months ago
- Code for "Language Models Can Learn from Verbal Feedback Without Scalar Rewards"☆65Jan 5, 2026Updated 9 months ago
- ADAG: Transluce's MLP neuron-level circuit tracing library☆43Apr 10, 2026Updated 6 months ago
- Implemented and benchmarked KV cache compression methods for LLM inference in Triton. Optimized kernels for KIVI and TurboQuant with Flas…☆18Jun 19, 2026Updated 4 months ago
- Typescript implementation of Kinetix☆12Feb 25, 2025Updated last year
- ☆12Mar 7, 2022Updated 4 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Basic world models☆33Oct 30, 2025Updated 11 months ago
- A comprehensive AI & ML project portfolio from the University of Texas at Austin PG Program, demonstrating real-world data science and ma…☆18Jan 25, 2026Updated 8 months ago
- A markdown native slides tool for academics building with agents.☆348Updated this week
- Functional JAX middleware using a compute Node abstraction☆22Sep 6, 2026Updated last month
- ☆46Aug 3, 2026Updated 2 months ago
- Measuring how well CLI agents like Claude Code or Codex CLI can post-train base LLMs on a single H100 GPU in 10 hours☆602Oct 2, 2026Updated 2 weeks ago
- Official Code for MIMETIC^2☆13Nov 19, 2024Updated last year
- Generic building-block toolbox for training neural networks with adaptive and recursive execution. It provides reusable components to con…☆27Jun 29, 2026Updated 3 months ago
- ☆19Jun 27, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆44Dec 29, 2025Updated 9 months ago
- [NeurIPS 2026] When Reasoning Meets Its Laws☆38Jan 2, 2026Updated 9 months ago
- ☆15Feb 25, 2026Updated 7 months ago
- RePo: Language Models with Context Re-Positioning☆85Mar 30, 2026Updated 6 months ago
- ☆32Nov 4, 2024Updated last year
- ☆16Sep 24, 2026Updated 3 weeks ago
- Custom trained GRACE-OFFF models☆11Jun 2, 2026Updated 4 months ago
- ☆10Nov 16, 2023Updated 2 years ago
- Code publication to the paper "Normalized Attention Without Probability Cage"☆17Aug 24, 2026Updated last month
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Offical implementation of "MetaLA: Unified Optimal Linear Approximation to Softmax Attention Map" (NeurIPS2024 Oral)☆36Jan 18, 2025Updated last year
- minimal pytorch 4D parallelism☆74Feb 16, 2026Updated 8 months ago
- The official github repo for "Diffusion Language Models are Super Data Learners".☆228Nov 6, 2025Updated 11 months ago
- [ACCV'26] Revisiting Multi-Task Visual Representation Learning☆22Jan 21, 2026Updated 9 months ago
- Physics of Language Models: Part 4.2, Canon Layers at Scale where Synthetic Pretraining Resonates in Reality☆362Aug 21, 2026Updated 2 months ago
- A model for unsupervised morphological analysis that integrates orthographic and semantic views of words.☆13Oct 10, 2023Updated 3 years ago
- Script to convert all MP4 videos in a zip archive to JPG frames at a desired FPS with unique names. It will then retrain the top layers o…☆12Jul 6, 2016Updated 10 years ago