mirror of https://github.com/hpcaitech/ColossalAI
ae1b58cd16
* [tensor] added linear implementation for the new sharding spec * polish code |
||
|---|---|---|
| .. | ||
| _ops | ||
| graph | ||
| layer | ||
| loss | ||
| lr_scheduler | ||
| metric | ||
| optimizer | ||
| parallel | ||
| __init__.py | ||
| init.py | ||