/flax-hybridization

Flax hybridization

Primary LanguageJupyter Notebook

Flax - Model Hybridization

An ML-model hybridization example written jax and flax.