Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Stateful Controls #559

Open
wants to merge 45 commits into
base: dev
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
45 commits
Select commit Hold shift + click to select a range
51c4c41
intermediate work
lockwo Dec 29, 2024
382d171
solver work
lockwo Dec 31, 2024
10b16bc
benchmark
lockwo Dec 31, 2024
9921407
timing
lockwo Dec 31, 2024
7956b5b
more work
lockwo Dec 31, 2024
7fb9baa
work
lockwo Jan 1, 2025
fd059f7
Merge branch 'patrick-kidger-main' into Owen/control_revamp
lockwo Jan 2, 2025
35dc705
work
lockwo Jan 3, 2025
427a594
some fixes
lockwo Jan 3, 2025
29138ed
testing work
lockwo Jan 3, 2025
7f76cdd
fixes
lockwo Jan 4, 2025
d24e6f1
add test
lockwo Jan 4, 2025
a1374f9
tests + examples
lockwo Jan 5, 2025
919abf9
format
lockwo Jan 5, 2025
12bcf5a
Merge branch 'main' into Owen/control_revamp
lockwo Jan 5, 2025
37640ed
remove todo
lockwo Jan 8, 2025
cc0d4bc
Allowing args into grad_f for ULD
ricor07 Dec 27, 2024
d304d9f
clean up
lockwo Jan 16, 2025
22d00ca
Merge branch 'dev' into Owen/control_revamp
lockwo Jan 16, 2025
1ad8dad
int
lockwo Jan 16, 2025
d0f161c
fix
lockwo Jan 22, 2025
16fedb2
ULD fix
lockwo Jan 22, 2025
9a19d68
more langevin fixes
lockwo Jan 22, 2025
4994982
adjoit
lockwo Jan 24, 2025
80fef54
shorten test
lockwo Jan 27, 2025
1d34946
Test fixes for v0.5.0 + args for langevin
patrick-kidger Jan 26, 2025
b8683f4
Merge branch 'dev' into Owen/control_revamp
lockwo Jan 28, 2025
1067c10
Fix for making vmap over diffeqsolve possible (#578)
LuggiStruggi Jan 28, 2025
5366e65
Tweak test name
patrick-kidger Jan 28, 2025
54e9e77
Update pyproject.toml to meet poetry conventions
joharkit Jan 28, 2025
92fb93d
Fixed a major source of bugs: ControlTerms no longer broadcast.
patrick-kidger Jan 12, 2025
7c2c720
Now using jaxtyping.Real for prettier documentation.
patrick-kidger Jan 12, 2025
583cd6d
Bumped minimum version of Python to 3.10
patrick-kidger Jan 28, 2025
4272270
Investigating if we can drop the typeguard dependency.
patrick-kidger Jan 28, 2025
9236a68
Merge branch 'dev' into Owen/control_revamp
lockwo Jan 30, 2025
96f8bf3
fix merge
lockwo Jan 30, 2025
1946a8b
fix tests
lockwo Feb 3, 2025
b3bb170
fix test2
lockwo Feb 3, 2025
03e5b92
trying larger stepsize
lockwo Feb 4, 2025
766b471
does splitting it up help? (passes locally, but github actions fails)
lockwo Feb 4, 2025
4d22b6f
Merge branch 'patrick-kidger:dev' into dev
lockwo Feb 9, 2025
865846b
Merge branch 'dev' into Owen/control_revamp
lockwo Feb 9, 2025
7865a16
update benchmark
lockwo Feb 9, 2025
20e700d
update jit results
lockwo Feb 9, 2025
e4cd2a3
return jit
lockwo Feb 9, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
remove todo
lockwo committed Jan 8, 2025
commit 37640edc8c8b0b9303fc771c1a34f450419d0fa4
1 change: 0 additions & 1 deletion diffrax/_brownian/path.py
Original file line number Diff line number Diff line change
@@ -118,7 +118,6 @@ def _generate_noise(
shape: jax.ShapeDtypeStruct,
max_steps: int,
) -> Float[Array, "..."]:
# TODO: merge into a single jr.normal call
if self.levy_area is SpaceTimeTimeLevyArea:
noise = jr.normal(key, (3, max_steps, *shape.shape), shape.dtype)
elif self.levy_area is SpaceTimeLevyArea: