diff --git a/.github/ISSUE_TEMPLATE/config.yml b/.github/ISSUE_TEMPLATE/config.yml new file mode 100644 index 000000000..3bdc1ea60 --- /dev/null +++ b/.github/ISSUE_TEMPLATE/config.yml @@ -0,0 +1,5 @@ +blank_issues_enabled: true +contact_links: + - name: "😊 Discussions" + url: https://github.com/hpcaitech/ColossalAI/discussions + about: Ask questions and discuss with other Colossal-AI community members in our forum diff --git a/README.md b/README.md index 3f8dfc556..f04f7776b 100644 --- a/README.md +++ b/README.md @@ -1,11 +1,10 @@ # Colossal-AI - +![logo](./docs/images/Colossal-AI_logo.png) +
+

Paper | Documentation | Forum | Blog

+
An integrated large-scale model training system with efficient parallelization techniques. -Paper: [Colossal-AI: A Unified Deep Learning System For Large-Scale Parallel Training](https://arxiv.org/abs/2110.14883) - -Blog: [Colossal-AI: A Unified Deep Learning System For Large-Scale Parallel Training](https://www.hpcaitech.com/blog) - ## Installation ### PyPI @@ -34,10 +33,6 @@ Install and enable CUDA kernel fusion (compulsory installation when using fused pip install -v --no-cache-dir --global-option="--cuda_ext" . ``` -## Documentation - -- [Documentation](https://www.colossalai.org/) - ## Use Docker @@ -82,7 +77,7 @@ model = ... # sampler by default train_dataset = ... train_dataloader = get_dataloader(dataset=dataset, - shuffle=True, + shuffle=True ) diff --git a/docs/images/Colossal-AI_logo.png b/docs/images/Colossal-AI_logo.png new file mode 100644 index 000000000..bdba4b952 Binary files /dev/null and b/docs/images/Colossal-AI_logo.png differ