paddlefx is an experimental project for building PaddlePaddle Python IR.
- 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.
pip install -r requirements_dev.txt
pip install -e .
pytest -n3 testsHere are some examples of what paddlefx can do:
- Capture and compile python bytecodes into a fx graph. 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.
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.