How to write a PyTensor Op to wrap Jax ODEs with multiple input parameters
|
|
7
|
798
|
July 28, 2024
|
Jax-Metal pm.Sample with numpyro on M1/M2
|
|
1
|
181
|
June 12, 2024
|
New distribution for PyMC: alpha-stable family [beta]
|
|
5
|
203
|
May 28, 2024
|
Recent error from Jax's (numpy) clip method
|
|
2
|
240
|
May 21, 2024
|
Jax sampling for bayesian neural network
|
|
3
|
168
|
March 27, 2024
|
Can jax be used to wrap a function that contained python libraries?
|
|
2
|
265
|
December 25, 2023
|
How to resolve Input Dimension Mis-match Error in Hierarchical Bayesian Inference with PyMC3
|
|
3
|
300
|
December 20, 2023
|
pm.Categorical with sample_numpyro_nuts
|
|
3
|
342
|
November 24, 2023
|
Dynamic shaping, "round" function, JAX, and a "few" more questions
|
|
28
|
1471
|
September 29, 2023
|
JAX Sampling Error with TruncatedNormal Distribution
|
|
6
|
896
|
July 27, 2023
|
Numpyro JAX sampling very slow
|
|
1
|
1329
|
July 9, 2023
|
Jax on Apple Silicon GPU
|
|
2
|
1857
|
June 13, 2023
|
NameError: unbound axis name raised during transformation of variables after sample_numpyro_nuts
|
|
6
|
900
|
June 13, 2023
|
Wild results for masked multinomial with jax sampler(s)
|
|
2
|
370
|
May 7, 2023
|
Pymc3 on GPU using jax
|
|
2
|
1478
|
April 20, 2023
|
Out of Memory when using pm.sampling.jax.sample_blackjax_nuts
|
|
2
|
830
|
March 23, 2023
|
How can I output a gradient in vector format in Op.grad instance?
|
|
7
|
852
|
January 14, 2023
|
How to use JAX ODEs and Neural Networks in PyMC
|
|
0
|
386
|
January 4, 2023
|
Numpyro Convergence Diagnostics
|
|
1
|
590
|
December 30, 2022
|
Out of memory when "transforming variables" in Numpyro & JAX
|
|
10
|
1193
|
December 12, 2022
|
Sample_numpyro_nuts hangs when parallelizing over datasets with multiprocessing
|
|
0
|
646
|
November 9, 2022
|
JAX Sampling Error when same model had previously worked without it
|
|
1
|
355
|
October 25, 2022
|
Slow inference for numpyro sampling on Colab GPU
|
|
3
|
825
|
October 5, 2022
|
Var_names not working with sample_numpyro_nuts
|
|
5
|
557
|
September 12, 2022
|
Pm.sampling_jax to sample a MvNormal()
|
|
4
|
707
|
August 3, 2022
|
Open all NUTS kwargs for sampling with Numpyro
|
|
5
|
691
|
August 1, 2022
|
Use of numpyro/Jax with pymc-dev
|
|
5
|
2080
|
July 28, 2022
|
Implementation of generalized linear mixed in the form X\beta + Zb
|
|
3
|
611
|
July 27, 2022
|
Gradients from external model in likelihood
|
|
3
|
649
|
July 1, 2022
|
How to debug a model that is not sampling?
|
|
0
|
629
|
June 30, 2022
|