Add effect handler and operation to handle non-smooth dynamics

BasisResearch / chirho

An experimental language for causal reasoning

https://basisresearch.github.io/chirho/getting_started.html

Apache License 2.0

172 stars 12 forks source link

Add effect handler and operation to handle non-smooth dynamics #147

Open agrawalraj opened 1 year ago

agrawalraj commented 1 year ago

The event function in https://github.com/rtqichen/torchdiffeq allows users to model change in dynamics at certain instances in time (e.g., bouncing ball example). Currently, we do not support such dynamics in #137.

SamWitty commented 1 year ago

An example syntax for what this could look like, but certainly not attached.

@Hybridify(EventFn, StateTransformFn)
def model(X, dX):
   ...