Hacker GitHub

← All repos

pyro-ppl/numpyro

View on GitHub ↗ ★ 2,777 Python

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

6 mentions on Hacker News by 6 people: 3 stories, 3 comments · first 2019-06-03 · latest 2024-11-16

Mentions per year 6 mentions · busiest year 2019 (2)
20082026
AllStoriesComments NewestTopOldest
  1. …library that runs on jax which makes it easy to put on the gpu. They have a nice suite of examples. https://github.com/pyro-ppl/numpyro
    comment by sito42 · 2024-02-10 · in Bayesian Analysis with Python
  2. …compiled; and it can run on GPUs. So that might be helpful for your problem with loads of data. [1] https://github.com/pyro-ppl/numpyro
    comment by jeremiecoullon · 2021-02-24 · in Saving the World with Bayesian Modeling
  3. …( https://github.com/deepmind/rlax ) * NumPyro, a probabilistic programming library on top of JAX ( https://github.com/pyro-ppl/numpyro ) * JAX-MD, a differentiable molecular dynamics package built on top of JAX (…
    comment by emu · 2020-04-08 · in Jax – Composable transformations of Python and NumPy programs