Flax

Flax is a neural network library for JAX that is designed for flexibility.

Machine Learning Model TrainingML FrameworksOpen Source

About

Rilasciato nel 2024, Flax NNX è una nuova API Flax semplificata progettata per rendere più facile la creazione, l'ispezione, il debug e l'analisi delle reti neurali in JAX. Questo viene realizzato aggiungendo un supporto di prima classe per la semantica di riferimento di Python. Ciò consente agli utenti di esprimere i propri modelli utilizzando oggetti Python regolari, abilitando la condivisione di riferimenti e la mutabilità.

Open Source Health

Not enough history
Stars
7,317
Forks
833
License
Apache-2.0
Last commit
1 months ago
Jupyter Notebook

Resources & Links

Related Categories

Usi Flax?

Tieni traccia del suo costo insieme al resto del tuo stack e ricevi un promemoria prima del rinnovo.

Aggiungi al mio stack

Vendor

Google

Organize the world's information and make it universally accessible.

View vendor profile

More by Google