equitrain/__init__.py,sha256=Oz6e31T_18Gvs6gH4qB3pZBEAUlJl3m0FttQS_Yo2eY,3082
equitrain/__main__.py,sha256=LoYQTgEwjTAlSR3vA0NsDISo7r7oH62hMvDDaEx-OHQ,1136
equitrain/_version.py,sha256=Qr0d737dKW0t4NMk4ugtJLvxQ1nNW-TRLJD_Xq0w_6I,528
equitrain/argparser.py,sha256=3Bmd36N9-GKIy-lNsgN234_5BRrHIOq_N5-iszUWp6Q,31650
equitrain/checkpoint.py,sha256=u4F-4onViGdd2_X1RzPooqnOFkBLfHaw1Id1UD4Dvns,1225
equitrain/evaluate.py,sha256=su_2Yx7fh9WbZ_X6OLsXcVBrd65oKFOf7HFBZ12ZUS8,718
equitrain/logger.py,sha256=nOqwleahgsII8sl3sMSDKJ5aPiPU1MTTKTwIQI4YmBo,3728
equitrain/loss_metrics.py,sha256=8ArNL3GfTz2VJfkfiwWciZyxQxHuPP74PEpV44quz5E,521
equitrain/model_wrappers.py,sha256=dfwoJgKFUCAqmZJ5X96HkXqFCK6UFoWv8C1Dude0rkU,820
equitrain/predict.py,sha256=Kot1xXeSQ9Fd8IYcYp1p2ZX9JUNQbP4d1q8pHTcd69o,3302
equitrain/preprocess.py,sha256=5WPBrkWuKpbjbFATqBnn_49P83WAjpX8mmYwIV_pNwA,15377
equitrain/train.py,sha256=ZU_DTuqzWdXV4vYoHJIxvqVX2Ioe1JHgqvgN-M_lwQs,1452
equitrain/backends/__init__.py,sha256=xnQKSy3Q-SIbbcmEiA0CJ_lrELYOFQG-A2HZxCU0NyQ,619
equitrain/backends/jax_backend.py,sha256=YGGNVkV2mjf0uMBg33yaA8JGzQDqIkcPWXx2yQC200I,52010
equitrain/backends/jax_checkpoint.py,sha256=e9ICzqGFaGtYDr81LuFrSY9tJ84a6AK9XBzmhYTqkrQ,9299
equitrain/backends/jax_evaluate.py,sha256=Zk3apjJyXM2ncLZ5kTL2ftbGimkKlkPF7cYRucAl3kU,6953
equitrain/backends/jax_freeze.py,sha256=yNMPlscAl7z2Ld5uX_xmC_jnu4DIvTdKO2_Mdxwj4f4,2834
equitrain/backends/jax_loss.py,sha256=_IW1nspmzqlPEIaXekpuhj555uQ-_7S-TA9x_0mDeUQ,1874
equitrain/backends/jax_loss_fn.py,sha256=GYK_I_vxJqt3Qu7lnq8_74XtTheHRQoa1KprLq_h5OA,15712
equitrain/backends/jax_loss_metrics.py,sha256=3CY8hnuEgL4r3HJIRaSnEvZGtJzQb-7Rqj_ArW1pAvQ,4010
equitrain/backends/jax_nnx_compat.py,sha256=HoX9AUGiaQ0dW8H9mmH--HvEsAV-Rc9_yCWPPZus-ng,6358
equitrain/backends/jax_optimizer.py,sha256=fKLMYeNheTinXYUo7cA2Arn5PPj8R6pkixMtGlp-uoU,2119
equitrain/backends/jax_predict.py,sha256=M10GxXgELh64lHK7cMh-QEfhQ8mHxrinoIrrvsQ6x2k,18827
equitrain/backends/jax_runtime.py,sha256=xJF7YIxVy_o15wJsxOfszbJVT6fH2Lp27hi-2Xyi9As,671
equitrain/backends/jax_scheduler.py,sha256=I1Gp31g8TaXAu9DM0mHCx6Xhr50t0vMeWaJEuVBoiSA,5644
equitrain/backends/jax_utils.py,sha256=LPnlJ79IGI-wDDR_ujvWZ1NkOAMN853vbcO_fh3bsD4,20016
equitrain/backends/scheduler_common.py,sha256=Z0fh4-NM5T8Eu6SHhhhS6TyTQJ6XjOCt1uUWXEcOmkI,928
equitrain/backends/torch_backend.py,sha256=Vs7d-dVEm_2xjmiuOvBtnAAkR2vHkRctBLiKyed_6IA,15388
equitrain/backends/torch_checkpoint.py,sha256=QVhMgauaqZ0pb3g7kFozEAYo6tqH6CR-_WGkjgB3UB8,11279
equitrain/backends/torch_evaluate.py,sha256=XQcAGODCjsVlxEUmD-TOYy_RBCkebIWSXzxDBWFdEF4,5824
equitrain/backends/torch_freeze.py,sha256=9FQVzK3ROq22Ry6vNuFIl0UqspEwQngJ8PIe-0ZgXBk,1254
equitrain/backends/torch_loss.py,sha256=xciezTI_gX2WoGnQy_fuqdFSj6UdrgNerx5dso5Ll8k,3226
equitrain/backends/torch_loss_fn.py,sha256=I536fgdsR2BilN4mh1WrWiqhkR2TWgd8baDTmrsdTjY,12519
equitrain/backends/torch_loss_metrics.py,sha256=6xiSvktKYD4c1QsX-_V_xt9wGeCl4WbyJNLNiCNfoLk,7792
equitrain/backends/torch_model.py,sha256=7m-cj1NR6xXs5SM-loFakIiQlTzCngS43yGLRDA4GUc,2509
equitrain/backends/torch_optimizer.py,sha256=v5Vv_3l3jB4UThiMJ7wLsaSzCas80C5eCNC2ef4dj88,3396
equitrain/backends/torch_predict.py,sha256=7TkwQW6DUvwJILVACvCMs_QXcgVgGe6JhitY-nfvy4o,5299
equitrain/backends/torch_scheduler.py,sha256=VYjmVqs-_pce50OZ4eZYCrBE1Ln-ll5oK8qVR_a5-wk,2353
equitrain/backends/torch_utils.py,sha256=zSffaNdm-cjHYAi5ohXcePP8PWMzJwe3Z0dHbKg_sI0,640
equitrain/backends/jax_wrappers/__init__.py,sha256=BxFVvy_QS8eOQ4BJm6tCwfJ9yRemRjAbeh4Blm-pSCM,3368
equitrain/backends/jax_wrappers/ani.py,sha256=3I2OSBqVQDOqrIospQ4X8FeD1CnjqQ-hJ2xPFIWvkS0,14976
equitrain/backends/jax_wrappers/m3gnet.py,sha256=jIJUad6B491yaUH0pnnbFQRoArf-Klf0QA0LIY2FM6s,20378
equitrain/backends/jax_wrappers/mace.py,sha256=FQHSCSC5xFEqtZU2S7A-ZMwl5V-hyDGEdx-HwLu7Ovo,4375
equitrain/backends/torch_derivatives/force.py,sha256=G_wLh-MB_BhCKVEr9gakuJuSRQr_z0sUBfM7UcrGtME,518
equitrain/backends/torch_derivatives/stress.py,sha256=tR5r-IAla2x_R7lvv1_ZT5g8AnnQ_nsaLn887lt_PCI,1239
equitrain/backends/torch_wrappers/__init__.py,sha256=QRaArP174P13Fp7zhKbGvTVq62WKkuIw-lYQi5SHqgk,1890
equitrain/backends/torch_wrappers/ani.py,sha256=KyTbsoVhN49P4RSSruxL7O-SVD_ISBgmxW1icZ5TXZU,8856
equitrain/backends/torch_wrappers/base.py,sha256=4StsvatFd9F_XX24Z9dI1u__hzQZW-IpsyFCYgvnT4o,1347
equitrain/backends/torch_wrappers/m3gnet.py,sha256=kamD-sQeQLyJA2typ_4as_ltQAMkt6in9ttvxHna6CI,7694
equitrain/backends/torch_wrappers/mace.py,sha256=3zizMdnsV_BBtwEoWVw0qLxy6_PJdZ_JxALhhB46LDw,8084
equitrain/backends/torch_wrappers/orb.py,sha256=Wnfh3-CMUWRh9LaGdDztb5_78C1cC1QkKnj-2IM0yls,11047
equitrain/backends/torch_wrappers/sevennet.py,sha256=GxzuUevUETfSihV0OvOndx19v7Ujb790CmSRnLtQWPE,2895
equitrain/calculators/__init__.py,sha256=33Y5B_QN9NFvSQGnWSFNk2InbWQsQw1BtTorHyZRgjw,439
equitrain/calculators/jax_wrapper.py,sha256=AVpUQ0-nn3JYyL-rDB9JxifisVu8gg6eVugckGRzQyA,12066
equitrain/calculators/torch_wrapper.py,sha256=Rhnv6sggaYD5wU15aGsbzpTLuhGN9fVLyEtm8oGcKa8,11837
equitrain/data/__init__.py,sha256=ARnhgeYwo9TZvAknE860M8g9q5usV1AGIBKO8IBKlHI,298
equitrain/data/atomic.py,sha256=GQc57_Cv2L5OzgO0U3_zNCRlqWctvj3eRdwcHWtZcgs,777
equitrain/data/configuration.py,sha256=7nsNt5DYzMGuJT_5wpiOz82R5ZTIQMG-KateqEs1Wvs,8438
equitrain/data/neighborhood.py,sha256=qnpTo9E60pEbC6NbjIipXL-o54XEkHEynIiCeEZr9-4,2303
equitrain/data/statistics_data.py,sha256=vIUCa7RKnWbOEAXkiyThrMLW-Ydmzijv-7yZos1eihs,3879
equitrain/data/backend_jax/__init__.py,sha256=KlLoRiBw139AjAxaK6HVPk5eM8Ppb95ILMyxL0PV61M,595
equitrain/data/backend_jax/atoms_to_graphs.py,sha256=AKzQzZOiabSApkorn_RPHwsh5FDYYx_XDz6CdavbBRY,2434
equitrain/data/backend_jax/atoms_to_graphs_impl.py,sha256=UVguF3k48bz4K3vNJGJ9311Rj5A5st1Psdfl4lWZves,9932
equitrain/data/backend_jax/loaders.py,sha256=suDoVNl5XpfCo1VpOZpPHvTq5gxMZIHzP5QhJCMSM84,3561
equitrain/data/backend_jax/loaders_impl.py,sha256=5SfkytAbjdvfV8oEQQb36ClK96m_VLo5lp_YrFKJKMk,38294
equitrain/data/backend_jax/loaders_impl_cache.py,sha256=5-oeGmHUrEvseAvNbJGfsj9G4Fg4O2O1h7EDGXkWKKY,4850
equitrain/data/backend_jax/statistics.py,sha256=pw_wXkdGB5XMTlquux8NVuK15jgC9wRX6zuj09MmHyQ,4756
equitrain/data/backend_torch/__init__.py,sha256=4ysc-rxV5G9zPATAIQpKDWBgeNPXlxtlxMN8IPWLbws,903
equitrain/data/backend_torch/atoms_to_graphs.py,sha256=IHrr5asebwWBbWo74TpJnIbEmm9Vz8mccP_qeYn_BCc,5563
equitrain/data/backend_torch/loaders.py,sha256=smMrTmHnCyOnmy4G5cCy2usMqUjYAv0wRAJr6GJE34A,5467
equitrain/data/backend_torch/loaders_impl.py,sha256=41z673x5URhOpHy8eQp02dqywbA9Xe98xwxsMkHBNuk,2700
equitrain/data/backend_torch/loaders_reaction.py,sha256=8-9cSL6Z6VeAWCGFpr26iji_5hE5yNa-x96qR__ssYA,14740
equitrain/data/backend_torch/scatter.py,sha256=a1r8nk-WsBEraH78QqEWVSQIrviyDF_T9jT7yRchFGQ,2917
equitrain/data/backend_torch/statistics.py,sha256=rYn15EeiF8wsGc5Ig9I94DPQjV7oEmBnh33LL_wDU9E,3591
equitrain/data/backend_torch/utility.py,sha256=U8iHlxFzeiFi2DswZaGFlLm26n6TCvqpkghzUXD53As,1354
equitrain/data/format_hdf5/__init__.py,sha256=I3VqS93Iy3XmQZhWzorFjcPI6ZjOn_EHTMWQ_0yfXyI,154
equitrain/data/format_hdf5/dataset.py,sha256=X_DSehHJKZI4NpA1ZLU3lmkZmiD99F5iqs1w_lO-CMI,15326
equitrain/data/format_lmdb/__init__.py,sha256=qdYGeVif5d2ng6u6wDk4HSqcViXeyjWytOJ4lEbGhnQ,172
equitrain/data/format_lmdb/lmdb.py,sha256=KMCNl0mDa-0Vikuk6CZdg43yr0khIrm_zHfyut7IR8k,7145
equitrain/data/format_xyz/__init__.py,sha256=UtNmHfQxZsYX4y491ejt6a5crmN2Jr9HEgNRqdYDTzs,109
equitrain/data/format_xyz/reader.py,sha256=KkwjjwvcDX60ujKLNS0-YUQ64PruKDR0tfbeO6wnpM4,3393
equitrain/finetune/__init__.py,sha256=Bi98UUmuqqq2pMDhN71pCxyN2To0eTLxk19OfuxJypc,1302
equitrain/finetune/_layer_selection.py,sha256=GTK2tcQAgXy4Vmy0LB0qj-HXou_uns-adES2hWwXeHI,8909
equitrain/finetune/_lora_common.py,sha256=faCHh-JKbZfTeldYHmymMg6J4yVR8FxLYR5TMpYZ3Dc,2210
equitrain/finetune/_torch_common.py,sha256=eZalWW1EVuzF20JBh191K-zOudGImntr2jrKql5eEDg,1700
equitrain/finetune/delta_jax.py,sha256=iGPpXpYBQ5m2_Ci-WED08v1laLV9wdPGTkkKCaYrpcU,3314
equitrain/finetune/delta_torch.py,sha256=eDzNkhaZz2MCa6N0Eg33-sQ_VpmguZ7Ey_wIMIV2_Qc,6860
equitrain/finetune/freeze_torch.py,sha256=bBXNxxyf-7z0ZM7Kl9HkdhGKITbq3ie3Lv9db7n5MSs,4359
equitrain/finetune/lora_jax.py,sha256=kNnRKbbdGsBwS_Cs4llZfJpZPyyZO0j0vA52oYXywQ0,7184
equitrain/finetune/lora_torch.py,sha256=CgebG9GrCaEIMGbSI6ymXfl0LyE9HFZ-5NEISAi2XFw,7830
equitrain/scripts/__init__.py,sha256=r-dfrB6hEUuU3Ah8mlENIMh3WE-SiPo5wh4fB_0xZHw,37
equitrain/scripts/equitrain.py,sha256=Hwzc4jVF-7lxCvwHirhs2pc2f2S43gr95yZWCqlNkiQ,302
equitrain/scripts/equitrain_evaluate.py,sha256=mPqjqOJdn4pSl8H5ouksBpAIoAL_YytX0vsYulPL41c,297
equitrain/scripts/equitrain_export.py,sha256=EUhiU-N4pIZe3rKVD8brAy_eFxl-ezGZcNu4EXTuN4w,8039
equitrain/scripts/equitrain_hdf5_benchmark.py,sha256=OucZHzE40GgbH_vJSy0qAVDE4BEMzgArFvtuGHKk-FY,3943
equitrain/scripts/equitrain_hdf5_info.py,sha256=poly9EoYabmZva4xJb-Funu5jKddeY-SOBmYVeE3Fy0,4346
equitrain/scripts/equitrain_inspect.py,sha256=My8EtD1SPIfCW2zRP-Bj6yU37eL_Kg3NFIJPg-tzcsk,1432
equitrain/scripts/equitrain_predict.py,sha256=379SZqaWDRA6U-u1MHCcp_M9CmUoNpqh_DEad8nTCt4,482
equitrain/scripts/equitrain_preprocess.py,sha256=yBJqHAZ0LvJT2-HQgYYPLszD6nR8lAdIMdlCSxTcDRQ,304
equitrain/utility_test/__init__.py,sha256=rWXpi04wv_IME3c5wKl7L_oZq3ZjJpH8doAnfw4WTl0,1313
equitrain/utility_test/mace_support.py,sha256=4KuHo0k5fpAwuLQ3LILnyQ4trUUckaJG90xcn1cIlvI,7520
equitrain/utility_test/wrapper_ani.py,sha256=IYuTeYd5ETrlUB40T0wlvEZVTrmVd7spvJKwdSz8aFo,1605
equitrain/utility_test/wrapper_m3gnet.py,sha256=V3kZ3Y7cLUdy8bb0ye93fFRSqp3hFNbJ-PVxyEkA9NM,3951
equitrain/utility_test/wrapper_mace.py,sha256=7-ZL_hghPI2GNVplfxue2qCTo7fceg2IuJPmUuBOEso,1874
equitrain/utility_test/wrapper_orb.py,sha256=rB5PqV-P-y30Dj-3sB-spzOCjr4GnQUVy0jJVuA1E54,6458
equitrain/utility_test/wrapper_sevennet.py,sha256=6cGZ5TTnR7V8xFwYpkZGvG1ssEZyCFCzQFNjdT7W5Qg,2166
equitrain-1.0.0.dist-info/licenses/LICENSE,sha256=HfTR8kPuxvjfJ7tJ4HegYcJ17gqKO-qG6NLWvrRTq-A,1068
equitrain-1.0.0.dist-info/METADATA,sha256=VEQjW7S9hIGP06OXRLxyAYWUqoeJS1dDeEpB0BMG_UU,9734
equitrain-1.0.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
equitrain-1.0.0.dist-info/entry_points.txt,sha256=qW7uH2YX6-ucegdkINHZqx5iA_bryoH867YBsUQzf4E,514
equitrain-1.0.0.dist-info/top_level.txt,sha256=FVOtSk2Uwe1o0KjOXBanD5g4gAzlOJU3NjvL0aiFbQQ,10
equitrain-1.0.0.dist-info/RECORD,,
