Initial support for 3.12

[bhavya01]

For #8946

I was able to train a resnet model after building the wheel from source.

(torch312) ➜  xla git:(master) ✗ python examples/train_resnet_base.py
WARNING:root:libtpu.so and TPU device found. Setting PJRT_DEVICE=TPU.
Epoch 1 train begin  6:06PM UTC on Jun 25, 2025
epoch: 1, step: 0, loss: 6.895828723907471, rate: 4.670206670408138
epoch: 1, step: 10, loss: 6.885840892791748, rate: 754.674309045292
epoch: 1, step: 20, loss: 6.875847339630127, rate: 1399.1818112606793
epoch: 1, step: 30, loss: 6.865854263305664, rate: 1658.248962950589
epoch: 1, step: 40, loss: 6.855878829956055, rate: 1755.7655474631229
epoch: 1, step: 50, loss: 6.845893383026123, rate: 1800.957394879235
epoch: 1, step: 60, loss: 6.835905075073242, rate: 1816.1105562107623
epoch: 1, step: 70, loss: 6.825912952423096, rate: 1825.6423765339405
epoch: 1, step: 80, loss: 6.815918445587158, rate: 1828.584844518641
epoch: 1, step: 90, loss: 6.8059306144714355, rate: 1831.154835138746
epoch: 1, step: 100, loss: 6.795934677124023, rate: 1830.7875810113103
epoch: 1, step: 110, loss: 6.785972595214844, rate: 1829.6420722475502
epoch: 1, step: 120, loss: 6.776001453399658, rate: 1830.370502407869
epoch: 1, step: 130, loss: 6.766016006469727, rate: 1831.0602388877992
epoch: 1, step: 140, loss: 6.756027698516846, rate: 1828.946538821459
epoch: 1, step: 150, loss: 6.746034622192383, rate: 1829.7046951690263
epoch: 1, step: 160, loss: 6.73603630065918, rate: 1830.356592581364
epoch: 1, step: 170, loss: 6.7260355949401855, rate: 1829.9008492807068
epoch: 1, step: 180, loss: 6.716034412384033, rate: 1829.6818910001666
epoch: 1, step: 190, loss: 6.706037521362305, rate: 1831.5200718576157
epoch: 1, step: 200, loss: 6.696044445037842, rate: 1830.4032785293643
epoch: 1, step: 210, loss: 6.6860575675964355, rate: 1831.3699679697675
epoch: 1, step: 220, loss: 6.676076889038086, rate: 1831.2460041785037
epoch: 1, step: 230, loss: 6.666107177734375, rate: 1831.5409170709086
epoch: 1, step: 240, loss: 6.6561503410339355, rate: 1830.7053866045135
epoch: 1, step: 250, loss: 6.646203994750977, rate: 1831.3891703503266
epoch: 1, step: 260, loss: 6.63612174987793, rate: 1830.271234197638
epoch: 1, step: 270, loss: 6.626188278198242, rate: 1828.9326157193525
epoch: 1, step: 280, loss: 6.616239547729492, rate: 1827.4076259705325
epoch: 1, step: 290, loss: 6.606279373168945, rate: 1828.684601676097
Epoch 1 train end  6:07PM UTC on Jun 25, 2025