Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Truncated normal dispatches #7506

Open
wants to merge 6 commits into
base: main
Choose a base branch
from
Open

Add Truncated normal dispatches #7506

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
e797cd9
Add jax dispatch for truncated normal
HarshvirSandhu Sep 17, 2024
8a9f17e
Add test for jax dispatch
HarshvirSandhu Sep 17, 2024
fd3b6d2
Fix jax dispatch for TruncatedNormal
HarshvirSandhu Sep 19, 2024
a7cb67b
Fix ruff error
HarshvirSandhu Sep 19, 2024
1de633c
Fix check-no-tests-are-ignored error
HarshvirSandhu Sep 19, 2024
b31dc5d
Fix failing test and parametrize test
HarshvirSandhu Oct 12, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix ruff error
  • Loading branch information
@HarshvirSandhu
HarshvirSandhu committed Sep 19, 2024
commit a7cb67bddff2aa8c6f6e397ba8fa3e01151e1641
5 changes: 3 additions & 2 deletions tests/dispatch/test_jax.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,12 @@
# limitations under the License.
import numpy as np
import pytest
from pymc.dispatch import dispatch_jax

from pytensor import function
import pymc as pm

import pymc as pm

from pymc.dispatch import dispatch_jax # noqa: F401

jax = pytest.importorskip("jax")

Expand Down
Loading