deps: make pip-git-installable (oryx direct ref + tfp-nightly via fork)

Plain `pip install git+<acopula>` previously succeeded but crashed at import:
pip ignores [tool.uv.sources]/[tool.uv] overrides, so it pulled stock oryx
0.2.9 (broken on jax 0.8) and stable tensorflow-probability alongside the
nightly (namespace collision -> pytype_aval_mappings).

- oryx: bare `oryx>=0.2.9` -> PEP 508 direct ref
  `oryx @ git+https://github.com/thisiscam/oryx.git@copulaad_support`
  (pip and uv both honor it). The fork now depends on tfp-nightly (not stable
  tensorflow-probability), so no stable tfp gets installed.
- jet-array: 0.1.0a1 (git) -> 0.1.0a2 (PyPI).
- Drop [tool.uv.sources] and the tfp override/constraint (superseded).
- hatch: allow-direct-references = true (required to build with the git dep).
- README / docs: `pip install git+...` is now the primary install; uv for dev.

Caveat: the git URL in [project.dependencies] makes the package
un-uploadable to PyPI until a jax-0.8-compatible oryx is published upstream.

Verified: clean-venv `pip install` resolves only tfp-nightly (no stable tfp),
oryx from the fork, jet-array 0.1.0a2; `import acopula` succeeds. uv path also
imports.

Author: thisiscam

README.md

@@ -17,7 +17,13 @@ AD* (arXiv:TBD).
 
 ## Install
 
-Install from source with [uv](https://docs.astral.sh/uv/):
+```bash
+pip install git+https://github.com/thisiscam/acopula
+```
+
+acopula depends on a patched [`oryx`](https://github.com/jax-ml/oryx) build
+(pulled from git as part of the install) because stock `oryx` is incompatible
+with jax 0.8. For development, clone and use [uv](https://docs.astral.sh/uv/):
 
 ```bash
 git clone https://github.com/thisiscam/acopula
@@ -26,12 +32,9 @@ uv sync                 # add --extra examples for the plotting examples
 uv run python examples/01_quickstart.py
 ```
 
-uv is required (for now) because `acopula` depends on a patched
-[`oryx`](https://github.com/jax-ml/oryx) build pinned via `[tool.uv.sources]`;
-plain `pip install` — including `pip install git+…` — ignores that pin, pulls
-the stock PyPI `oryx` (incompatible with jax 0.8), and fails at import. A
-`pip install acopula` from PyPI is **not available yet**; it is gated on an
-upstream `oryx` release compatible with jax 0.8.
+A PyPI `pip install acopula` is **not available yet**: the git-pinned `oryx`
+dependency can't be uploaded to PyPI, so this is gated on an upstream `oryx`
+release compatible with jax 0.8.
 
 `acopula` pins `jax>=0.8,<0.9` because it relies on JAX-internal
 APIs in the [jet-array](https://github.com/thisiscam/jet-array) backend.

docs/index.md

@@ -11,7 +11,13 @@ forward pass via Taylor-mode AD, scaling polynomially in the dimension.
 
 ## Install
 
-Install from source with [uv](https://docs.astral.sh/uv/):
+```bash
+pip install git+https://github.com/thisiscam/acopula
+```
+
+acopula depends on a patched [`oryx`](https://github.com/jax-ml/oryx) build
+(pulled from git during install) because stock `oryx` is incompatible with
+jax 0.8. For development, clone and use [uv](https://docs.astral.sh/uv/):
 
 ```bash
 git clone https://github.com/thisiscam/acopula
@@ -20,12 +26,9 @@ uv sync                 # add --extra examples for the plotting examples
 uv run python examples/01_quickstart.py
 ```
 
-uv is required (for now): `acopula` depends on a patched
-[`oryx`](https://github.com/jax-ml/oryx) build pinned via `[tool.uv.sources]`,
-and plain `pip install` (including `pip install git+…`) ignores that pin, pulls
-the stock PyPI `oryx` (incompatible with jax 0.8), and fails at import. A PyPI
-`pip install acopula` is **not available yet** — it is gated on an upstream
-`oryx` release compatible with jax 0.8.
+A PyPI `pip install acopula` is **not available yet**: the git-pinned `oryx`
+dependency can't be uploaded to PyPI, so it is gated on an upstream `oryx`
+release compatible with jax 0.8.
 
 `acopula` pins `jax>=0.8,<0.9` because it relies on JAX-internal APIs in the
 [jet-array](https://github.com/thisiscam/jet-array) backend.

pyproject.toml

@@ -32,12 +32,17 @@ classifiers = [
 ]
 dependencies = [
     "jax>=0.8,<0.9",
-    "jet-array==0.1.0a1",
+    "jet-array==0.1.0a2",
     "jaxopt>=0.8",
     "optimistix>=0.0.6",
     "quadax>=0.2.9",
     "equinox>=0.13",
-    "oryx>=0.2.9",
+    # Direct ref to the patched fork (jax-0.8 import fixes + copy_p ILDJ +
+    # tfp-nightly dep).  pip and uv both honor this, so `pip install
+    # git+<acopula>` works.  NOTE: a git URL here makes the package
+    # un-uploadable to PyPI; swap back to `oryx>=<ver>` once a jax-0.8
+    # compatible oryx is published.
+    "oryx @ git+https://github.com/thisiscam/oryx.git@copulaad_support",
     "tfp-nightly==0.26.0.dev20251116",
     "scipy>=1.11",
     "sympy>=1.12",
@@ -69,35 +74,11 @@ docs = [
 required-environments = [
     "sys_platform == 'linux' and platform_machine == 'x86_64'",
 ]
-# oryx pulls in stable `tensorflow-probability`, which collides on disk with
-# `tfp-nightly` (both write into the `tensorflow_probability/` namespace).
-# Depending on uv's install order, stable's broken `ops.py` (referencing the
-# removed `jax.interpreters.xla.pytype_aval_mappings`) wins and crashes at
-# import.  Drop stable tfp from resolution; tfp-nightly satisfies every oryx
-# import path.
-override-dependencies = [
-    "tensorflow-probability ; sys_platform == 'never'",
-]
-# Pin tfp-nightly to the version that works with jax 0.8.x.  Newer nightlies
-# remove `jax.interpreters.xla.pytype_aval_mappings` and crash at import.
-constraint-dependencies = [
-    "tfp-nightly==0.26.0.dev20251116",
-]
 
-# PyPI's oryx 0.2.9 imports `from jax import util`, which jax 0.8.x removed.
-# The `copulaad_support` branch of the patched fork uses `from jax._src import
-# util` and ships the lax.copy_p ILDJ shim acopula expects.  This source
-# override applies only under uv; the published `oryx>=0.2.9` PyPI dep stays
-# in `dependencies` so a future plain `pip install acopula` works once the
-# upstream fix lands.
-[tool.uv.sources]
-oryx = { git = "https://github.com/thisiscam/oryx.git", branch = "copulaad_support" }
-# PyPI's jet-array==0.1.0a1 predates the log-space Taylor expansion
-# (sign, log|c|) representation that `bell.py`'s `log_jet=True` path
-# uses, and also predates the log_add NaN-gradient shift-guard at
-# structural zeros.  Pin to the public fork until a new wheel is
-# released.
-jet-array = { git = "https://github.com/thisiscam/jet-array.git", branch = "main" }
+[tool.hatch.metadata]
+# acopula depends on the oryx fork via a git direct reference (see
+# [project.dependencies]); hatchling requires opting in to build it.
+allow-direct-references = true
 
 [tool.hatch.build.targets.wheel]
 packages = ["acopula"]