chore: repo is now called jflux

pyproject.toml

@@ -1,5 +1,5 @@
 [project]
-name = "flux-jax"
+name = "jflux"
 version = "0.1.0"
 description = "Inference codebase for Flux in Jax"
 readme = "README.md"
@@ -9,7 +9,7 @@ dependencies = [
     "einops>=0.8.0",
     "fire>=0.6.0",
     "flax>=0.9.0",
-    "flux-jax",
+    "jflux",
     # FIXME: Allow for local installation without GPUs as well `jax[cuda12]`
     "jax>=0.4.31",
     "mypy>=1.11.2",
@@ -28,7 +28,7 @@ dev-dependencies = [
 ]
 
 [tool.uv.sources]
-flux-jax = { workspace = true }
+jflux = { workspace = true }
 flux = { git = "https://github.com/black-forest-labs/flux.git" }
 
 [tool.ruff.lint]