Skip to content

Commit

Permalink
first commit for a decorator that transforms JAX to pytensor
Browse files Browse the repository at this point in the history
  • Loading branch information
jdehning committed Dec 12, 2024
1 parent 231a977 commit 0c6304e
Showing 3 changed files with 451 additions and 1 deletion.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -78,7 +78,7 @@ tests = [
"pytest-sphinx",
]
rtd = ["sphinx>=5.1.0,<6", "pygments", "pydot", "pydot2", "pydot-ng"]
jax = ["jax", "jaxlib"]
jax = ["jax", "jaxlib", "equinox"]
numba = ["numba>=0.57", "llvmlite"]

[tool.setuptools.packages.find]
Loading

0 comments on commit 0c6304e

Please sign in to comment.