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
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆188Updated 3 years ago
• EleutherAI / DeeperSpeed
  DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
  ☆170Updated last month
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆215Updated last year
• huggingface / optimum-habana
  Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)
  ☆201Updated last week
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆87Updated last year
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆220Updated last year
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆119Updated last year
• google / praxis
  ☆190Updated 3 weeks ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆210Updated 2 years ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆276Updated 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 / flaxformer
  ☆363Updated last year
• sholtodouglas / scalingExperiments
  ☆62Updated 3 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• r-three / git-theta
  git extension for {collaborative, communal, continual} model development
  ☆215Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• huggingface / fuego
  [WIP] A 🔥 interface for running code in the cloud
  ☆85Updated 2 years ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆122Updated 9 months 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
• sgugger / torchdynamo-tests
  ☆19Updated 2 years ago
• rom1504 / cc2dataset
  Easily convert common crawl to a dataset of caption and document. Image/text Audio/text Video/text, ...
  ☆320Updated last year
• SeanNaren / min-LLM
  Minimal code to train a Large Language Model (LLM).
  ☆172Updated 3 years ago
• google / aqt
  ☆337Updated last week
• huggingface / transformers_bloom_parallel
  Techniques used to run BLOOM at inference in parallel
  ☆37Updated 3 years ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 3 years ago
• llm-efficiency-challenge / neurips_llm_efficiency_challenge
  NeurIPS Large Language Model Efficiency Challenge: 1 LLM + 1GPU + 1Day
  ☆256Updated 2 years ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆433Updated last month
• rasbt / pytorch-memory-optim
  This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…
  ☆91Updated 2 years ago