search

Oneflow-Inc / libai

LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training
https://libai.readthedocs.io
Apache License 2.0
390 stars 55 forks source link

replace cuda to device in folder libai #545

Closed ShawnXuan closed 3 months ago

ShawnXuan commented 3 months ago

replace cuda to device in subfolder libai