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