| .. |
|
__init__.py
|
9d81716bfd
[v0.5.3] Release Candidate (#388)
|
hace 8 meses |
|
cuda_wrapper.py
|
cc6399792f
fix: keep consistent with how pytorch finds libcudart.so
|
hace 5 meses |
|
custom_all_reduce.py
|
4d4e767838
ci: take one of fixing lint issues
|
hace 5 meses |
|
custom_all_reduce_utils.py
|
4d4e767838
ci: take one of fixing lint issues
|
hace 5 meses |
|
pynccl.py
|
6b1fdd07bd
chore: add isort and refactor formatting script and utils
|
hace 5 meses |
|
pynccl_wrapper.py
|
f91991f584
fix: f-string fixes
|
hace 5 meses |
|
shm_broadcast.py
|
bcd8e72dbd
fix: zmq hangs with large requests
|
hace 5 meses |
|
tpu_communicator.py
|
8d88814475
chore: reduce XLA compile times
|
hace 5 meses |
AlpinDale
4d4e767838
ci: take one of fixing lint issues