huggingface / optimum-graphcore

Related projects ⓘ

Alternatives and complementary repositories for optimum-graphcore

• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆113Updated 6 months ago
• huggingface / bloom-jax-inference
  ☆64Updated 2 years ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆248Updated 2 years ago
• sholtodouglas / scalingExperiments
  ☆57Updated 2 years ago
• google / praxis
  ☆177Updated last week
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆210Updated 9 months ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆249Updated this week
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆197Updated 8 months ago
• huggingface / optimum-habana
  Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)
  ☆153Updated this week
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆195Updated last year
• pytorch-labs / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆168Updated 3 weeks ago
• huggingface / optimum-tpu
  Google TPU optimizations for transformers models
  ☆75Updated this week
• microsoft / varuna
  ☆237Updated 3 months ago
• mgmalek / efficient_cross_entropy
  ☆77Updated 5 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆105Updated 2 weeks ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆194Updated this week
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆212Updated 3 months ago
• yandex-research / DeDLOC
  Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)
  ☆116Updated 2 years ago
• google / aqt
  ☆268Updated this week
• google / flaxformer
  ☆334Updated 7 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 10 months ago
• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆33Updated 2 years ago
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆80Updated 11 months ago
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆146Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆344Updated this week
• apple / ml-recurrent-drafter
  ☆101Updated last month
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆77Updated 2 years ago
• HanGuo97 / flute
  Fast Matrix Multiplications for Lookup Table-Quantized LLMs
  ☆187Updated this week
• huggingface / fuego
  [WIP] A 🔥 interface for running code in the cloud
  ☆86Updated last year
• graphcore / tutorials
  Training material for IPU users: tutorials, feature examples, simple applications
  ☆87Updated last year