hpcaitech/TensorNVMe hpcaitech/TensorNVMe

  • Stars
    star
    92
  • Rank 362,966 (Top 8 %)
  • Language
    C++
  • Created over 2 years ago
  • Updated over 1 year ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
hpcaitech/TensorNVMe
github

hpcaitech

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

A Python library transfers PyTorch tensors between CPU and NVMe

More Repositories

1

ColossalAI

Making large AI models cheaper, faster and more accessible
Python
38,629
star
2

Open-Sora

Open-Sora: Democratizing Efficient Video Production for All
Python
21,795
star
3

EnergonAI

Large-scale model inference.
Python
629
star
4

FastFold

Optimizing AlphaFold Training and Inference on GPU Clusters
Python
490
star
5

ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI
Python
335
star
6

PaLM-colossalai

Scalable PaLM implementation of PyTorch
Python
192
star
7

CachedEmbedding

A memory efficient DLRM training solution using ColossalAI
Python
93
star
8

SkyComputing

Sky Computing: Accelerating Geo-distributed Computing in Federated Learning
Python
89
star
9

ColossalAI-Benchmark

Performance benchmarking with ColossalAI
Python
38
star
10

Titans

A collection of models built with ColossalAI
Python
32
star
11

ColossalAI-Documentation

Documentation for Colossal-AI
JavaScript
24
star
12

ColossalAI-Pytorch-lightning

Python
23
star
13

Oh-My-Dockerfile

A collection of dockerfiles for various tasks
Dockerfile
15
star
14

Elixir

Elixir: Train a Large Language Model on a Small GPU Cluster
Python
12
star
15

GPT-Demo

GPT Demo with hybrid distributed training
Python
9
star
16

public_assets

Storing publicly available assets such as images, animations and texts
Python
8
star
17

OPT-Benchmark

Python
4
star
18

mmdetection-examples

Train mmdetection models with ColossalAI.
Python
2
star