Alternatives and similar repositories for torchprime

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

• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆60Updated last month
• AI-Hypercomputer / kithara
  ☆13Updated last month
• mgmalek / efficient_cross_entropy
  ☆106Updated 11 months ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆109Updated 3 months ago
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated last year
• cat-state / tinypar
  ☆20Updated last year
• HeegyuKim / torch-xla-SPMD
  Pytorch/XLA SPMD Test code in Google TPU
  ☆23Updated 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…
  ☆245Updated this week
• google / praxis
  ☆186Updated 2 weeks ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆119Updated last week
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆37Updated 2 years ago
• google / saxml
  ☆138Updated 2 weeks ago
• IST-DASLab / peft-rosa
  A fork of the PEFT library, supporting Robust Adaptation (RoSA)
  ☆14Updated 9 months ago
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆224Updated 9 months ago
• bminixhofer / zett
  Code for Zero-Shot Tokenizer Transfer
  ☆127Updated 4 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆125Updated 5 months ago
• warner-benjamin / optimi
  Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers
  ☆92Updated 10 months ago
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆265Updated 2 weeks ago
• pytorch-tpu / transformers
  🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
  ☆17Updated last month
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆118Updated 11 months ago
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆218Updated last year
• JonasGeiping / linear_cross_entropy_loss
  A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
  ☆67Updated 9 months ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆195Updated this week
• mobiusml / gemlite
  Fast low-bit matmul kernels in Triton
  ☆299Updated this week
• cloneofsimo / min-fsdp
  ☆79Updated 10 months ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆204Updated 3 weeks ago
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆85Updated 3 weeks ago
• imoneoi / multipack_sampler
  Multipack distributed sampler for fast padding-free training of LLMs
  ☆189Updated 9 months ago
• MekkCyber / CutlassAcademy
  A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS
  ☆173Updated last week
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆221Updated 4 months ago