Hemil Desai

Happy to help with this.

Not sure why `CI / Test (Python) (3.6, ubuntu-20.04)` is failing. The error seems to be `No matching distribution found for tensorflow-cpu==2.4.0 (from -r requirements-test.txt (line 4))`.

Cool, thanks for looking into it. Let me know if I can help.

@bfirsh it looks like the pip version is the culprit - https://github.com/tensorflow/tensorflow/issues/39130#issuecomment-623335820

Yeah, sometimes with issues like this, it can be quite frustrating.

Hey @bfirsh, let me know if this is good to go or if there are any changes required.

@bfirsh Sure. Can you point me to where the binary is installed so that I can change it to be installed in the venv?

Thanks for the response. In [fx2ait vision example](https://github.com/facebookincubator/AITemplate/tree/main/fx2ait/fx2ait/example/02_vision_model), it is mentioned that ``` Notice that because AIT supports channel last, while pytorch supports channel first operation, we need to permute...