Commit History

E2e device cuda (#575)
2414673
unverified

winglian commited on

Model parallel (#538)
f6060a6
unverified

winglian commited on

Fix crash when running without CUDA
15f6e57

chargoddard commited on

improve GPU logging to break out pytorch cache and system mem
7b55fe6

tmm1 commited on

log GPU memory usage
e303d64

tmm1 commited on