Commits

Commits on Jul 8, 2026

Automated Code Change

h-joo
authored and
DistraxDev
committed
Automated Code Change

h-joo
authored and
DistraxDev
committed
Add pytype ignore statements related to deprecated JAX APIs

Jake VanderPlas
authored and
DistraxDev
committed

Commits on Jun 12, 2026

Release v0.1.9

hbq1
authored and
DistraxDev
committed
fix tests and drop python 3.11

hbq1
authored and
DistraxDev
committed

Commits on May 13, 2026

[JAX] Replace uses of jnp.float_, jnp.int_, and jnp.complex_ with float, int, and complex in .astype, jnp.zeros, and jnp.ones.

hawkinsp
authored and
DistraxDev
committed

Commits on Apr 18, 2026

Avoid usage of deprecated `jax.core` APIs.

Jake VanderPlas
authored and
DistraxDev
committed

Commits on Apr 16, 2026

Release v0.1.8

Jake Bruce
authored and
DistraxDev
committed

Commits on Apr 1, 2026

PR #36092: lax.mul: add out_dtype argument

jakevdp
authored and
DistraxDev
committed

Commits on Mar 24, 2026

Fix Distrax from Jax API change.

DistraxDev
authored and
DistraxDev
committed

Commits on Jan 23, 2026

[JAX] Remove stale references to //jax:experimental, which is being removed.

hawkinsp
authored and
DistraxDev
committed

Commits on Dec 9, 2025

Replace deprecated `jax.core.get_aval` with `jax.typeof`

Jake VanderPlas
authored and
DistraxDev
committed

Commits on Oct 11, 2025

Migrate from `jax.experimental.enable_x64` to `jax.enable_x64`.

Jake VanderPlas
authored and
DistraxDev
committed

Commits on Sep 1, 2025

release v0.1.7

hbq1
authored and
DistraxDev
committed
migrate to pyproject.toml

hbq1
authored and
DistraxDev
committed
release v0.1.6

hbq1
authored and
DistraxDev
committed
* migrate away from dm-tree

hbq1
authored and
DistraxDev
committed
Disable bad-ident warnings.

hbq1
authored and
DistraxDev
committed

Commits on Jul 10, 2025

remove `pjit_p`, leaving only `jax.extend.core.primitives.jit_p`

froystig
authored and
DistraxDev
committed

Commits on May 19, 2025

[distrax] Replace usages of deprecated `jax.interpreters.xla.abstractify`

Jake VanderPlas
authored and
DistraxDev
committed

Commits on May 2, 2025

Configure tests to explicitly use jax_threefry_partitionable=False.

jburnim
authored and
DistraxDev
committed

Commits on Apr 17, 2025

Configure tests to explicitly use jax_threefry_partitionable=False.

jburnim
authored and
DistraxDev
committed

Commits on Mar 28, 2025

Add accuracy field to unary ops

hanrach9
authored and
DistraxDev
committed

Commits on Feb 25, 2025

Fix wrong call signatures for jnp.shape, jnp.size, and jnp.ndim

Jake VanderPlas
authored and
DistraxDev
committed

Commits on Dec 14, 2024

Migrate from jax.core to jax.extend.core for several deprecated symbols

Jake VanderPlas
authored and
DistraxDev
committed

Commits on Dec 11, 2024

Remove references to jax.core.raise_to_shaped

Jake VanderPlas
authored and
DistraxDev
committed

Commits on Nov 25, 2024

Add sqrt/square primitives to the inverse registry after https://github.com/openxla/stablehlo/pull/2623

hbq1
authored and
DistraxDev
committed

Commits on Sep 18, 2024

Clean up and fix primal type to tangent type mapping

dougalm
authored and
DistraxDev
committed

Commits on Jun 27, 2024

Update distrax von_mises to use new jnp.clip syntax.

DistraxDev
authored and
DistraxDev
committed
Fix pylint errors found by the updated linter.

DistraxDev
authored and
DistraxDev
committed

Commits on Mar 5, 2024

The main tensorflow_probability BUILD target is being changed to only load the

ThomasColthurst
authored and
DistraxDev
committed

Commits on Feb 17, 2024

Automated Code Change

tomwardio
authored and
DistraxDev
committed

Commits on Dec 14, 2023

Futureproof distrax against the removal of jax.linear_util, which was deprecated in jax v0.4.16 and will be removed in jax v0.4.23

Jake VanderPlas
authored and
DistraxDev
committed

Commits on Nov 21, 2023

Install dependencies before running version checks in the release workflow.

hbq1
authored and
DistraxDev
committed

Commits on Nov 20, 2023

Install setuptools for python>=3.12.

hbq1
authored and
DistraxDev
committed