Enable docstring and README code testing

[jerryzh168]

To make sure the code in docs are runnable, we should do some automatic testing, we can do something similar to https://github.com/pytorch/pytorch/blob/main/test/quantization/core/test_docs.py

cc @HDCharles @vayuda