38d ago · PyTorch Blog →
This post explores the FBTriton kernel design for Table Batched Embedding (TBE) forward and backward passes.