Here is the Hakaru IR AST https://github.com/hakaru-dev/hakaru-rktjit/blob/master/ast.rkt the expansion is in a comment and the transformation is using create-rpass

I will do the performance comparison and ablation study this weekend.

I am also available both Monday 11am and today between 10:30am and 1pm

I have a meeting at 11:30 or so, otherwise I’m available then

Let’s talk today at 10:45–11:30 to start

https://github.com/hakaru-dev/hakaru-rktjit/blob/master/pass/simplifications.rkt#L79 for a use of create-rpass