Alternatives and similar repositories for torchprime

Users that are interested in torchprime are comparing it to the libraries listed below

• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆112Updated 4 months ago
• google / saxml
  ☆138Updated 2 weeks ago
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆60Updated 2 months ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆144Updated this week
• mgmalek / efficient_cross_entropy
  ☆108Updated last year
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆37Updated 2 years ago
• HeegyuKim / torch-xla-SPMD
  Pytorch/XLA SPMD Test code in Google TPU
  ☆23Updated last year
• AI-Hypercomputer / kithara
  ☆13Updated 2 weeks ago
• catie-aq / flashT5
  A fast implementation of T5/UL2 in PyTorch using Flash Attention
  ☆105Updated 2 months ago
• cat-state / tinypar
  ☆20Updated last year
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆271Updated last week
• MekkCyber / CutlassAcademy
  A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS
  ☆183Updated last month
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆185Updated this week
• erfanzar / EasyDeL
  Accelerate, Optimize performance with streamlined training and serving options with JAX.
  ☆274Updated this week
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated last year
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆249Updated this week
• yixiaoer / tpux
  A set of Python scripts that makes your experience on TPU better
  ☆54Updated 11 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆223Updated 10 months ago
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆229Updated 4 months ago
• hkproj / triton-flash-attention
  ☆169Updated 5 months ago
• google / praxis
  ☆186Updated this week
• warner-benjamin / optimi
  Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers
  ☆94Updated 10 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆215Updated this week
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆118Updated last year
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆125Updated 2 months ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆212Updated last year
• gpu-mode / profiling-cuda-in-torch
  ☆157Updated last year
• huggingface / picotron_tutorial
  ☆190Updated 3 months ago
• NVIDIA / NeMo-Run
  A tool to configure, launch and manage your machine learning experiments.
  ☆153Updated this week
• gau-nernst / learn-cuda
  Learn CUDA with PyTorch
  ☆21Updated this week