Adds dependencies and extras for torch 2.3.0 with new xformers versions

[nathan-az]

Does not change any colab requirements - not sure if colab uses torch 2.3.0 by default. Enables testing with torch.compile.

[danielhanchen]

OO thanks for this - highly appreciate it - I will check it out! Might also have to edit the readme (if you're up for it :))

[nathan-az]

Readme updated in accordance with the current style!

It's getting a bit repetitive - I suggest a separate PR that consolidates and explains the install pattern (and if Colab is very popular maybe keeping a specific section for it). Then the sections could be briefer: conda/pip/colab