jax-metrics pypi

This project isn't being tracked yet.