Alternatives and similar repositories for optimum-graphcore

Users that are interested in optimum-graphcore are comparing it to the libraries listed below

• huggingface / bloom-jax-inference
  ☆66Updated 3 years ago
• EleutherAI / DeeperSpeed
  DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
  ☆171Updated 2 months ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆223Updated last year
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆216Updated last year
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆189Updated 3 years ago
• huggingface / optimum-habana
  Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)
  ☆201Updated last week
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆120Updated last year
• google / praxis
  ☆190Updated 3 weeks ago
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆87Updated last year
• sholtodouglas / scalingExperiments
  ☆62Updated 3 years ago
• huggingface / fuego
  [WIP] A 🔥 interface for running code in the cloud
  ☆86Updated 2 years ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆210Updated 2 years ago
• sgugger / torchdynamo-tests
  ☆19Updated 3 years ago
• graphcore / tutorials
  Training material for IPU users: tutorials, feature examples, simple applications
  ☆87Updated 2 years ago
• microsoft / varuna
  ☆252Updated last year
• google / aqt
  ☆337Updated 2 weeks ago
• google / flaxformer
  ☆364Updated last year
• SeanNaren / min-LLM
  Minimal code to train a Large Language Model (LLM).
  ☆172Updated 3 years ago
• huggingface / transformers_bloom_parallel
  Techniques used to run BLOOM at inference in parallel
  ☆37Updated 3 years ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆277Updated 3 years ago
• fpgaminer / GPTQ-triton
  GPTQ inference Triton kernel
  ☆316Updated 2 years ago
• HazyResearch / H3
  Language Modeling with the H3 State Space Model
  ☆521Updated 2 years ago
• rom1504 / cc2dataset
  Easily convert common crawl to a dataset of caption and document. Image/text Audio/text Video/text, ...
  ☆322Updated 2 years ago
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆242Updated 2 years ago
• huggingface / optimum-benchmark
  🏋️ A unified multi-backend utility for benchmarking Transformers, Timm, PEFT, Diffusers and Sentence-Transformers with full support of O…
  ☆320Updated 2 months ago
• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆161Updated 2 months ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆124Updated 10 months ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• meta-pytorch / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆226Updated last year
• IST-DASLab / qmoe
  Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".
  ☆278Updated 2 years ago