Repositories list

• ColossalAI

  Public
  Making large AI models cheaper, faster and more accessible

  aihpcdistributed-computing+ 9inferencebig-modellarge-scaledata-parallelismmodel-parallelismpipeline-parallelismfoundation-models+ 2

  Python

  •

  Apache License 2.0

  •4.5k•41k•430•44•UpdatedJul 10, 2025Jul 10, 2025

• public_assets

  Public
  Storing publicly available assets such as images, animations and texts

  Python

  •

  Apache License 2.0

  •18•13•0•0•UpdatedJun 10, 2025Jun 10, 2025

• ColossalAI-Documentation

  Public
  Documentation for Colossal-AI

  JavaScript

  •

  Apache License 2.0

  •12•23•4•1•UpdatedJun 6, 2025Jun 6, 2025

• Open-Sora

  Public
  Open-Sora: Democratizing Efficient Video Production for All

  Python

  •

  Apache License 2.0

  •2.6k•27k•2•2•UpdatedApr 30, 2025Apr 30, 2025

• Open-Sora-Demo

  Public
  0•6•0•0•UpdatedMar 19, 2025Mar 19, 2025

• Oh-My-Dockerfile

  Public
  A collection of dockerfiles for various tasks

  Dockerfile

  •

  Apache License 2.0

  •9•20•0•0•UpdatedFeb 20, 2025Feb 20, 2025

• ColossalAI-Platform-CLI

  Public
  CLI for ColossalAI Platform

  Python

  •

  Apache License 2.0

  •2•11•1•0•UpdatedFeb 20, 2025Feb 20, 2025

• TensorNVMe

  Public
  A Python library transfers PyTorch tensors between CPU and NVMe

  deep-learningpytorchnvme+ 1colossal-ai

  C++

  •25•116•8•0•UpdatedNov 27, 2024Nov 27, 2024

• transformers

  Public
  🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

  Python

  •

  Apache License 2.0

  •30k•11•0•0•UpdatedNov 19, 2024Nov 19, 2024

• LLaVA-NeXT

  Public
  Python

  •

  Apache License 2.0

  •378•1•0•0•UpdatedNov 18, 2024Nov 18, 2024

• graphrag

  Public
  A modular graph-based Retrieval-Augmented Generation (RAG) system

  Python

  •

  MIT License

  •2.7k•1•0•0•UpdatedAug 3, 2024Aug 3, 2024

• FastFold

  Public
  Optimizing AlphaFold Training and Inference on GPU Clusters

  gpuprotein-structureparallelism+ 6protein-foldingalphafold2evoformerhabana-gaudicudapytorch

  Python

  •

  Apache License 2.0

  •90•605•39•5•UpdatedJul 16, 2024Jul 16, 2024

• Cloud-Platform-Docs

  Public
  Documentation for our cloud platform

  JavaScript

  •3•1•0•1•UpdatedApr 3, 2024Apr 3, 2024

• CANN-Installer

  Public
  This repository contains Huawei Ascend CANN files

  Apache License 2.0

  •1•1•1•0•UpdatedFeb 27, 2024Feb 27, 2024

• SwiftInfer

  Public
  Efficient AI Inference & Serving

  deep-learninginferenceartificial-intelligence+ 5llamagptllm-servingllm-inferencellama2

  Python

  •

  Apache License 2.0

  •29•471•3•0•UpdatedJan 8, 2024Jan 8, 2024

• EnergonAI

  Public archive
  Large-scale model inference.

  Python

  •

  Apache License 2.0

  •88•630•40•2•UpdatedSep 12, 2023Sep 12, 2023

• pytest-testmon

  Public
  Selects tests affected by changed files. Executes the right tests first. Continuous test runner when used with pytest-watch.

  Python

  •

  GNU Affero General Public License v3.0

  •64•0•0•0•UpdatedJul 25, 2023Jul 25, 2023

• Elixir

  Public
  Elixir: Train a Large Language Model on a Small GPU Cluster

  efficientmemory-managementlarge-language-models

  Python

  •5•15•0•1•UpdatedJun 8, 2023Jun 8, 2023

• ColossalAI-Examples

  Public archive
  Examples of training models with hybrid parallelism using ColossalAI

  Python

  •

  Apache License 2.0

  •103•340•29•2•UpdatedMar 23, 2023Mar 23, 2023

• mmdetection-examples

  Public archive
  Train mmdetection models with ColossalAI.

  Python

  •0•2•1•0•UpdatedFeb 18, 2023Feb 18, 2023

• PaLM-colossalai

  Public archive
  Scalable PaLM implementation of PyTorch

  Python

  •

  Apache License 2.0

  •27•190•7•2•UpdatedDec 19, 2022Dec 19, 2022

• GPT-Demo

  Public archive
  GPT Demo with hybrid distributed training

  Python

  •

  Apache License 2.0

  •7•10•2•0•UpdatedDec 1, 2022Dec 1, 2022

• Titans

  Public archive
  A collection of models built with ColossalAI

  Python

  •

  Apache License 2.0

  •16•32•6•1•UpdatedNov 22, 2022Nov 22, 2022

• ColossalAI-Pytorch-lightning

  Public
  Python

  •6•24•0•0•UpdatedNov 22, 2022Nov 22, 2022

• OPT-Benchmark

  Public archive
  Python

  •

  Apache License 2.0

  •5•4•1•0•UpdatedNov 22, 2022Nov 22, 2022

• CachedEmbedding

  Public
  A memory efficient DLRM training solution using ColossalAI

  nlpembeddingsdlrm+ 4recommandation-systemcolossal-aideep-learningpytorch

  Python

  •

  Apache License 2.0

  •14•105•2•0•UpdatedNov 22, 2022Nov 22, 2022

• SkyComputing

  Public archive
  Sky Computing: Accelerating Geo-distributed Computing in Federated Learning

  Python

  •

  Apache License 2.0

  •21•91•1•0•UpdatedNov 22, 2022Nov 22, 2022

• torchrec

  Public
  Pytorch domain library for recommendation systems

  Python

  •

  BSD 3-Clause "New" or "Revised" License

  •538•1•0•0•UpdatedOct 21, 2022Oct 21, 2022

• ColossalAI-Benchmark

  Public archive
  Performance benchmarking with ColossalAI

  Python

  •

  Apache License 2.0

  •16•39•5•0•UpdatedJul 6, 2022Jul 6, 2022