A JAX Research Toolkit for Visualizing, Manipulating, and Understanding Gemma Models with Multi-modal Support based on Penzai.
☆99Jan 13, 2026Updated 7 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 4 months ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- Programmable chat templates for LLM training and inference.☆155Updated this week
- ADAG: Transluce's MLP neuron-level circuit tracing library☆37Apr 10, 2026Updated 4 months ago
- Code for "Language Models Can Learn from Verbal Feedback Without Scalar Rewards"☆65Jan 5, 2026Updated 8 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.
- Official implementation of "Data Mixture Inference: What do BPE tokenizers reveal about their training data?"☆23May 15, 2025Updated last year
- Fluid Language Model Benchmarking☆29Sep 16, 2025Updated 11 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☆548Updated this week
- ☆52Apr 12, 2026Updated 4 months ago
- This is an implementation of the paper "Are We Done with Object-Centric Learning?"☆14Jun 21, 2026Updated 2 months ago
- ☆12Mar 7, 2022Updated 4 years ago
- Basic world models☆33Oct 30, 2025Updated 10 months ago
- ☆15Feb 25, 2026Updated 6 months ago
- Generic building-block toolbox for training neural networks with adaptive and recursive execution. It provides reusable components to con…☆27Jun 29, 2026Updated 2 months ago
- 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.
- ☆27Jan 22, 2026Updated 7 months ago
- RePo: Language Models with Context Re-Positioning☆84Mar 30, 2026Updated 5 months ago
- When Reasoning Meets Its Laws☆38Jan 2, 2026Updated 8 months ago
- This repository contains the code for the paper "TaylorShift: Shifting the Complexity of Self-Attention from Squared to Linear (and Back)…☆15Feb 25, 2026Updated 6 months ago
- Physics of Language Models: Part 4.2, Canon Layers at Scale where Synthetic Pretraining Resonates in Reality☆361Aug 21, 2026Updated 2 weeks ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆93Oct 30, 2024Updated last year
- Attention visualization in CLIP☆17Dec 7, 2022Updated 3 years ago
- The official github repo for "Diffusion Language Models are Super Data Learners".☆227Nov 6, 2025Updated 10 months ago
- ☆15Dec 3, 2022Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- minimal pytorch 4D parallelism☆73Feb 16, 2026Updated 6 months ago
- The product documentation for Weights & Biases, published at https://docs.wandb.ai☆25Updated this week
- FID computation in Jax/Flax.☆29Jul 17, 2024Updated 2 years ago
- ☆32Nov 4, 2024Updated last year
- Revisiting Multi-Task Visual Representation Learning☆22Jan 21, 2026Updated 7 months ago
- a whirlwind tour to deep learning and deep learning systems☆83Updated this week
- A model for unsupervised morphological analysis that integrates orthographic and semantic views of words.☆13Oct 10, 2023Updated 2 years ago
- The State Of The Art, intelligence☆165Aug 12, 2025Updated last year
- Code for the experiments and websites of the paper "Same Task, Different Circuits"☆37Jul 21, 2026Updated last month
- 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.
- [ACL 2025] Analyzing LLMs' Multilingual Knowledge Boundary Cognition Across Languages Through the Lens of Internal Representations☆19Oct 18, 2025Updated 10 months ago
- ☆17Jul 31, 2025Updated last year
- Code and data related to "Efficient, Compositional, Order-Sensitive n-gram Embeddings" (EACL 2017)☆15Apr 6, 2017Updated 9 years ago
- ☆68Aug 16, 2024Updated 2 years ago
- OpenVision (ICCV 2025), OpenVision 2 (CVPR 2026), and OpenVision 3☆495Aug 22, 2026Updated 2 weeks ago
- 「Low-Rank Decay」的官方实现。☆19Nov 25, 2025Updated 9 months ago
- Code repository dedicated to experimenting and research with tiny reasoning language model☆52Nov 24, 2025Updated 9 months ago