Optax

Optax funge da libreria di elaborazione e ottimizzazione dei gradienti specificamente progettata per JAX, facilitando la ricerca fornendo blocchi modulari che possono essere facilmente combinati in ottimizzatori personalizzati. Mira a migliorare la produttività della ricerca con implementazioni efficienti di componenti fondamentali, consentendo agli utenti di creare soluzioni su misura. La libreria include vari ottimizzatori e funzioni di perdita popolari, rendendola adatta per applicazioni di apprendimento automatico.