Thomas Christie

icon company University of Tübingen icon location Tübingen, Germany icon location PhD Student in Machine Learning @ University of Tübingen

Results 3 comments of Thomas Christie

Problem training model with jax.jvp in forward pass

Great, thanks, that worked! And yeah, I think having `nnx.jvp` would probably be a bit more intuitive - I actually searched to see if this existed whilst debugging.

[WIP] Add skeleton code for dealing with priors

@thomaspinder have just added some skeleton code for adding priors to kernel parameters. As suggested, I was going to deal with priors by adding a `prior` field to the `param_field`...

feat: Attach priors to parameters

Will we be wanting to enable users to specify whether their prior is over the *constrained* or *unconstrained* parameters, or just force users to place the prior over the *constrained*...