v0.1.3

Cutting a new release to support new Python versions

v0.1.2

Merge pull request #78 from jax-ml:backwards-xla

PiperOrigin-RevId: 508552584

v0.1.1

Merge pull request #6 from jax-ml/multidevice

Lazily load CUDA modules before launch