Similar repositories to dogeplusplus/flax-u2net:

• dogeplusplus/flax-u2net github similar
• danistefanovic/build-your-own-x github similar
• google-research/runtime-error-prediction github similar
• khdlr/augmax github similar
• 8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen github similar
• cgarciae/jax_metrics github similar
• deepmind/dks github similar
• lucidrains/flash-attention-jax github similar
• borisdayma/vit-vqgan github similar
• jax-ml/oryx github similar
• google/flaxformer github similar
• lucidrains/attention github similar
• DarshanDeshpande/jax-models github similar
• deepmind/jaxline github similar
• lucidrains/x-unet github similar
• google/CommonLoopUtils github similar
• lucidrains/PaLM-jax github similar
• matthias-wright/flaxmodels github similar
• google/sympy2jax github similar
• RobertTLange/evosax github similar
• cgarciae/treex github similar
• RobertTLange/gymnax github similar
• google/jaxtyping github similar
• ml-jku/hopular github similar
• deepmind/dm_pix github similar
• vopani/jaxton github similar
• karpathy/lecun1989-repro github similar
• deepmind/distrax github similar
• blackjax-devs/blackjax github similar
• deepmind/chex github similar
• patrick-kidger/equinox github similar
• google/evojax github similar
• google-research/big_vision github similar
• google/ml_collections github similar
• hardmaru/slimevolleygym github similar
• deepmind/xmanager github similar
• google-research/t5x github similar
• phlippe/uvadlc_notebooks github similar
• srush/Tensor-Puzzles github similar
• deepmind/optax github similar