[doc] Fix JIT code highlighting (#37338)
Summary:
Fix https://github.com/pytorch/pytorch/issues/36216
| Before | After |
| --- | --- |
| |  |
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37338
Differential Revision: D21262421
Pulled By: mrshenli
fbshipit-source-id: 4fb62cce9543e6a4852828f58a279c36565f8c44
