[dependencies] migrate to uv official wheels

[erictang000]

Migrate the pyproject.toml to use https://wheels.astral.sh/ instead of our custom built ones

[erictang000]

made some progress in #2023, need to update TE to >=2.16 to switch flash-attn and TE