Skip to content

Request: PyPI release for jax 0.9.x support #247

@m9h

Description

@m9h

Hi maintainers,

The current PyPI release of jwave (0.2.1, from September 2024) depends on jaxdf <0.3.0, which in turn caps jax <0.5.0. This prevents installing jwave from PyPI in environments running modern JAX (0.9.x).

However, jaxdf 0.3.0 — which supports jax 0.9.x — is already available on PyPI, and the jwave main branch already depends on jaxdf >=0.3.0 and works with jax 0.9.x. So it seems like the only missing piece is a new PyPI release of jwave itself.

Would it be possible to cut a new release (e.g. 0.3.0) to PyPI from the current main branch?

Our use case: we maintain openlifu-python, a library for transcranial focused ultrasound simulation that depends on jwave. We need jax 0.9.x because it is the earliest version providing CUDA support on aarch64 (NVIDIA GH/GB-series platforms), and the jax <0.5.0 cap from the current PyPI release blocks us from using jwave in that environment.

Thank you for your work on jwave — happy to help test a pre-release if that would be useful.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions