eBay is interested. FLAX, a high-level API on JAX provides a basic mixed precision training API. More info here: https://github.com/google/flax/discussions/2027
You won't be notified about changes to this idea.