Tom Andersson

Memory leak when drawing many tfd.GaussianProcess samples?

3

When instantiating many 2-dimensional GPs with exponentiated quadratic kernels and sampling over several thousand points, I'm getting memory errors: `ResourceExhaustedError: failed to allocate memory [Op:Mul]` I am able to produce...

Control resolution of rasterized plots

9

I'm using `ds.hvplot.quadmesh` to project and visualise a heatmap of geospatial data. Due to the size of the data, I have to set `rasterize=True`. As far as I'm aware (and...

NaNs in 6-hourly analysis-ready dataset for 2m temperature`

4

Hi there! I've come across NaNs in the `2m_temperature` variable in the 6-hourly analysis-ready dataset -- MWE below -- does this reproduce for you? Three strange observations: - I've tried...

Thanks for this fantastic initiative, it's great to be able to access ERA5 data straight into RAM with just a few lines of code. Currently only an hourly, instantaneous time...

[FR] Ability to draw cheap AR samples with subset-condition-predict procedure with `nps.ar_predict`

4

In our [AR CNP paper in ICLR](https://openreview.net/forum?id=OAsXFPBfTBh), we describe some ways to make AR sampling cheaper through a subset-AR sample-condition-predict procedure. For example, in Appendix K: > The AR samples...

Currently, calling `B.jit` on a `logpdf`-based objective works for a model with a single context set and single target set. However, a `ValueError` is raised in the case where a...

[FR] Allow `nps.loglik`, `nps.ar_loglik` and `nps.ar_predict` to work with numpy arrays

3

Currently, with a TensorFlow backend, I believe all the array inputs to `nps.loglik`, `nps.ar_loglik` and `nps.ar_sample` need to be TensorFlow tensors, e.g. `tf.EagerTensor`s. It would be convenient if users with...

[Feature]: Hyperparameter sliders for logged media

3

### Description I'm doing a grid search where the primary output is a `matplotlib` figure, rather than a scalar value like loss or accuracy. To be specific, I'm searching over...