Similar repositories to sebamenabar/meta-learning-representations-jax:

• sebamenabar/meta-learning-representations-jax github similar
• cgarciae/flax-tools github similar
• alexub/jax-meta-learning github similar
• gcucurull/maml_flax github similar
• tristandeleu/metax github similar
• nestordemeure/jochastic github similar
• deepmind/jax github similar
• google/orbax github similar
• tristandeleu/jax-comln github similar
• google-research/jax-influence github similar
• brentyi/jax_dataclasses github similar
• google/paxml github similar
• afqueiruga/StatefulOdeNets github similar
• jax-ml/oryx github similar
• jax-ml/jax-triton github similar
• openxla/xla github similar
• ucl-bug/jaxdf github similar
• cgarciae/treeo github similar
• jeremiecoullon/SGMCMCJax github similar
• IvanYashchuk/jax-fenics-adjoint github similar
• google/trajax github similar
• mitmath/18337 github similar
• deepmind/jmp github similar
• lucidrains/attention github similar
• ericjang/nf-jax github similar
• deepmind/jax_verify github similar
• google/CommonLoopUtils github similar
• brentyi/jaxfg github similar
• google/tensorstore github similar
• lucidrains/PaLM-jax github similar
• google/learned_optimization github similar
• brentyi/jaxlie github similar
• awf/functional-transformer github similar
• matthias-wright/flaxmodels github similar
• google/fiddle github similar
• ott-jax/ott github similar
• google/sympy2jax github similar
• cgarciae/treex github similar
• mpi4jax/mpi4jax github similar
• RobertTLange/gymnax github similar