Merge pull request #39 from drisspg/update-readme

make readme prettier

Author: msaroufim

README.md

@@ -9,26 +9,26 @@ The torchao package contains apis and workflows used to apply AO techniques like
 **Note: this library makes liberal use of several new features in pytorch, its recommended to use it with the current pytorch nightly if you want full feature coverage. If not, the subclass APIs may not work, though the module swap api's will still work.**
 
 1. From PyPI:
-```
+```Shell
 pip install torchao
 ```
 
 2. From Source:
 
-```
+```Shell
 git clone https://github.com/pytorch-labs/ao
 cd ao
 python setup.py install
 ```
 
 Verify Installation:
 
-```
+```Shell
 pip list | grep torchao
 ```
 
 Expected Output
-```
+```Shell
 torchao                            0.0.1                   <install dir>
 ```
 
@@ -50,7 +50,7 @@ converts the floating point linear matmul of the original linear op to a dynamic
 
 Example
 
-```
+```Python
 import torch
 from torchao.quantization import quant_api
 
@@ -76,7 +76,7 @@ converts the floating point linear matmul of the original linear op to a weight
 
 Example
 
-```
+```Python
 # some user model and example input
 ...
 
@@ -97,7 +97,7 @@ converts the floating point linear matmul of the original linear op to a weight
 
 Example
 
-```
+```Python
 # some user model and example input
 ...
 
@@ -123,7 +123,7 @@ Due to requiring calibration, the API is slightly more complicated and currently
 
 Example
 
-```
+```Python
 import torch
 from torchao.quantization.smoothquant import swap_linear_with_smooth_fq_linear, smooth_fq_linear_to_inference