Tim Reichelt

icon indicating a website link treigerm.github.io

icon company University of Oxford icon location Postdoc at the University of Oxford

Results 19 comments of Tim Reichelt

Implementation of DCC inference algorithm

I have now added tests and documentation. Please let me know if you think anything is missing! After this PR is merged my plan would be to add the [SDVI...

Implementation of DCC inference algorithm

Thanks, @fehiepsi ! I'm currently unsure why the tests are failing, it seems to be an import error that it is not able to find the new `numpyro.contrib.stochastic_support.dcc` module. For...

Implementation of DCC inference algorithm

Ah yes of course! I have added an `__init__` file now. I also added the `XLA_FLAGS="--xla_force_host_platform_device_count=2"` flag to the `contrib` module tests to make the `parallel` chain sampling method pass....

[FR] Add inference method for programs with stochastic support

Thank you for the response! Yes, I think there is a reasonably large space of the programs with stochastic support which can be succinctly described in Pyro and for which...

[FR] Add inference method for programs with stochastic support

Yes, will do! I will open a draft PR with a simple bare bones implementation of DCC so that we have a starting point to discuss design decisions. I am...

GroupNorm missing from NNX normalization layer

Great, will look at the other layers and will send a PR by the end of this week.

Functional interface for error metrics

Thanks a lot!

Relative error bound results in out-of-bounds access after zfp_stream_maximum_size

Hello, thanks so much for your help! I'm still having trouble getting the relative error bound functionality to work but I might be using the API wrong. I'm running ZFP...

[Bug] Memory leak when using ExactGP module

This does indeed fix the memory bug in the example above. So adjusting the loss function to: ``` def loss(y): inputs = torch.randn(20, 2, device="cuda:2") # Option 1: likelihood =...