Flax hybridization
Primary LanguageJupyter Notebook
An ML-model hybridization example written jax and flax.