注册并分享邀请链接,可获得视频播放与邀请奖励。

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 粉丝
Modern AI workloads rely on a growing range of specialized GPU kernels, and no single kernel generation backend is optimal across every operator. At #PyTorchCon# North America 2026, Liz Li and Jiahui Cao (@AMD) will present “Extending TorchInductor with FlyDSL: A New MLIR-Native Backend for High-Performance GEMMs.” Liz and Jiahui will share how FlyDSL, AMD’s Python-native and MLIR-based GPU kernel DSL, integrates with TorchInductor’s compilation and autotuning pipeline while preserving the torch.compile user experience. The session will also present performance results for transformer training and inference on AMD Instinct GPUs, comparing Triton and FlyDSL implementations. Register for PyTorch Conference North America 2026:
显示更多