MANIFEST.in
README.md
setup.py
bmtrain/__init__.py
bmtrain/block_layer.py
bmtrain/checkpointing.py
bmtrain/debug.py
bmtrain/global_var.py
bmtrain/init.py
bmtrain/layer.py
bmtrain/param_init.py
bmtrain/parameter.py
bmtrain/pipe_layer.py
bmtrain/store.py
bmtrain/synchronize.py
bmtrain/utils.py
bmtrain/wrapper.py
bmtrain.egg-info/PKG-INFO
bmtrain.egg-info/SOURCES.txt
bmtrain.egg-info/dependency_links.txt
bmtrain.egg-info/requires.txt
bmtrain.egg-info/top_level.txt
bmtrain/benchmark/__init__.py
bmtrain/benchmark/all_gather.py
bmtrain/benchmark/reduce_scatter.py
bmtrain/benchmark/send_recv.py
bmtrain/benchmark/shape.py
bmtrain/benchmark/utils.py
bmtrain/distributed/__init__.py
bmtrain/distributed/ops.py
bmtrain/inspect/__init__.py
bmtrain/inspect/format.py
bmtrain/inspect/model.py
bmtrain/inspect/tensor.py
bmtrain/loss/__init__.py
bmtrain/loss/cross_entropy.py
bmtrain/lr_scheduler/__init__.py
bmtrain/lr_scheduler/cosine.py
bmtrain/lr_scheduler/exponential.py
bmtrain/lr_scheduler/linear.py
bmtrain/lr_scheduler/no_decay.py
bmtrain/lr_scheduler/noam.py
bmtrain/lr_scheduler/warmup.py
bmtrain/nccl/__init__.py
bmtrain/nccl/enums.py
bmtrain/optim/__init__.py
bmtrain/optim/adam.py
bmtrain/optim/adam_offload.py
bmtrain/optim/optim_manager.py
csrc/adam_cpu.cpp
csrc/adam_cuda.cpp
csrc/cross_entropy_loss.cpp
csrc/nccl.cpp
csrc/cuda/adam.cu
csrc/cuda/cross_entropy.cu
csrc/cuda/has_inf_nan.cu
csrc/cuda/reduce.cuh
csrc/include/nccl.h
csrc/include/nccl_net.h