Similar repositories to google/jax-md:

• google/jax-md github similar
• danistefanovic/build-your-own-x github similar
• torchmd/torchmd github similar
• deepmind/jraph github similar
• google/jaxopt github similar
• deepmind/optax github similar
• mir-group/nequip github similar
• atomistic-machine-learning/schnetpack github similar
• e3nn/e3nn github similar
• aiqm/torchani github similar
• hanxiao/bert-as-service github similar
• tootsuite/mastodon github similar
• tuvtran/project-based-learning github similar
• deepmind/chex github similar
• mitmath/18337 github similar
• openmm/openmm github similar
• deepmind/dm-haiku github similar
• patrick-kidger/diffrax github similar
• deepmodeling/deepmd-kit github similar
• lab-ml/annotated_deep_learning_paper_implementations github similar
• deepmind/ferminet github similar
• mpi4jax/mpi4jax github similar
• pyro-ppl/numpyro github similar
• google/jax-cfd github similar
• patrick-kidger/equinox github similar
• google/neural-tangents github similar
• google/objax github similar
• FabianFuchsML/se3-transformer-public github similar
• torchmd/mdgrad github similar
• deepmind/rlax github similar
• pandas-profiling/pandas-profiling github similar
• meilisearch/MeiliSearch github similar
• n2cholas/awesome-jax github similar
• MDAnalysis/mdanalysis github similar
• deepqmc/deepqmc github similar
• tum-pbs/PhiFlow github similar
• pyscf/pyscf github similar
• google-research/torchsde github similar
• cgnorthcutt/cleanlab github similar
• google/brax github similar