def sample_from_activation(self, vmap): rate = self.rate_from_activation(vmap) return samplers.exponential(rate) # lambda = -activation!
def sample_from_activation(self, vmap): return samplers.exponential(-vmap[self]) # lambda = -activation!