Metadata-Version: 2.4
Name: lammps-jax-pjrt-cpu
Version: 0.11.2
Summary: XLA CPU PJRT plugin for pair_style jax/kk on host Kokkos builds of LAMMPS
License-Expression: Apache-2.0
Description-Content-Type: text/markdown
License-File: LICENSE
License-File: THIRD_PARTY_NOTICES
Requires-Dist: jax<=0.11.2
Dynamic: license-file

# lammps-jax-pjrt-cpu

XLA's CPU PJRT plugin for [lammps-jax](https://github.com/abhijeetgangan/lammps-jax)
on host Kokkos builds of LAMMPS, built from the XLA of the jaxlib with the same
version; it runs bundles from that jax or older.

```bash
pip install lammps-jax-pjrt-cpu
python -c 'import lammps_jax_pjrt_cpu as p; print(p.PLUGIN_PATH)'
```
