Similar repositories to google/neural-tangents:

• google/neural-tangents github similar
• danistefanovic/build-your-own-x github similar
• deepmind/dm-haiku github similar
• google/flax github similar
• deepmind/optax github similar
• pandas-profiling/pandas-profiling github similar
• cvxgrp/cvxpylayers github similar
• deepmind/jraph github similar
• DiffEqML/torchdyn github similar
• pyro-ppl/numpyro github similar
• google-research/torchsde github similar
• cornellius-gp/gpytorch github similar
• pytorch/functorch github similar
• google/jaxopt github similar
• tancik/fourier-feature-networks github similar
• facebookresearch/higher github similar
• google/uncertainty-baselines github similar
• hanxiao/bert-as-service github similar
• patrick-kidger/equinox github similar
• tuvtran/project-based-learning github similar
• arogozhnikov/einops github similar
• n2cholas/awesome-jax github similar
• rtqichen/torchdiffeq github similar
• tomgoldstein/loss-landscape github similar
• deepmind/chex github similar
• deepmind/rlax github similar
• Zymrael/awesome-neural-ode github similar
• pytorch/botorch github similar
• google/edward2 github similar
• patrick-kidger/diffrax github similar
• tristandeleu/pytorch-meta github similar
• google/objax github similar
• facebookresearch/nevergrad github similar
• deepmind/distrax github similar
• cgnorthcutt/cleanlab github similar
• bayesiains/nflows github similar
• google/jax-md github similar
• yang-song/score_sde github similar
• ml-jku/hopfield-layers github similar
• tensorflow/probability github similar