# Pathfinder running with gpu

**URL:** <https://discourse.pymc.io/t/pathfinder-running-with-gpu/17817>\
**Category:** General\
**Created:** [June 4, 2026, 2:52pm UTC](https://discourse.pymc.io/t/pathfinder-running-with-gpu/17817 "2026-06-04T14:52:23Z")\
**Posts on this page:** 3\
**Page:** 1

<div class="post-metadata">

**Author:** ![skalilopa](https://yyz2.discourse-cdn.com/flex036/user_avatar/discourse.pymc.io/skalilopa/32/9604_2.png) [@skalilopa](https://discourse.pymc.io/u/skalilopa)\
**Post date:** [June 4, 2026, 2:52pm UTC](https://discourse.pymc.io/t/pathfinder-running-with-gpu/17817/1 "2026-06-04T14:52:23Z")

</div>

Hello, I want to know if there is any new release of pathfinder running in GPU, because with pymc-extras still run on cpu.

Thanks.

---

<div class="post-metadata">

**Author:** ![jessegrabowski](https://yyz2.discourse-cdn.com/flex036/user_avatar/discourse.pymc.io/jessegrabowski/32/5010_2.png) [@jessegrabowski](https://discourse.pymc.io/u/jessegrabowski)\
**Post date:** [June 4, 2026, 3:00pm UTC](https://discourse.pymc.io/t/pathfinder-running-with-gpu/17817/2 "2026-06-04T15:00:24Z")

</div>

You can set `compile_kwargs={'mode':'JAX'}` to put the log\_dlogp function on GPU. The actual algorithm will be run on CPU, so you will pay the cost of going back and forth. blackjax offers are pure jax implementation of the end-to-end pathfinder algorithm, so you can bring a jax-compiled logp\_dlogp function over there and get the whole thing on GPU. Full disclosure, I have never tried the blackjax route so I don’t know the API. I know @junpenglao has been upgrading it recently to include multi-path, so that’s cool.

---

<div class="post-metadata">

**Author:** ![skalilopa](https://yyz2.discourse-cdn.com/flex036/user_avatar/discourse.pymc.io/skalilopa/32/9604_2.png) [@skalilopa](https://discourse.pymc.io/u/skalilopa)\
**Post date:** [June 4, 2026, 3:02pm UTC](https://discourse.pymc.io/t/pathfinder-running-with-gpu/17817/3 "2026-06-04T15:02:30Z")

</div>

thank you for your response.