atom_jepa/__init__.py,sha256=WnXgjKLz9kxaoCN6EiaLWCi7r1HFoTbbm-pfsTYccck,502
atom_jepa/api.py,sha256=DYTfLrOx9QEcbAmNeS8TjJ36oTpsxnyXjS8HcCTxAng,14137
atom_jepa/checkpoint.py,sha256=2cHF5NuPltCqL8qQbswyJ7qOFxYvw515QOjvqOBj398,4125
atom_jepa/conformers.py,sha256=1hsy34hxPs9lqhrLoi44D8b18iOWvAfLE-sJrebOBJM,1885
atom_jepa/cue.py,sha256=W15pd0QC2TwhaGvFf-oGoJfPCszq-0wqrIxRk9UHi4k,3412
atom_jepa/execution.py,sha256=2VjLX0mK6Fc8HvYKzbpsTVUu1qKPWs1rqtSE8CPlTZI,4041
atom_jepa/info.py,sha256=qBOM_FOcCFYAVa_2e46633-fSStcOfmwkoVS09zZM-w,1020
atom_jepa/data/__init__.py,sha256=EJi36vBEGJd_3MPfIOBfXbaswj9ELN-e1NCUEkh22XE,172
atom_jepa/data/collate.py,sha256=-VCxm-F5xGURXYJ3VkqAvmTuZEd31YCbs-jOBva-REY,8244
atom_jepa/data/elements.py,sha256=wHkj0CJyoirTAXS35vcIYM3NcX9XKCAEhaZDojyuSns,1138
atom_jepa/data/graphs.py,sha256=9Mru1Pe1joblyQ-_oycn-Np1wdBhtTgzBgq7pAoU1po,4820
atom_jepa/data/masking.py,sha256=t-0WTXl63t1b40aF6nvgDBWCK83_A2nxV-wd8-93tYQ,4542
atom_jepa/data/splits.py,sha256=RPOI5k26KUpWF-GXWha33woAAlzVUs2vIC_jCH3XTT4,649
atom_jepa/data/structures.py,sha256=iaruDtb-BmUTzinkjc3vt3G_XWQ4Vuy5LYYb0TRy-AI,2368
atom_jepa/models/__init__.py,sha256=3E1ALcid-WvD5TgiNPAZQjwqWRmoKm7eDA622Seu9lo,79
atom_jepa/models/eqv3_backbone.py,sha256=U_HyM-5sM9AGwE7HZoo6NZnq4ZuLLcedIc2XvMUzfRQ,17191
atom_jepa/models/jepa_equiformer.py,sha256=2SBMNYTmxVNFrytyTTSuQuauxuuti65mJWDMIVMvNWw,20073
atom_jepa/models/equiformer_v3/Jd.pt,sha256=tAWcRb4kbctsScVFZwtlxWVQ6wwuepyStLUKktNw2-I,21697
atom_jepa/models/equiformer_v3/LICENSE,sha256=8tPIkmUdj6DcaVenTN0P8Ba9c5aKXhJ8QhkZY2tKNb4,1078
atom_jepa/models/equiformer_v3/LICENSE-fairchem,sha256=PhCCeTiq6eWRZHciyDAGWpWzrych8NfwLzc4abQP6B0,1093
atom_jepa/models/equiformer_v3/__init__.py,sha256=prD_nkI7GDcQt4ACgN9qPrOhIo-sHkQBxDd6-ySNmgE,538
atom_jepa/models/equiformer_v3/activation.py,sha256=Z7bFENgOyZkRReFB_Acw89C0OLDaZTMVoFD4nKh9trI,20538
atom_jepa/models/equiformer_v3/admet_cue_ops.py,sha256=djfFMjAlT6j4_EE6YhKaqh1-2NypbaBfJkZpvj6r8LI,6404
atom_jepa/models/equiformer_v3/drop.py,sha256=nNauN3Yn3X55kWI-iWgEE5lwYDpZ9RFNI2bVBL4Kj38,3436
atom_jepa/models/equiformer_v3/edge_rot_mat.py,sha256=i8Rro0ZXX_dr1GbNQeSuMrNSMsht0ykSsbskZKfJ9Gc,4028
atom_jepa/models/equiformer_v3/envelope.py,sha256=rkhyafh43ZEl0Wo_a3fEYTuvfjLNZv-RD1qFNTTOXXY,1264
atom_jepa/models/equiformer_v3/input_block.py,sha256=Kz68WeUxccQ4tspkCZwOeGtHpHIn30ysfk6KsM7qArM,4745
atom_jepa/models/equiformer_v3/layer_norm.py,sha256=5JBb-K8s3zxtJIOVtDEckSTSASjmckzJrRXjYdQvu0s,14266
atom_jepa/models/equiformer_v3/radial_function.py,sha256=qiGA9doQ_RU96QXiQVk6uaA8zDwcjkVQpzXQ5DM71OY,4353
atom_jepa/models/equiformer_v3/so2_ops.py,sha256=7Z46G_OAZCWOzfHsYqXu8Ame_XVdmg_GusL7d3GC80g,6266
atom_jepa/models/equiformer_v3/so3.py,sha256=dJBxx8GO6f6C4PNgpoq_3dAeyPtgumRyE_umWSiKA9I,23924
atom_jepa/models/equiformer_v3/softmax.py,sha256=xEBgl7jk60CAgzsooNgXO6FujAedfX4yRiFfht3iAg4,3227
atom_jepa/models/equiformer_v3/transformer_block.py,sha256=oxhf7ckK1RpsKTQ18CehEw6pnHh3uja5siAGhlG6ymA,37085
atom_jepa/models/equiformer_v3/utils.py,sha256=C1YF28-MbSYM-ttPScGBNmQs2-FvL1aR0Ek_dza4lRI,254
atom_jepa/models/equiformer_v3/wigner.py,sha256=2wH5vkGOrzk43PpLr4efDAJiOHgouKQpccg64D-sVyI,2390
atom_jepa-0.3.0.dist-info/licenses/LICENSE,sha256=ynPzXILhF1hE6HP7wAXGbKRMj5EW1YZcvMqVYN-W4g0,1069
atom_jepa-0.3.0.dist-info/licenses/atom_jepa/models/equiformer_v3/LICENSE,sha256=8tPIkmUdj6DcaVenTN0P8Ba9c5aKXhJ8QhkZY2tKNb4,1078
atom_jepa-0.3.0.dist-info/licenses/atom_jepa/models/equiformer_v3/LICENSE-fairchem,sha256=PhCCeTiq6eWRZHciyDAGWpWzrych8NfwLzc4abQP6B0,1093
atom_jepa-0.3.0.dist-info/METADATA,sha256=lOEI92-8SGXVdzv0HPud37afDPt_mE-ka6F93I3GoAE,7022
atom_jepa-0.3.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
atom_jepa-0.3.0.dist-info/top_level.txt,sha256=sm4rEXvZPM2gTxdsMsP1KmKDj80mq9-Fpi-mpzoIE-c,10
atom_jepa-0.3.0.dist-info/RECORD,,
