How to write a PyTensor Op to wrap Jax ODEs with multiple input parameters
|
|
7
|
794
|
July 28, 2024
|
Jax-Metal pm.Sample with numpyro on M1/M2
|
|
1
|
168
|
June 12, 2024
|
New distribution for PyMC: alpha-stable family [beta]
|
|
5
|
197
|
May 28, 2024
|
Recent error from Jax's (numpy) clip method
|
|
2
|
236
|
May 21, 2024
|
Jax sampling for bayesian neural network
|
|
3
|
162
|
March 27, 2024
|
Can jax be used to wrap a function that contained python libraries?
|
|
2
|
262
|
December 25, 2023
|
How to resolve Input Dimension Mis-match Error in Hierarchical Bayesian Inference with PyMC3
|
|
3
|
296
|
December 20, 2023
|
pm.Categorical with sample_numpyro_nuts
|
|
3
|
337
|
November 24, 2023
|
Dynamic shaping, "round" function, JAX, and a "few" more questions
|
|
28
|
1467
|
September 29, 2023
|
JAX Sampling Error with TruncatedNormal Distribution
|
|
6
|
893
|
July 27, 2023
|
Numpyro JAX sampling very slow
|
|
1
|
1308
|
July 9, 2023
|
Jax on Apple Silicon GPU
|
|
2
|
1844
|
June 13, 2023
|
NameError: unbound axis name raised during transformation of variables after sample_numpyro_nuts
|
|
6
|
898
|
June 13, 2023
|
Wild results for masked multinomial with jax sampler(s)
|
|
2
|
369
|
May 7, 2023
|
Pymc3 on GPU using jax
|
|
2
|
1469
|
April 20, 2023
|
Out of Memory when using pm.sampling.jax.sample_blackjax_nuts
|
|
2
|
826
|
March 23, 2023
|
How can I output a gradient in vector format in Op.grad instance?
|
|
7
|
850
|
January 14, 2023
|
How to use JAX ODEs and Neural Networks in PyMC
|
|
0
|
386
|
January 4, 2023
|
Numpyro Convergence Diagnostics
|
|
1
|
586
|
December 30, 2022
|
Out of memory when "transforming variables" in Numpyro & JAX
|
|
10
|
1187
|
December 12, 2022
|
Sample_numpyro_nuts hangs when parallelizing over datasets with multiprocessing
|
|
0
|
639
|
November 9, 2022
|
JAX Sampling Error when same model had previously worked without it
|
|
1
|
354
|
October 25, 2022
|
Slow inference for numpyro sampling on Colab GPU
|
|
3
|
823
|
October 5, 2022
|
Var_names not working with sample_numpyro_nuts
|
|
5
|
556
|
September 12, 2022
|
Pm.sampling_jax to sample a MvNormal()
|
|
4
|
706
|
August 3, 2022
|
Open all NUTS kwargs for sampling with Numpyro
|
|
5
|
685
|
August 1, 2022
|
Use of numpyro/Jax with pymc-dev
|
|
5
|
2057
|
July 28, 2022
|
Implementation of generalized linear mixed in the form X\beta + Zb
|
|
3
|
604
|
July 27, 2022
|
Gradients from external model in likelihood
|
|
3
|
647
|
July 1, 2022
|
How to debug a model that is not sampling?
|
|
0
|
628
|
June 30, 2022
|