Skip to content

@hpcaitech HPC-AI Tech

All

    Repositories list

    33 repositories

    • ColossalAI

      Public
      Making large AI models cheaper, faster and more accessible
      aideep-learninghpc
      aideep-learninghpcdistributed-computinginferencebig-modellarge-scaledata-parallelismmodel-parallelismpipeline-parallelism
      Python
      •
      Apache License 2.0
      •4.5k•41k•446•68•Updated Oct 11, 2026Oct 11, 2026

    • Open-Sora

      Public
      Open-Sora: Democratizing Efficient Video Production for All
      Python
      •
      Apache License 2.0
      •3.1k•30k•0•14•Updated Apr 9, 2026Apr 9, 2026

    • public_assets

      Public
      Storing publicly available assets such as images, animations and texts
      Python
      •
      Apache License 2.0
      •19•14•0•0•Updated Apr 9, 2026Apr 9, 2026

    • HPC-AI-SDK

      Public
      HPC-AI TECH 's Fine-tuning SDK
      Python
      •
      Apache License 2.0
      •3•6•0•1•Updated Jan 29, 2026Jan 29, 2026

    • TensorRT-LLM

      Public
      TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and support state-of-the-art optimizations to perform inferenc…
      C++
      •
      Apache License 2.0
      •2.8k•0•0•0•Updated Oct 13, 2025Oct 13, 2025

    • TensorRT-Model-Optimizer

      Public
      A unified library of state-of-the-art model optimization techniques like quantization, pruning, distillation, speculative decoding, etc. It compresses deep lear…
      Python
      •
      Apache License 2.0
      •737•1•0•0•Updated Oct 12, 2025Oct 12, 2025

    • ColossalAI-Documentation

      Public
      Documentation for Colossal-AI
      JavaScript
      •
      Apache License 2.0
      •11•24•4•1•Updated Jun 6, 2025Jun 6, 2025

    • Open-Sora-Demo

      Public
      0•10•0•0•Updated Mar 19, 2025Mar 19, 2025

    • Oh-My-Dockerfile

      Public
      A collection of dockerfiles for various tasks
      Dockerfile
      •
      Apache License 2.0
      •9•24•0•0•Updated Feb 20, 2025Feb 20, 2025

    • ColossalAI-Platform-CLI

      Public
      CLI for ColossalAI Platform
      Python
      •
      Apache License 2.0
      •2•10•1•0•Updated Feb 20, 2025Feb 20, 2025

    • xfold

      Public
      Democratizing AlphaFold3: an PyTorch reimplementation to accelerate protein structure prediction
      Python
      •
      Other
      •13•1•0•0•Updated Dec 16, 2024Dec 16, 2024

    • TensorNVMe

      Public
      A Python library transfers PyTorch tensors between CPU and NVMe
      deep-learningpytorchnvme
      deep-learningpytorchnvmecolossal-ai
      C++
      •27•125•10•1•Updated Nov 27, 2024Nov 27, 2024

    • transformers

      Public
      🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
      Python
      •
      Apache License 2.0
      •35k•10•0•0•Updated Nov 19, 2024Nov 19, 2024

    • LLaVA-NeXT

      Public
      Python
      •
      Apache License 2.0
      •471•1•0•0•Updated Nov 18, 2024Nov 18, 2024

    • graphrag

      Public
      A modular graph-based Retrieval-Augmented Generation (RAG) system
      Python
      •
      MIT License
      •3.8k•1•0•0•Updated Aug 3, 2024Aug 3, 2024

    • FastFold

      Public
      Optimizing AlphaFold Training and Inference on GPU Clusters
      gpuprotein-structurecuda
      gpuprotein-structurecudapytorchparallelismprotein-foldingalphafold2evoformerhabana-gaudi
      Python
      •
      Apache License 2.0
      •90•616•38•6•Updated Jul 16, 2024Jul 16, 2024

    • Cloud-Platform-Docs

      Public
      Documentation for our cloud platform
      JavaScript
      •2•1•0•0•Updated Apr 3, 2024Apr 3, 2024

    • CANN-Installer

      Public
      This repository contains Huawei Ascend CANN files
      Apache License 2.0
      •0•1•1•0•Updated Feb 27, 2024Feb 27, 2024

    • SwiftInfer

      Public
      Efficient AI Inference & Serving
      deep-learninginferenceartificial-intelligence
      deep-learninginferenceartificial-intelligencellamagptllm-servingllm-inferencellama2
      Python
      •
      Apache License 2.0
      •31•474•3•0•Updated Jan 8, 2024Jan 8, 2024

    • EnergonAI

      Public archive
      Large-scale model inference.
      Python
      •
      Apache License 2.0
      •83•629•40•2•Updated Sep 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
      •84•0•0•0•Updated Jul 25, 2023Jul 25, 2023

    • Elixir

      Public
      Elixir: Train a Large Language Model on a Small GPU Cluster
      efficientmemory-managementlarge-language-models
      efficientmemory-managementlarge-language-models
      Python
      •4•15•0•0•Updated Jun 8, 2023Jun 8, 2023

    • ColossalAI-Examples

      Public archive
      Examples of training models with hybrid parallelism using ColossalAI
      Python
      •
      Apache License 2.0
      •100•339•29•2•Updated Mar 23, 2023Mar 23, 2023

    • mmdetection-examples

      Public archive
      Train mmdetection models with ColossalAI.
      Python
      •0•2•1•0•Updated Feb 18, 2023Feb 18, 2023

    • PaLM-colossalai

      Public archive
      Scalable PaLM implementation of PyTorch
      Python
      •
      Apache License 2.0
      •26•190•7•2•Updated Dec 19, 2022Dec 19, 2022

    • GPT-Demo

      Public archive
      GPT Demo with hybrid distributed training
      Python
      •
      Apache License 2.0
      •7•9•2•0•Updated Dec 1, 2022Dec 1, 2022

    • Titans

      Public archive
      A collection of models built with ColossalAI
      Python
      •
      Apache License 2.0
      •16•32•6•1•Updated Nov 22, 2022Nov 22, 2022

    • ColossalAI-Pytorch-lightning

      Public
      Python
      •6•24•0•0•Updated Nov 22, 2022Nov 22, 2022

    • OPT-Benchmark

      Public archive
      Python
      •
      Apache License 2.0
      •5•4•1•0•Updated Nov 22, 2022Nov 22, 2022

    • CachedEmbedding

      Public
      A memory efficient DLRM training solution using ColossalAI
      nlpdeep-learningpytorch
      nlpdeep-learningpytorchembeddingsdlrmrecommandation-systemcolossal-ai
      Python
      •
      Apache License 2.0
      •13•107•2•0•Updated Nov 22, 2022Nov 22, 2022
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.