Mike Dusenberry issues

Repositories
Issues
Comments

Results 4 issues of


                                            Mike Dusenberry

Add smoke tests for all TF-based training scripts.

Also reproducibility tests if we can do it with TF.

Update the design of weight sampling in the BNN layers.

TFP has been updated such that `tfp.distributions.*` objects can be initialized in a tape-safe manner. I.e., it's now possible to create a distribution within one tape, and use it in...

Improve robustness of ed RVs in mixed-precision settings.

Currently, ed RandomVariable sampling for reparameterized distributions can encounter errors in mixed-precision settings (such as bfloat16) due to a mismatch between the the dtype of output of the `sample`computation in...

[WIP] Adding surgery functions to convert models to fully-conv versions

This is a work-in-progress for a set of functions to convert our models to fully-convolutional versions in order to be able to make predictions on the full images for increased...