Similar repositories to deepmind/dm-haiku:

• deepmind/dm-haiku github similar
• google/flax github similar
• deepmind/optax github similar
• danistefanovic/build-your-own-x github similar
• deepmind/rlax github similar
• patrick-kidger/equinox github similar
• pytorch/functorch github similar
• deepmind/jraph github similar
• google/jax github similar
• n2cholas/awesome-jax github similar
• hanxiao/bert-as-service github similar
• deepmind/chex github similar
• pyro-ppl/numpyro github similar
• google/brax github similar
• deepmind/acme github similar
• huggingface/accelerate github similar
• google-research/scenic github similar
• patrick-kidger/diffrax github similar
• arogozhnikov/einops github similar
• openai/triton github similar
• pandas-profiling/pandas-profiling github similar
• aqlaboratory/openfold github similar
• google/evojax github similar
• huggingface/diffusers github similar
• google/jaxopt github similar
• tuvtran/project-based-learning github similar
• poets-ai/elegy github similar
• google-research/t5x github similar
• google/jaxtyping github similar
• microsoft/DeepSpeed github similar
• deepmind/deepmind-research github similar
• pixijs/pixi.js github similar
• google/trax github similar
• deepmind/alphafold github similar
• kingoflolz/mesh-transformer-jax github similar
• deepmind/mujoco github similar
• lucidrains/x-transformers github similar
• facebookresearch/metaseq github similar
• deepmind/mctx github similar
• karpathy/micrograd github similar