HeegyuKim / torch-xla-SPMDView external linksLinks
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
- (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX☆29Feb 2, 2026Updated 2 weeks 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 11 months ago
- 트랜스포머 블록을 활용한 상품명 자연어처리 기반 카테고리 분류 모델☆10Dec 5, 2022Updated 3 years ago
- ML AI Case Studies and Projects☆10Jun 11, 2020Updated 5 years ago
- YouTube Assistant☆12May 15, 2023Updated 2 years ago
- Transformer from scratch with einsum method☆11Jul 8, 2021Updated 4 years ago
- PhoneGap NFC peer to peer demo☆22Jan 6, 2017Updated 9 years ago
- A real-time visual analysis software for depth tracking in noisy oil and gas environments.☆12Jan 13, 2026Updated last month
- OpenAI ROS☆12Mar 7, 2019Updated 6 years ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- An API wrapper for @convertio (convertio.co) written in Python.☆12Jun 30, 2025Updated 7 months ago
- ☆11Feb 9, 2024Updated 2 years ago
- The official implementation of Bi-Mamba☆14Oct 22, 2025Updated 3 months ago
- ☆14Mar 8, 2025Updated 11 months ago
- 4-bit Shampoo for Memory-Efficient Network Training (NeurIPS 2024)☆13Feb 13, 2025Updated last year
- See https://github.com/cuda-mode/triton-index/ instead!☆11May 8, 2024Updated last year
- A common protocol for AI agent tools☆10Oct 21, 2024Updated last year
- ☆10Dec 21, 2024Updated last year
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆337Updated this week
- Extremely simple MoE implementation, mostly based off Switch Transformer☆13Feb 26, 2024Updated last year
- An implementation of the AlphaZero algorithm by Google Deepmind. Research paper here: https://arxiv.org/abs/1911.08265☆12Oct 10, 2024Updated last year
- Perf monitoring CLI tool for Apple Silicon☆11Jan 25, 2023Updated 3 years ago
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Nov 23, 2023Updated 2 years ago
- Predict scalar coupling in molecules☆15Mar 14, 2021Updated 4 years ago
- ☆13May 21, 2024Updated last year
- ☆13Jun 3, 2023Updated 2 years ago
- GRPC client CLI, like grpcurl, but in Rust; GRPC Client UI, like postman, but in Rust☆20Jan 29, 2026Updated 2 weeks ago
- Patch-based inpainting Python library☆12Mar 5, 2023Updated 2 years ago
- ☆14Dec 9, 2021Updated 4 years ago
- Hinton's Forward-Forward Algorithm for Deep Learning☆10Feb 6, 2023Updated 3 years ago
- Docker container for ocropus3 OCR system☆12Aug 19, 2018Updated 7 years ago
- GPT-J 6B inference on TensorRT with INT-8 precision☆11Apr 5, 2023Updated 2 years ago
- KANs and MLPs☆12Jun 7, 2024Updated last year
- A custom Docker image containing the theia-ide for Java development☆15Nov 30, 2021Updated 4 years ago
- Example implementation of Zeebe workflows using pyzeebe.☆12Jun 1, 2021Updated 4 years ago
- NMT with ssp☆11Oct 28, 2021Updated 4 years ago
- Official implementation for "PEAC: Unsupervised Pre-training for Cross-Embodiment Reinforcement Learning" (NeurIPS 2024)☆18Oct 13, 2024Updated last year
- sound stretch python module☆11May 1, 2019Updated 6 years ago