Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.
☆24Sep 29, 2024Updated last year
Alternatives and similar repositories for midGPT
Users that are interested in midGPT are comparing it to the libraries listed below
Sorting:
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- ☆13Dec 15, 2025Updated 2 months ago
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆21Sep 24, 2025Updated 5 months ago
- Digital Trigger Unit for V2&V3 Gearbox☆15Jan 11, 2021Updated 5 years ago
- Pure Triton kernels for Qwen3.5-27B inference on NVIDIA B200☆66Updated this week
- ☆44Feb 27, 2026Updated last week
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- ☆27Jul 9, 2024Updated last year
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆158Nov 11, 2025Updated 3 months ago
- 详细双语注释版word2vec源码，well-annotated word2vec☆10Oct 3, 2021Updated 4 years ago
- 💧 Inland water systems are essential to our environment because they are vital ecosystems that are bio-diverse. Thus, finding innovative…☆11Apr 26, 2020Updated 5 years ago
- ☆28Dec 3, 2025Updated 3 months ago
- ☆12Jan 10, 2026Updated last month
- LLVM Essentials 中文版☆12Feb 18, 2025Updated last year
- YouTube Assistant☆12May 15, 2023Updated 2 years ago
- Make triton easier☆50Jun 12, 2024Updated last year
- ☆14Feb 11, 2026Updated 3 weeks ago
- Persistent dense gemm for Hopper in `CuTeDSL`☆15Aug 9, 2025Updated 6 months ago
- ☆18Jun 6, 2025Updated 9 months ago
- The official implementation of Bi-Mamba☆14Oct 22, 2025Updated 4 months ago
- Code for reproducing the results in "Mining Semantic Affordances of Visual Object Categories"☆12Jun 10, 2024Updated last year
- Gazebo plugins for running Orocos RTT components in the gazebo process.☆12Jul 28, 2016Updated 9 years ago
- The Spezi FHIR Standard & Related Modules☆13Jan 11, 2026Updated last month
- Implementation of Reinforce for educational purposes.☆12Jun 12, 2023Updated 2 years ago
- ☆14Mar 8, 2025Updated 11 months ago
- ☆10Dec 18, 2023Updated 2 years ago
- Speeding Up Your Python Codes 1000x☆12Apr 2, 2025Updated 11 months ago
- Collection of simple General Matrix Multiplication - GEMM implementations☆13Feb 26, 2024Updated 2 years ago
- Fast, simple, cryptographically strong random numbers in C++. Experimental.☆19Dec 12, 2013Updated 12 years ago
- ☆23Jul 11, 2025Updated 7 months ago
- an implementation of paper"Retentive Network: A Successor to Transformer for Large Language Models" https://arxiv.org/pdf/2307.08621.pdf☆11Jul 25, 2023Updated 2 years ago
- BigBang-Proton is a LLM pretrained on cross-scale, cross-structure, cross-discipline real-world scientific tasks to construct a scienti…☆22Nov 8, 2025Updated 3 months ago
- Large language models to diffusion finetuning code☆24Jun 2, 2025Updated 9 months ago
- ☆13Jan 7, 2025Updated last year
- Pytorch routines for (Ker)nel (Mac)hines☆11Oct 10, 2025Updated 4 months ago
- Unofficial implementation of paper : Exploring the Space of Key-Value-Query Models with Intention☆12May 24, 2023Updated 2 years ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 6 months ago
- Code for "What really matters in matrix-whitening optimizers?"☆22Oct 31, 2025Updated 4 months ago