☆16Jul 8, 2024Updated 2 years ago
Alternatives and similar repositories for tpu-training-example
Users that are interested in tpu-training-example are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- JAX implementation of the Mistral 7b v0.2 model☆35Jul 3, 2024Updated 2 years ago
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated 2 years ago
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated last year
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated 2 years ago
- DiT (training + flow matching) in Jax☆12Jan 5, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Reduce kernel based on CUTLASS CuTe and TMA.☆12Sep 25, 2025Updated last year
- ☆10Jun 8, 2024Updated 2 years ago
- JAX implementation of the T5 model: Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer☆24Jun 10, 2023Updated 3 years ago
- 苏州大学本科毕业论文LaTeX模板☆12May 27, 2020Updated 6 years ago
- 🎼 Claude Code skill: an unattended nightly loop that turns Linear tickets into reviewed PRs☆12Oct 23, 2026Updated last week
- ESM2 protein language models in JAX/Flax☆18Oct 10, 2022Updated 4 years ago
- Graph Networks with Jax☆11Mar 23, 2022Updated 4 years ago
- High-performance large-scale embedding acceleration for JAX on Google TPU SparseCores.☆38Updated this week
- Simple Audio Embedding Toolkit☆12Aug 9, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- So, I trained a MoE based a 124M (8x12M) architecture I coded from ground up to build a small instruct model, going through the below-men…☆14Jul 18, 2025Updated last year
- MATLAB code to replicate Koop and Korobilis (2014) A new index of financial conditions. European Economic Review☆23Jun 4, 2025Updated last year
- Triton kernels for Flux☆23Jul 7, 2025Updated last year
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- Neural likelihood-free methods in PyTorch.☆39Feb 11, 2020Updated 6 years ago
- Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning☆25May 12, 2025Updated last year
- ☆23May 5, 2025Updated last year
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated last year
- GPU Performance Advisor☆66Jul 25, 2022Updated 4 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- A Top-Down Profiler for GPU Applications☆24Feb 29, 2024Updated 2 years ago
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆16Jun 24, 2023Updated 3 years ago
- ☆14Aug 8, 2023Updated 3 years ago
- supporting pytorch FSDP for optimizers☆85Dec 8, 2024Updated last year
- Paper introducing jax-cosmo☆13Apr 27, 2023Updated 3 years ago
- JAX implementation of the Llama 2 model☆217Feb 2, 2024Updated 2 years ago
- Everything you want to know about Google Cloud TPU☆573Jul 16, 2024Updated 2 years ago
- 粵語 Emoji☆13Jun 25, 2026Updated 4 months ago
- PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)☆67Mar 24, 2025Updated last year
- 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.
- A lightweight open-source package to fine-tune embedding models.☆22Feb 4, 2024Updated 2 years ago
- Code for the paper "Towards Reliable Simulation-Based Inference with Balanced Neural Ratio Estimation".☆15Nov 14, 2022Updated 3 years ago
- Online BaseHangul Encoder And Decoder☆13Jan 30, 2023Updated 3 years ago
- Loengfan (粵語兩分) is the Cantonese version of the Liang Fen input method☆16Mar 3, 2022Updated 4 years ago
- Generates Graphviz image URL that can be used directly on any website without a need to host them on a server☆15Feb 21, 2026Updated 8 months ago
- Run Jupyter Lab as a desktop application☆14Jun 12, 2020Updated 6 years ago
- talk to sarvam APIs directly, without code.☆12Jun 9, 2025Updated last year