Align the jax stack, name the CUDA extra for CUDA, and drop Python 3.11 - #10

Node.js 20 is deprecated. The following actions target Node.js 20 but are being forced to run on Node.js 24: actions/checkout@v4, actions/setup-python@v5, actions/upload-artifact@v4, astral-sh/setup-uv@v1. For more information see: https://github.blog/changelog/2025-09-19-deprecation-of-node-20-on-github-actions-runners/

Failed to save: <h2>Our services aren't available right now</h2><p>We're working to restore all services as soon as possible. Please check back soon.</p>0VhSSagAAAADS+X6E2veQR7JRVDhIqUhIUEhMMzBFREdFMDEyMgBFZGdl