Similar repositories to deepmind/jmp:

• deepmind/jmp github similar
• danistefanovic/build-your-own-x github similar
• NVIDIA/TRTorch github similar
• hanxiao/bert-as-service github similar
• lab-ml/annotated_deep_learning_paper_implementations github similar
• google/CommonLoopUtils github similar
• deepmind/jaxline github similar
• lab-ml/labml github similar
• rwightman/efficientnet-jax github similar
• deepmind/distrax github similar
• deepmind/chex github similar
• google/tree-math github similar
• deepmind/dm_pix github similar
• deepmind/tensor_annotations github similar
• deepmind/kfac-jax github similar
• google/jaxtyping github similar
• google/etils github similar
• mitmath/18337 github similar
• cgnorthcutt/cleanlab github similar
• deepmind/jax_verify github similar
• google-research/vdm github similar
• dfm/extending-jax github similar
• cgarciae/treex github similar
• gehring/fax github similar
• SymJAX/SymJAX github similar
• pandas-profiling/pandas-profiling github similar
• mpi4jax/mpi4jax github similar
• google/flaxformer github similar
• deepmind/optax github similar
• brentyi/jaxfg github similar
• deepmind/einshape github similar
• coax-dev/coax github similar
• PrincetonLIPS/randomized_telescopes github similar
• kingoflolz/swarm-jax github similar
• 4rtemi5/imax github similar
• ku2482/rljax github similar
• n2cholas/awesome-jax github similar
• IvanYashchuk/jax-fenics-adjoint github similar
• google/jaxopt github similar
• deepmind/jax github similar