search

mlcommons / chakra

Repository for MLCommons Chakra schema and tools
https://mlcommons.org/working-groups/research/chakra/
Apache License 2.0
62 stars 32 forks source link

Documentation on how to use Chakra with PyTorch #147

Open whatdhack opened 2 months ago

whatdhack commented 2 months ago

A simple 101 example to setup, run and visualize/analyze a PyTorch run would be very useful.

TaekyungHeo commented 2 months ago

Please see if https://github.com/mlcommons/chakra/wiki/Chakra-Execution-Trace-Collection-%E2%80%90-A-Comprehensive-Guide-on-Merging-PyTorch-and-Kineto-Traces works for you.