Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)
Primary LanguagePythonMIT LicenseMIT