RTX PRO 6000 /SM_120 incompatibility

Hello,

is it possible to run AF2 on Blackwell cards? Would you be so kind as to provide dockerfile/requirements.txt ?

I tried a dozen of possible combinations between jax, cuda, haiku, etc., but none seem to work (XLA fails to initialise cuda, cannot see card, etc.)

thanks!