Similar repositories to deepmind/distrax:

• deepmind/distrax github similar
• danistefanovic/build-your-own-x github similar
• deepmind/chex github similar
• blackjax-devs/blackjax github similar
• google/jaxtyping github similar
• google/jaxopt github similar
• pandas-profiling/pandas-profiling github similar
• deepmind/optax github similar
• hanxiao/bert-as-service github similar
• lab-ml/annotated_deep_learning_paper_implementations github similar
• patrick-kidger/equinox github similar
• ChrisWaites/jax-flows github similar
• deepmind/mctx github similar
• deepmind/jmp github similar
• rlouf/mcx github similar
• deepmind/rlax github similar
• deepmind/jraph github similar
• deepmind/dm_pix github similar
• cgarciae/treex github similar
• mpi4jax/mpi4jax github similar
• deepmind/jaxline github similar
• lab-ml/labml github similar
• thomaspinder/GPJax github similar
• n2cholas/awesome-jax github similar
• dfm/extending-jax github similar
• patrick-kidger/diffrax github similar
• RobertTLange/gymnax github similar
• google/CommonLoopUtils github similar
• google/tree-math github similar
• jeremiecoullon/SGMCMCJax github similar
• Information-Fusion-Lab-Umass/NuX github similar
• poets-ai/elegy github similar
• pyro-ppl/numpyro github similar
• google/objax github similar
• SymJAX/SymJAX github similar
• mackelab/sbi github similar
• deepmind/dm-haiku github similar
• jacobjinkelly/easy-neural-ode github similar
• AaltoML/BayesNewton github similar
• cgnorthcutt/cleanlab github similar