7f2a703ed5
* Merge with private repo * Update README * Update README * Update README * Add PyTorch requirements * Fix sync scopes for MQA logits (#256) * Update README
5 lines
116 B
Python
5 lines
116 B
Python
from . import math, layout
|
|
from .layout import *
|
|
from .math import *
|
|
from .dist import init_dist, uneven_all_gather
|