piwheels

jax-md

Differentiable, Hardware Accelerated, Molecular Dynamics

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install jax-md

Dependencies

• jaxlib
• optax
• flax
• frozendict
• e3nn-jax
• absl-py
• einops
• numpy
• jraph
• pymatgen
• jax
• ml-collections

Releases

Version	Released	Bookworm Python 3.11	Trixie Python 3.13	Files
0.2.29	2026-06-10
jax_md-0.2.29-py3-none-any.whl (363 KB)
0.2.28	2026-03-22
jax_md-0.2.28-py3-none-any.whl (228 KB)
0.2.27	2025-11-07
0.2.26	2025-10-30
0.2.8	2023-08-09
jax_md-0.2.8-py3-none-any.whl (147 KB)
0.2.7	2023-07-24
jax_md-0.2.7-py3-none-any.whl (147 KB)
0.2.6	2023-07-24
jax_md-0.2.6-py3-none-any.whl (143 KB)
0.2.5	2022-11-28
jax_md-0.2.5-py3-none-any.whl (137 KB)
0.2.4	2022-11-27
jax_md-0.2.4-py3-none-any.whl (137 KB)
0.2.3	2022-10-30
jax_md-0.2.3-py3-none-any.whl (117 KB)
0.2.2	2022-10-28
jax_md-0.2.2-py3-none-any.whl (115 KB)
0.2.1	2022-10-09
jax_md-0.2.1-py3-none-any.whl (102 KB)
0.2.0	2022-08-19
jax_md-0.2.0-py3-none-any.whl (102 KB)
0.1.29	2022-07-18
jax_md-0.1.29-py3-none-any.whl (89 KB)
0.1.28	2022-05-02
jax_md-0.1.28-py3-none-any.whl (94 KB)
0.1.27	2022-04-29
jax_md-0.1.27-py3-none-any.whl (91 KB)
0.1.26	2022-04-23
jax_md-0.1.26-py3-none-any.whl (89 KB)
0.1.25	2022-03-30
jax_md-0.1.25-py3-none-any.whl (89 KB)
0.1.24	2022-01-21
jax_md-0.1.24-py3-none-any.whl (88 KB)
0.1.23	2022-01-15
jax_md-0.1.23-py3-none-any.whl (88 KB)
0.1.22	2021-12-01
jax_md-0.1.22-py3-none-any.whl (81 KB)
0.1.21	2021-12-01
jax_md-0.1.21-py3-none-any.whl (80 KB)
0.1.20	2021-10-29
jax_md-0.1.20-py3-none-any.whl (80 KB)
0.1.19	2021-09-24
jax_md-0.1.19-py3-none-any.whl (76 KB)
0.1.18	2021-09-08
jax_md-0.1.18-py3-none-any.whl (76 KB)
0.1.17	2021-08-17
jax_md-0.1.17-py3-none-any.whl (75 KB)
0.1.16	2021-06-26
jax_md-0.1.16-py2.py3-none-any.whl (71 KB)
0.1.15	2021-04-24
jax_md-0.1.15-py2.py3-none-any.whl (70 KB)
0.1.14	2021-04-22
jax_md-0.1.14-py2.py3-none-any.whl (70 KB)
0.1.13	2021-03-12
jax_md-0.1.13-py2.py3-none-any.whl (66 KB)
0.1.12	2021-02-05
jax_md-0.1.12-py2.py3-none-any.whl (65 KB)
0.1.11	2021-02-01
jax_md-0.1.11-py2.py3-none-any.whl (64 KB)
0.1.10	2020-11-24
jax_md-0.1.10-py2.py3-none-any.whl (62 KB)
0.1.9	2020-10-28
jax_md-0.1.9-py2.py3-none-any.whl (61 KB)
0.1.8	2020-09-27
jax_md-0.1.8-py2.py3-none-any.whl (53 KB)
0.1.7	2020-08-17
jax_md-0.1.7-py2.py3-none-any.whl (51 KB)
0.1.6	2020-06-11
jax_md-0.1.6-py2.py3-none-any.whl (51 KB)
0.1.5	2020-05-01
jax_md-0.1.5-py2.py3-none-any.whl (43 KB)
0.1.3	2020-04-15
jax_md-0.1.3-py2.py3-none-any.whl (41 KB)
0.1.2	2020-03-26
jax_md-0.1.2-py2.py3-none-any.whl (41 KB)
0.1.1	2020-02-14
jax_md-0.1.1-py2.py3-none-any.whl (41 KB)
0.1.0	2019-12-11
jax_md-0.1.0-py2.py3-none-any.whl (35 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue