|
Wild results for masked multinomial with jax sampler(s)
|
|
2
|
56
|
May 7, 2023
|
|
NameError: unbound axis name raised during transformation of variables after sample_numpyro_nuts
|
|
5
|
291
|
May 5, 2023
|
|
Pymc3 on GPU using jax
|
|
2
|
235
|
April 20, 2023
|
|
Out of Memory when using pm.sampling.jax.sample_blackjax_nuts
|
|
2
|
165
|
March 23, 2023
|
|
How to write a PyTensor Op to wrap Jax ODEs with multiple input parameters
|
|
2
|
125
|
March 16, 2023
|
|
How can I output a gradient in vector format in Op.grad instance?
|
|
7
|
289
|
January 14, 2023
|
|
How to use JAX ODEs and Neural Networks in PyMC
|
|
0
|
113
|
January 4, 2023
|
|
Numpyro Convergence Diagnostics
|
|
1
|
168
|
December 30, 2022
|
|
Out of memory when "transforming variables" in Numpyro & JAX
|
|
10
|
565
|
December 12, 2022
|
|
Sample_numpyro_nuts hangs when parallelizing over datasets with multiprocessing
|
|
0
|
207
|
November 9, 2022
|
|
JAX Sampling Error with TruncatedNormal Distribution
|
|
2
|
291
|
October 27, 2022
|
|
JAX Sampling Error when same model had previously worked without it
|
|
1
|
125
|
October 25, 2022
|
|
Slow inference for numpyro sampling on Colab GPU
|
|
3
|
358
|
October 5, 2022
|
|
Var_names not working with sample_numpyro_nuts
|
|
5
|
216
|
September 12, 2022
|
|
Pm.sampling_jax to sample a MvNormal()
|
|
4
|
337
|
August 3, 2022
|
|
Open all NUTS kwargs for sampling with Numpyro
|
|
5
|
250
|
August 1, 2022
|
|
Use of numpyro/Jax with pymc-dev
|
|
5
|
953
|
July 28, 2022
|
|
Implementation of generalized linear mixed in the form X\beta + Zb
|
|
3
|
220
|
July 27, 2022
|
|
Gradients from external model in likelihood
|
|
3
|
218
|
July 1, 2022
|
|
How to debug a model that is not sampling?
|
|
0
|
237
|
June 30, 2022
|
|
JAX Numpyro backend "IndentationError: unexpected indent"
|
|
16
|
670
|
June 29, 2022
|