yixiaoer / tpu-training-exampleView external linksLinks
☆16Jul 8, 2024Updated last year
Alternatives and similar repositories for tpu-training-example
Users that are interested in tpu-training-example are comparing it to the libraries listed below
Sorting:
- JAX implementation of the Mistral 7b v0.2 model☆35Jul 3, 2024Updated last year
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated 4 months ago
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated last year
- ☆15Oct 30, 2025Updated 3 months ago
- DiT (training + flow matching) in Jax☆11Jan 5, 2025Updated last year
- ☆32Jul 2, 2025Updated 7 months ago
- A Top-Down Profiler for GPU Applications☆22Feb 29, 2024Updated last year
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- ☆22May 4, 2021Updated 4 years ago
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- GPU Performance Advisor☆65Jul 25, 2022Updated 3 years ago
- Learning problem-solving, logic/set, math, physics, economics through functional programming using Haskell☆19Oct 16, 2015Updated 10 years ago
- An experimental communicating attention kernel based on DeepEP.☆35Jul 29, 2025Updated 6 months ago
- PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)☆66Mar 24, 2025Updated 10 months ago
- A library to extract plaintexts from the JSON dump file of namu wiki☆26Oct 6, 2022Updated 3 years ago
- ☆26Dec 3, 2025Updated 2 months ago
- Free ChatGPT API Key，免费ChatGPT API，支持GPT4 API，ChatGPT国内可用免费转发API，直连无需代理。☆13Aug 28, 2024Updated last year
- ☆33Nov 4, 2024Updated last year
- [ACL 2023] Code for ContraCLM: Contrastive Learning For Causal Language Model☆35Dec 20, 2023Updated 2 years ago
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 2 months ago
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- Repository for go shared libraries (for now).☆11Dec 1, 2025Updated 2 months ago
- 详细双语注释版word2vec源码，well-annotated word2vec☆10Oct 3, 2021Updated 4 years ago
- A TUI-based utility for real-time monitoring of InfiniBand traffic and performance metrics on the local node☆63Dec 19, 2025Updated last month
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆86Jul 28, 2024Updated last year
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆106Jun 28, 2025Updated 7 months ago
- Asynchronous Wiki Engine☆10Sep 4, 2023Updated 2 years ago
- A large-scale training and benchmarking framework for rPPG.☆10Nov 26, 2024Updated last year
- ☆14Jan 23, 2025Updated last year
- A flexible utility for converting tensor precision in PyTorch models and safetensors files, enabling efficient deployment across various …☆11Aug 24, 2023Updated 2 years ago
- Make Git monorepos like a boss.☆13Dec 18, 2025Updated last month
- 235,886 Words for Go☆12Nov 16, 2018Updated 7 years ago
- Visualizing 230 years of US Census data☆12Feb 23, 2020Updated 5 years ago
- seqax = sequence modeling + JAX☆170Jul 23, 2025Updated 6 months ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆407Jan 5, 2026Updated last month
- ☆114May 16, 2025Updated 8 months ago
- ☆52Jun 10, 2024Updated last year
- Neural likelihood-free methods in PyTorch.☆39Feb 11, 2020Updated 6 years ago
- Concurrent hash tries for C++ 14 with no memory management whatsoever.☆10Aug 30, 2016Updated 9 years ago