Nice! Thanks for testing it out. Hopefully we can speed that up even more, this is just the first rough cut. Did you have any heavy matmuls in the model, or was it mostly just basic element-wise stuff?
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| PYMC 5 significant speedup of default sampler (Pytensor) | 4 | 1449 | January 25, 2023 | |
| PyMC runtime question (Jax, BLAS, aesara) | 1 | 825 | October 1, 2022 | |
| Is it possible to speed up PyMC sampling? | 3 | 3419 | May 20, 2022 | |
| Bambi is faster than my pure pymc implementation; what did I do wrong? | 2 | 1021 | July 19, 2022 | |
| Sampling time GPU vs CPU | 1 | 2933 | February 7, 2023 |