covariate support

2025-10-28 22:29:40 +00:00
parent 80f1a1eafe
commit c24db2afd2
2 changed files with 234 additions and 1 deletions
@@ -30,6 +30,10 @@ flax = [
    "jaxtyping",
    "jax[cuda]"
 ]
+xreg = [
+    "jax[cuda]",
+    "scikit-learn",
+]

 [tool.ruff]
 line-length = 88