Search results

Do not use "fork" as default mp_ctx when compiling JAX functions in the PyMC sampler

jax

samplers

Status: Open.

#7668 In pymc-devs/pymc;

· ricardoV94 opened on Jan 31, 2025

Add Truncated normal dispatches

jax

Status: Open (in progress).

pymc-devs/pymcnumber 7506

#7506 In pymc-devs/pymc;

· HarshvirSandhu opened on Sep 17, 2024

ENH: Support foward sampling via JAX

feature request

hackathonSuitable for hackathonSuitable for hackathon

jax

Status: Open.

#7348 In pymc-devs/pymc;

· jessegrabowski opened on Jun 4, 2024

ENH: Get sampling working using Apple Silicon GPU via jax backend

jax

macOSmacOS relatedmacOS related

Status: Open.

#7332 In pymc-devs/pymc;

· drbenvincent opened on May 23, 2024

Refactor/Upgrade <code>find_MAP</code>

feature request

jax

numba

request discussion

Status: Open.

#7308 In pymc-devs/pymc;

· jessegrabowski opened on May 12, 2024

Revert Temporarily pin Scipy in JAX job #7255

Github CI/CD

jax

maintenance

Status: Open.

#7256 In pymc-devs/pymc;

· ricardoV94 opened on Apr 15, 2024

Make VI compatible with JAX backend

jax

VIVariational InferenceVariational Inference

Status: Open (in progress).

pymc-devs/pymcnumber 7103

#7103 In pymc-devs/pymc;

· ferrine opened on Jan 15, 2024

Refactor jax internals to support dense_mass kwarg for numpyro

enhancements

jax

Status: Draft (not ready).

pymc-devs/pymcnumber 7050

#7050 In pymc-devs/pymc;

· ferrine opened on Dec 6, 2023

Numpyro keep warmup

enhancements

jax

request discussion

Status: Open (in progress).

pymc-devs/pymcnumber 6875

#6875 In pymc-devs/pymc;

· waldie11 opened on Aug 23, 2023

<code>Interpolated</code> not supported in JAX/Numba backends

feature request

jax

numba

Status: Open.

#6838 In pymc-devs/pymc;

· FBruzzesi opened on Jul 30, 2023

Use same API for defining internal and external Nuts kwargs

bug

help wanted

jax

Status: Open.

#6757 In pymc-devs/pymc;

· ricardoV94 opened on Jun 7, 2023

<code>var_names</code> in <code>sample_numpyro_nuts</code> not working

bug

jax

Status: Open.

#6116 In pymc-devs/pymc;

· rtdew1 opened on Sep 9, 2022