Deep Learning Frameworks - Public

[JAX] AMP support

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

  • Guest
  • Jul 1 2022
  • Planned
  • Attach files