Deep Learning Frameworks - Public

Add a new idea

Filter by status

1 Vote
- Add my vote +1
- +2
- +3
- +4
- +5
- +6
- +7
- +8
- +9
- +10
[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

Created 01 Jul 05:45

0 Planned