Skip to content

@hpcaitech HPC-AI Tech

Change the repository type filter

All

    Repositories list

    33 repositories

    • ColossalAI

      Public
      Making large AI models cheaper, faster and more accessible
      aideep-learninghpcdistributed-computinginferencebig-modellarge-scaledata-parallelismmodel-parallelismpipeline-parallelism
      Python
      •
      Apache License 2.0
      •4.5k•41k•442•50•Updated Feb 23, 2026Feb 23, 2026

    • HPC-AI-SDK

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

    • public_assets

      Public
      Storing publicly available assets such as images, animations and texts
      Python
      •
      Apache License 2.0
      •19•14•0•0•Updated Nov 28, 2025Nov 28, 2025

    • 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.1k•1•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
      •281•1•0•1•Updated Oct 12, 2025Oct 12, 2025

    • ColossalAI-Documentation

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

    • Open-Sora

      Public
      Open-Sora: Democratizing Efficient Video Production for All
      Python
      •
      Apache License 2.0
      •2.9k•29k•1•9•Updated Apr 30, 2025Apr 30, 2025

    • Open-Sora-Demo

      Public
      0•8•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
      •11•1•0•0•Updated Dec 16, 2024Dec 16, 2024

    • TensorNVMe

      Public
      A Python library transfers PyTorch tensors between CPU and NVMe
      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
      •32k•10•0•0•Updated Nov 19, 2024Nov 19, 2024

    • LLaVA-NeXT

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

    • graphrag

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

    • FastFold

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

    • Cloud-Platform-Docs

      Public
      Documentation for our cloud platform
      JavaScript
      •3•1•0•1•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-intelligencellamagptllm-servingllm-inferencellama2
      Python
      •
      Apache License 2.0
      •31•479•3•0•Updated Jan 8, 2024Jan 8, 2024

    • EnergonAI

      Public archive
      Large-scale model inference.
      Python
      •
      Apache License 2.0
      •85•627•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
      •74•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
      Python
      •5•15•0•1•Updated Jun 8, 2023Jun 8, 2023

    • ColossalAI-Examples

      Public archive
      Examples of training models with hybrid parallelism using ColossalAI
      Python
      •
      Apache License 2.0
      •102•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
      •27•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•10•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-learningpytorchembeddingsdlrmrecommandation-systemcolossal-ai
      Python
      •
      Apache License 2.0
      •14•107•2•0•Updated Nov 22, 2022Nov 22, 2022