1 paper · 1 filter
Alexander Gräfe, Sebastian Trimpe
Mixed-precision training has emerged as an indispensable tool for enhancing the efficiency of neural network training in recent years. Concurrently, JAX has grown in popularity as…