Name: A torch.fx Based Compression Toolkit Empowered by torch_musa - Fan Mo, Moore Threads
Start: 2025-06-07T11:50:00+0800
End: 2025-06-07T12:00:00+0800

In-person | 2025 June 7
Learn more on our website

The Sched app allows you to build your schedule but is not a substitute for your event registration. You must be registered to participate in the sessions. If you have not registered but would like to join us, please visit the BAAI Conference webpage.

Please note: This schedule is automatically displayed in China Standard Time (UTC+08:00). To see the schedule in your preferred timezone, please select from the drop-down located at the bottom of the menu to the right.

IMPORTANT NOTE: Timing of sessions and room locations are subject to change.

Saturday June 7, 2025 11:50 - 12:00 CST

TBA

An introduction of torch_musa, which is built by Moore Threads to provide PyTorch backend support for MUSA architecture, and with its customized support for quantization, we built a light-weight model compression toolkit named NeuroTrim, which can quantize a variety of deep learning models and run it on native PyTorch.
We expanded the graph tracing capability of PyTorch by integrating fx graph, torchInductor, and ExportedIR, and added many lowered operators in torch_musa, which can be easily used for debugging, and then export to ONNX or other formats for custom accelerator hardwares.

Speakers

Fan Mo

Machine Learning Engineer, Moore Threads

Fan Mo received a M.S degree in 2020 from Shanghai JiaoTong University, who has held machine learning engineer at Moore Threads. His current work focuses on AI Infrastructure, and is the principal maintainer of torch_musa, an open-source project that provides PyTorch backend support... Read More →

Saturday June 7, 2025 11:50 - 12:00 CST
TBA

PyTorch Ecosystem and Tools

PyTorch Day China 2025

Fan Mo

Sign up or log in to save this to your schedule, view media, leave feedback and see who's attending!