PyTorch details FBTriton kernels for Table Batched Embeddings · NewsAI