PyTorch
@PyTorch
Tensors and neural networks in Python with strong hardware acceleration. PyTorch is an open source project at the Linux Foundation. #PyTorchFoundation#
가입 September 2016
106 팔로잉 중 519.9K 팬
Replacing CUDA with FBTriton for Table Batched Embedding (TBE) kernels delivered massive gains for our rec sys models.
@Meta contributors achieved up to 1.28x faster forward passes, 2x faster backward passes, and a huge boost in developer velocity. Check out the engineering deep-dive:
✍️ Daohang Shi, Oleksandr Stashuk, Rupert Wu, Liangbei Xu, Rich Zhu
더 보기
