Loreon
Labs
Platform
Docs
Home
Ecosystems
Other
tpu-raiden
saikat-royc/tpu-raiden
Other
Emerging
GitHub
Stars
—
Forks
—
Contributors
6
Last push
13h ago
Recent commits
Latest commits.
Fixed the msan and tsan failures for `socket_transport_test`.
829533a
datenglin
16d ago
Fixed the msan and asan failures for `weight_synchronizer_ffi_test`.
ff6cbea
datenglin
16d ago
Refactor TPU Raiden host memory allocator:
8c6b042
justinlu
16d ago
Improved the latency of `prepare_reshard` by fusing the initialization of weight synchronizer and d2h in the source.
932dc07
datenglin
16d ago
Refactor TransferEngineBase to use KVCacheManager's BlockTransport for H2H transfers
afd52bd
sixiang
17d ago
Fix Bazel build scripts and FFI linkage setup.
bf164aa
amylin
17d ago
test(jax): add pure C++ unit tests for raw transfer on TAP
38916b4
fhzhang
18d ago
test(jax): add weight synchronizer FFI unit tests on TAP
f3da50c
fhzhang
18d ago
Top contributors
Builders behind this project.
fenghuizhang
35 commits
juncgu-google
8 commits
sixiang-google
8 commits
linamy85
7 commits
justinlu13
4 commits
datenglin
3 commits
