Sometimes some sort of computation function is needed for a use case and writing said functions in Theano is common. For example, the adstock function referenced in a 2017 Google paper on Media Mix Modeling: import theano.tensor as tt def geometric_adstock(x, theta, alpha,L=12): w = tt.as_tenso…

Here is an example by @dfm : deterministic_op.ipynb · GitHub

Custom functions in Jax?

jbuddy_13 April 4, 2021, 5:30pm 2

I think I read somewhere that this is totally a thing, and numpy too, maybe?

Topic		Replies	Views
Auto-differentiation of user defined functions? Questions	2	924	December 2, 2020
Pymc3 likelihood math with non-theano function Questions theano	4	1088	September 25, 2017
Implement deterministic nontrivial function of random variables Questions theano	4	1116	August 18, 2019
Theano error with element wise multiplication Questions theano	5	2525	July 12, 2017
Inquiry about how to define a function as pm.Uniform in pymc3 Questions development , theano	4	677	August 9, 2018