jax_md/__init__.py,sha256=Qe7JGV8Hjlnvw3umATXBIfp4XnCZ4jjv6C4N9aLtnsU,1015
jax_md/custom_partition.py,sha256=D8DonbHqiwWqFip452pKLf1t2R_ykG0MZKC60uc5xJI,42876
jax_md/custom_smap.py,sha256=cY6Ioh56VbGxrZVRzF2ETA6D5AeiLpi6vZJNpGJmOn8,11647
jax_md/dataclasses.py,sha256=BJgy1KvSojbxlVFCw8YY61oen5Vyyb9IZp1Ki3shzGM,3769
jax_md/elasticity.py,sha256=XdrDIn85cz8cXpOSCxFJXsOpAA9TmeEfDDHG_ElsZ5I,22313
jax_md/energy.py,sha256=6rtaABB1lG29PCywMnaQy8Q1_5oSlgpQcVDPPX9j22E,102262
jax_md/interpolate.py,sha256=AytAPT0HAWBPKKB0aho-eT6t89kDNaYE4LYJzsnbUjQ,2954
jax_md/minimize.py,sha256=aNYAafcCfAsh5y8kdTM-zxW_2PTO39OqRQHPRNjrebw,44554
jax_md/nn.py,sha256=Y4OFLmZwIWEhiOLybvLMyuHqSlqcqa0k_qzGZJZM0T4,11697
jax_md/partition.py,sha256=8hoQs--H9N_N-ZPRvPft6fVzQv0-ATaHODJpMR5zUMo,42811
jax_md/quantity.py,sha256=HJWMd4IiS21GyzUHu7o46QYNveBucs9rG5T8q-LPLoY,25347
jax_md/rigid_body.py,sha256=7lvNo9J9mAci_oQXlxMhFO25z5mOJx3R22mkRD5MR14,38497
jax_md/simulate.py,sha256=qxWqqewwNKJTF2uNcioLDz6T2HPnc-muM_5OJYi8xhg,60577
jax_md/smap.py,sha256=HVZom4wpcrBbEIlmPDTKSXdtsd3B8NxGuj4fW8wMOic,40368
jax_md/space.py,sha256=g_Tvz-ZjmuLJ9EgNo6seVApiQbIaojSQSjKT7Snq00Q,16708
jax_md/test_util.py,sha256=7JxwEvUcyP7f4q0WZd3JQJrW04PjT7W3830OoOoFKlY,12622
jax_md/tpu.py,sha256=DZXm6pZnYb-uDodgrG-FsQ4wVp4ZZ99gx4a-en5g2Tc,56724
jax_md/units.py,sha256=FF0QF9ZQRj2ZazpxUZFGA_FaSxZmz7NWycKrWSw82H4,4280
jax_md/util.py,sha256=mRYQTmv7MCZNymrIuujlctyqV4Uqrq60lvkqTQLcgtA,4164
jax_md/_energy/__init__.py,sha256=0wntVYIQ66zgOkRcd5ZztzkPRCULCpVsxJlX0BGz4Mc,606
jax_md/_energy/electrostatics.py,sha256=Ppki3Sh_v0y_Mn6umB0wm2K36-uwC_zoF1mRoOZEHkY,10327
jax_md/_nn/__init__.py,sha256=pZK3-4Dy-VJZ7PNBf0t7I0OjzV2CkaVhWZeGn7VGW_o,686
jax_md/_nn/behler_parrinello.py,sha256=L-HtSydre_vNUCS-Afu-kEnrpaiGszG_WWxSdDt0aqI,17837
jax_md/_nn/e3nn_layer.py,sha256=gjHf8WoknXMbpQ1p6M2w-Mg4_jfdF3vCPMwsGz0p2Ls,14797
jax_md/_nn/gnome.py,sha256=RFTsq7GMM7BpA8ZyuK2QZ9WNFXGevrMGDutwzOM0HgA,5121
jax_md/_nn/nequip.py,sha256=jWEfgzJ56VI89oV9L6JZZd8QUZxez7lMec1GYoFv_Xs,15590
jax_md/_nn/util.py,sha256=BaXm8sARZi7bMaQjNmu6BbRGusUn6IA-54sMZuVdW84,8446
jax_md/_nn/mace/__init__.py,sha256=FaerGtMVy4GZe8o4u0NcNP5TmJoWuqyBBnarSK9QEIs,31
jax_md/_nn/mace/featurizer.py,sha256=VVqkkSSFzwpapMVSDYKk0vx88cOuJG93own_VwLKCCU,9163
jax_md/_nn/mace/mace_jax_from_torch.py,sha256=-eVJfrrKj2-rkpoxp9cCd190NdSpnwPxDXZWnD5YxOQ,7572
jax_md/_nn/mace/model_builder.py,sha256=3X_4xCsx79zwrNRyLt2mEMSGrW4hIDuOfX3_8VzpVvU,13117
jax_md/_nn/uma/README.md,sha256=SjK-gOt3h4mE2P2vI7J5am8U4-n2k-1RrrZUNTEfKrs,6154
jax_md/_nn/uma/__init__.py,sha256=E8jBaxK7YXzXVuP5fgXbx5HO1tnLolPRO_sApZOZQso,1619
jax_md/_nn/uma/blocks.py,sha256=AAx6gU76jwelDkPLKQhYOXCRE97hZ7Am9KT6YSYDCME,13224
jax_md/_nn/uma/featurizer.py,sha256=2Dbq7LLgIJ_puLFDsGE2euGDWbsrwPkpWfVIMlcJqT8,6688
jax_md/_nn/uma/heads.py,sha256=2QVW-wEa_97ssM46KS15Bi2aemQK5JEPL2zB3gfpaC0,4302
jax_md/_nn/uma/kernels.py,sha256=etFx8YTIEIU7F6qlTAqN81Hrj7kQNnST5Gndci-3IEo,107599
jax_md/_nn/uma/model.py,sha256=Y5YPsog6WXxQIb9mnUcwYvg03sybszn2Z3v5LtIsiZ4,11066
jax_md/_nn/uma/model_moe.py,sha256=u2nMXO_UW5bJphrs8jERaiJWJu-nXTymFnBpmHSeVxE,44676
jax_md/_nn/uma/pretrained.py,sha256=LpjXWkWIagmwrj0QxhTx--X4FIeUzXHtlm1180YFFKY,21016
jax_md/_nn/uma/weight_conversion.py,sha256=AYtJBy1ZUnFOeD6mgEugo5Rj3dPsybXe9aWP1py7OTk,14338
jax_md/_nn/uma/Jd_npy/Jd_0.npy,sha256=rCmYCjl-UDqS5KmiMD32FZOmRWbjltTnvbi9jO9Mib8,132
jax_md/_nn/uma/Jd_npy/Jd_1.npy,sha256=dGXHfD1S9okdntKvfHcz82Hio3DYzZbvP-HiBgNxTbY,164
jax_md/_nn/uma/Jd_npy/Jd_2.npy,sha256=af7PuiR9-3zW-YqwZf7NkZUr46h1zlp8ExYcdMgxosw,228
jax_md/_nn/uma/Jd_npy/Jd_3.npy,sha256=hvTeMAl6KI6yLX-AV2MePzR40vHzAXj2luH4vlSOEdA,324
jax_md/_nn/uma/Jd_npy/Jd_4.npy,sha256=5FrVWLr-K_84U30ZgM0cQx225wE-qcTsrknHinjQoFs,452
jax_md/_nn/uma/common/__init__.py,sha256=ROu3KVXXs8ASmxyBGrmDyPwrEQZ8aYqwZRkIkyet4ss,465
jax_md/_nn/uma/common/rotation.py,sha256=hFB70nmWrN1OGUJJK5YttyplHyDJtCdQko2vKEiOFAc,11734
jax_md/_nn/uma/common/so3.py,sha256=4wG3LHA1d6tYAVV_E10OixcLt0wamNjiA0zgx7QcCJQ,13083
jax_md/_nn/uma/nn/__init__.py,sha256=bfOYfocq3b5whFZK1YcvPwEkoRtZH1lP9isl3Gu1bWY,946
jax_md/_nn/uma/nn/activation.py,sha256=KVmnLKTtiV5pRIB8kqvqTnRVNMHZXXIxyODB43AYsqI,5847
jax_md/_nn/uma/nn/embedding.py,sha256=gSp0ZRk-jWf8IBkuG_85tUZcaZ7N7Rxpr5x1Tt6J8Dg,8181
jax_md/_nn/uma/nn/layer_norm.py,sha256=WgDTWTo9k1bsU910JlZ3booe_vExb1rLWDzY5spzoKs,8652
jax_md/_nn/uma/nn/mole.py,sha256=6cK_S52gA1iGfqG4gakh8ADQDxWfADnjASVPc4VPP30,7561
jax_md/_nn/uma/nn/radial.py,sha256=ifDyhZefzT_LJRRAzrFY_BJQwMyE0FXf4KxBGJ7X7LE,3100
jax_md/_nn/uma/nn/so2_layers.py,sha256=MmfANEq_Serp7RICDYsJahXOYERppcYKQZBRcKwlZCA,7530
jax_md/_nn/uma/nn/so3_layers.py,sha256=dlWPynbGfavsqooYeT2odlGDNgHgOW6lrPIa2EOaP0k,2529
jax_md/a2c/__init__.py,sha256=zoK2kImtXgGbwh_U8LPxhiHnumTGU6CreWx9qxFggZI,68
jax_md/a2c/a2c_workflow.py,sha256=qQo1a5kpW_uWmLU7VHKQK6aLw1ORct8-FZoWVduoDwU,8051
jax_md/a2c/crystallizer_utils.py,sha256=cY8_L_hw82JE5oCtjQoEbaMpBHAmQuYbPD2xqPHRtoY,12426
jax_md/a2c/make_amorphous_utils.py,sha256=hjtulghdvjT3SGi1oLpMXP_KBxkvAQiExTsfF0gJ7CU,4567
jax_md/colab_tools/__init__.py,sha256=d-z7eJqmhtevSYf1uWNchdiJ427Mw3y40DiJUIKGLO8,616
jax_md/colab_tools/renderer.py,sha256=LApFSmyH5qLCZ5QOV577iQM_Oyttts1GF04sNvC36WU,9791
jax_md/colab_tools/visualization.html,sha256=J6uj-8DD_YwYa2ZZGE7NOYJeDCggit2JqX6pG_t-D2o,32973
jax_md/mm_forcefields/README.md,sha256=pKEF-LiOZrhqyON7VIIgcOJrUFjFdwXHd6j-5Kj-1dY,4464
jax_md/mm_forcefields/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
jax_md/mm_forcefields/base.py,sha256=0LX06nV-897McZ9gMza-0iMZCCkUVXwSQDo4wx34G4A,9455
jax_md/mm_forcefields/neighbor.py,sha256=eRl3DyrmDOG8A2aaRq46BlzVgKfnkJJHsdkmfcD2dXc,4073
jax_md/mm_forcefields/amber/__init__.py,sha256=90loOaM8sP55AM3f-aGDVBHDNi1mvzhnMKVrEUw0EYM,154
jax_md/mm_forcefields/amber/constraints.py,sha256=a8XCBpYK0lOwnbfvpXFS6xMOeFGIM0jvUTyu1IN4J98,19721
jax_md/mm_forcefields/amber/energy.py,sha256=On254dhNYxm_BQWm6BYzprrotCq6_GcvxYJZEheY154,45943
jax_md/mm_forcefields/io/__init__.py,sha256=BqeK_4lnK22v-wXbwxReUoMjRpVDD6nxQo1DvGXd5FI,430
jax_md/mm_forcefields/io/charmm.py,sha256=pIvz8BNSiBxeXfEWoK6KapVaHRJG-iNT3mDzfF4-6cQ,8908
jax_md/mm_forcefields/io/openmm.py,sha256=YJwfzuuSuVf4D3LPT4WlIUmYzRikL3y5fUUv6_4m3hE,44950
jax_md/mm_forcefields/nonbonded/__init__.py,sha256=MrYUH8IvQh2VnzercdDZRZc0JAYMQgFSJ-piMPgSM1c,320
jax_md/mm_forcefields/nonbonded/electrostatics.py,sha256=f-DYDh2ZUdV28nshR9-z2CM5unnytXyY9UcQgBuAfNY,26053
jax_md/mm_forcefields/oplsaa/__init__.py,sha256=-Sayw2W9AvJnQ83uOWWNvq_J2nYx517p6rJ6ANUv4ak,1706
jax_md/mm_forcefields/oplsaa/energy.py,sha256=53akJQ9LjT9iHrmOn6hEIspIBH42jCBc63kGXm4dVAM,8874
jax_md/mm_forcefields/oplsaa/io.py,sha256=S62v99RGsgQx3Hev08h2FJ-YLxkKUhWTQ39hXjHtIBY,8029
jax_md/mm_forcefields/oplsaa/params.py,sha256=V9-QXUlYabBzDGMkpFx3LBLq7QTONxyg1jX84aAP8lg,5083
jax_md/mm_forcefields/oplsaa/topology.py,sha256=XmHKOHJX8HDJVSAr4_yR83xPi0mk6Zvb8xuabzDNmV0,3479
jax_md/mm_forcefields/reaxff/__init__.py,sha256=iW2rphPhfoxpPZCd2ObToWGT1UHPq2HZX1lqZsNl6Ls,122
jax_md/mm_forcefields/reaxff/reaxff_energy.py,sha256=6QEAMJjFiJGzK6agHF8HJdKoIKDnysiOGub_nhmyT_c,41340
jax_md/mm_forcefields/reaxff/reaxff_forcefield.py,sha256=YE1IfazlOq4nehrvlkj79Vci9D2HPSFRy-M_C4Fsm20,8281
jax_md/mm_forcefields/reaxff/reaxff_helper.py,sha256=_G3A02_eOdWfI8VbZqFqQVcUd7rypMP_rpGJwPB5bs0,31108
jax_md/mm_forcefields/reaxff/reaxff_interactions.py,sha256=mIXcBZGl4phSI50zAI7Wk5DNjONsL7uindW1Mz3GRUk,27563
jax_md/_nn/uma/Jd.pt,sha256=tAWcRb4kbctsScVFZwtlxWVQ6wwuepyStLUKktNw2-I,21697
jax_md-0.2.29.dist-info/METADATA,sha256=hVL-YcKNcwv3gYjoLtcI_-Gy35z7jiwFKWYU1cFiqmo,21828
jax_md-0.2.29.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
jax_md-0.2.29.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
jax_md-0.2.29.dist-info/licenses/LICENSE_SHORT,sha256=vA2mla530c1P0vxmuvT0ikVi8TpDlP828TxphQQxV7I,574
jax_md-0.2.29.dist-info/RECORD,,
