+3 −1

mode.py

pytensor/compile/mode.py

+1 −0

__init__.py

pytensor/link/jax/dispatch/__init__.py

+61 −0

loop.py

pytensor/link/jax/dispatch/loop.py

+1 −0

utils.py

pytensor/link/utils.py

0

__init__.py

pytensor/loop/__init__.py

+199 −0

basic.py

pytensor/loop/basic.py

+478 −0

op.py

pytensor/loop/op.py

+20 −0

basic.py

pytensor/typed_list/basic.py

+4 −6

test_basic.py

tests/link/jax/test_basic.py

+114 −0

test_loop.py

tests/link/jax/test_loop.py

0

__init__.py

tests/loop/__init__.py

+146 −0

test_basic.py

tests/loop/test_basic.py

+186 −0

test_op.py

tests/loop/test_op.py

+4 −0

test_type.py

tests/typed_list/test_type.py