paddlefx

paddlefx is an experimental project for building PaddlePaddle Python IR.

Community activities

• Tracking issue for PaddleSOT project: PaddlePaddle/PaddleSOT#133
• Online tech discussion meeting: PaddlePaddle/PaddleSOT#149
• China open source software innovation challenge: PaddlePaddle/Paddle#53172 (comment)

If you are interested in these activities, please send a email to ext_paddle_oss@baidu.com, we'll invite you into wechat group.

Quick Start

pip install -r requirements_dev.txt

pip install -e .

pytest -n3 tests

Explore its Capabilities

Here are some examples of what paddlefx can do:

• Capture and compile python bytecodes into a fx graph. See this example.
• Run the captured ResNet fx Graph on TVM. See this example
• Trace ResNet and 10 other vision models. See this example and this test.
• Edit fx graphs. See this example.
• Profile ResNet. See this example.
• Demonstrate how to lower the fx graph IR to a native interpreter. See this example.

Contribution

This is a community driven project, maintenance of this project is on a "best effort" basis. The ideas and even lots of codes are borrowed from pytorch 2.0.

If you'd like to contribute, please feel free to raise a pull request, discuss in issues or discussions.