Pytorch/XLA SPMD Test code in Google TPU
☆23Apr 3, 2024Updated last year
Alternatives and similar repositories for torch-xla-SPMD
Users that are interested in torch-xla-SPMD are comparing it to the libraries listed below
Sorting:
- See details in https://github.com/pytorch/xla/blob/r1.12/torch_xla/distributed/fsdp/README.md☆25Dec 22, 2022Updated 3 years ago
- A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/…☆34Mar 4, 2025Updated last year
- The repository for the course "Astroinformatics" offered at Institute of Astronomy, National Central University, from Sep/2022 to Jan/202…☆10Jun 4, 2024Updated last year
- YouTube Assistant☆12May 15, 2023Updated 2 years ago
- This repository provides a small Python wrapper for the Matlab tool SNR Eval provided by Labrosa: https://labrosa.ee.columbia.edu/project…☆12Jun 22, 2022Updated 3 years ago
- 4-bit Shampoo for Memory-Efficient Network Training (NeurIPS 2024)☆13Feb 13, 2025Updated last year
- ☆14Mar 8, 2025Updated last year
- OpenAI ROS☆12Mar 7, 2019Updated 7 years ago
- ☆13Jun 22, 2025Updated 8 months ago
- 用Paddle复现Recipes for building an open-domain chatbot论文☆11Nov 1, 2021Updated 4 years ago
- ☆10Dec 21, 2024Updated last year
- 0-Shot Tokenizer Transplant☆14May 16, 2025Updated 9 months ago
- The official implementation of Bi-Mamba☆14Oct 22, 2025Updated 4 months ago
- MySQL Query MCP server for AI assistants - execute read-only MySQL queries☆12May 31, 2025Updated 9 months ago
- Transformer from scratch with einsum method☆11Jul 8, 2021Updated 4 years ago
- See https://github.com/cuda-mode/triton-index/ instead!☆11May 8, 2024Updated last year
- ☆11Feb 9, 2024Updated 2 years ago
- A real-time visual analysis software for depth tracking in noisy oil and gas environments.☆12Jan 13, 2026Updated last month
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆339Updated this week
- Hinton's Forward-Forward Algorithm for Deep Learning☆10Feb 6, 2023Updated 3 years ago
- KANs and MLPs☆12Jun 7, 2024Updated last year
- GPT-J 6B inference on TensorRT with INT-8 precision☆11Apr 5, 2023Updated 2 years ago
- Perf monitoring CLI tool for Apple Silicon☆10Jan 25, 2023Updated 3 years ago
- ☆13Dec 15, 2025Updated 2 months ago
- An implementation of the AlphaZero algorithm by Google Deepmind. Research paper here: https://arxiv.org/abs/1911.08265☆12Oct 10, 2024Updated last year
- ☆12Nov 15, 2022Updated 3 years ago
- Example implementation of Zeebe workflows using pyzeebe.☆12Jun 1, 2021Updated 4 years ago
- ☆13Jun 3, 2023Updated 2 years ago
- ☆14Dec 9, 2021Updated 4 years ago
- ☆13May 21, 2024Updated last year
- Extremely simple MoE implementation, mostly based off Switch Transformer☆13Feb 26, 2024Updated 2 years ago
- This project is the official implementation of our ACM MM 2024 paper, OmniStitch: Depth-aware Stitching Framework for Omnidirectional Vis…☆17Aug 5, 2024Updated last year
- Examples of Debug Adapter Protocol (DAP) implementation☆13Jun 22, 2021Updated 4 years ago
- Implementation of True Online TD(lambda) with a Fourier Basis function approximator.☆13May 9, 2015Updated 10 years ago
- LLM Proxy☆12Aug 26, 2024Updated last year
- Official implementation for "PEAC: Unsupervised Pre-training for Cross-Embodiment Reinforcement Learning" (NeurIPS 2024)☆18Oct 13, 2024Updated last year
- An implementation of DecorrelatedBN by tensorflow☆13Jun 30, 2022Updated 3 years ago
- Reimplementation of https://github.com/montemac/algebraic_value_editing in pure PyTorch for efficiency on large models☆11Jun 28, 2023Updated 2 years ago