Patrick Kidger
Patrick Kidger
At the moment all global interpolation routines require that the sequence of times be strictly increasing. In practice it's often helpful to admit weakly increasing times as well, e.g. to...
At least as an option, this might be a nice thing to have. Would need to think about the correct way of backpropagating through it + making sure XLA doesn't...
At the moment these custom operations aren't yet part of the public API, but they should make it possible to replace `unvmap_*` and `nondifferentiable_output`, which are currently implemented as custom...
At the moment it is just the same as a `jax.jvp` of `solution.evaluate`; in particular it is the derivative of the numerical spline approximation; not an approximation to the true...
At the moment there are several `stop_gradient`s in there, primarily to avoid instability observed in the backward pass. There's been a few bugs fixed in `IController` since then, so this...
That is, classes supporting the `__torch_function__` protocol. This shouldn't be too difficult -- most of the necessary work has already been done. - There's some places where we have `torch.Tensor`...
e.g. `cat(a: TensorType["x"], b: TensorType["y"]) -> TensorType["x + y"]`. Given that `torchtyping` operates at runtime then arbitrary Python expressions should be possible. Essentially just call `eval` with the appropriate constants...
Essentially: `typeguard` uses a "memo" object as a place to store information over the duration of a function call. At the moment, the checking is primarily performed by: - Storing...