torchebm/__init__.py,sha256=JVE1JhbhGDnJrhRVbYijgDokVeDiWcDcTXSh7yfymwo,1077
torchebm/_deprecation.py,sha256=AdKsPu4kJHg7ILFDggzEC9VQpoQLU_460J4Ajq0ctO4,11655
torchebm/_version.py,sha256=66eOA1xsjDh661ArDMFAC1B_h5SYJRTgKX2S_WjKYlg,528
torchebm/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
torchebm/core/__init__.py,sha256=8xotZUbNiXU8yml9sS2s4QDhdFJ4je-1iMflC61ftfE,2164
torchebm/core/base_coupling.py,sha256=LNgrVZ-gHR0XJO-ayfGFKSNcXXpHposuWEdiIciYcdM,13018
torchebm/core/base_integrator.py,sha256=oAWNGu4zTXRkb5jS-hSb8B-tlOcwtoLh7iL5mv1CtYM,33742
torchebm/core/base_interpolant.py,sha256=cNz49afzS_cvqi0ZQfRfohDD7L4rEuPLT3iPnXFSGjk,8172
torchebm/core/base_loss.py,sha256=0Lfze067NsDjVn_H5Kc2QBOsuXDEeKO3IJWzz-DYOI0,49873
torchebm/core/base_model.py,sha256=IVGm7IFedobHMT7QKjSEMgkbV09KJiQipbtcBWJ_j7Q,13163
torchebm/core/base_module.py,sha256=sL3ZQOywnWHjliMlfLGmGs2arrM59XnE82dOikZX6yc,8946
torchebm/core/base_sampler.py,sha256=XQCOLlaIG9unqfsmd0sOwd7aXGYMBAVVSrixn96Q200,6683
torchebm/core/base_scheduler.py,sha256=9s62xyY8MAiHPWBPzNY_BUpeg_ipQWgyV-Cj-yMPHMg,34438
torchebm/core/base_trainer.py,sha256=mMuESgE1vaFPP7nnqd_SFCMdaCHMHjaUXxR0UZHkko4,17359
torchebm/core/schedulable.py,sha256=0ITE7CUqCRkfwNbJYmGUVYeZScydT_79Mij8stoMk_4,2893
torchebm/couplings/__init__.py,sha256=FP7mtCAY2QfcKbCbg88qj06enPt7N-cIOZHwvSYsj0M,1328
torchebm/couplings/coupling_utils.py,sha256=8OQYKSBeuzNX8LYeMElLww6O240rwp6nmDaRfP-lDtU,2463
torchebm/couplings/independent.py,sha256=6LWmNqWLpwn8qjXNEcpQSuqSvd83M4Ap1a-beoD3jWc,1091
torchebm/couplings/model_induced.py,sha256=ThkS5AsFwZUiJsJk83rYrHabkK7eHdDIL5BWdY3-Qbs,2916
torchebm/couplings/ot.py,sha256=-ZRT5PeAD2C1GOauzxfF3M8DvA9mrxwFpOirqsPyxVw,16047
torchebm/cuda/__init__.py,sha256=6ZmN8eJbsWAyfI8aHjlIq-nbG_ImqUBYiXzlHrAyw2k,216
torchebm/cuda/fused_langevin.py,sha256=AAW_Xwyo8c226xAdux4jR5qxZtBUQdwVY7zD92JauYU,8819
torchebm/datasets/__init__.py,sha256=bVSJwKaARlDjRmz4iBmakIV7eBmw4w9-PtJ9gaB2KBg,542
torchebm/datasets/generators.py,sha256=iDilrE8wDMmdRdJeALUNWr8c-HGfwMMPTG9smfsW-Ys,44117
torchebm/integrators/__init__.py,sha256=7X3oFRwjnhptw3CT1s9w6RRB5ADGZA5r3O3gxDGKoV8,1579
torchebm/integrators/adaptive_heun.py,sha256=cwmnhp906C1KweGAQ_kVW4ye9ZGNqJiwAuNkwyJPEuQ,3378
torchebm/integrators/bosh3.py,sha256=IBXhsPFXibn0tepalPEEfs6eMHkBRFUl61Gn5itfwcM,4152
torchebm/integrators/dopri.py,sha256=kDoLKSUEkKE5Z41tg09zc2IWj9iCvNgMmmTjzooqivM,9411
torchebm/integrators/euler_maruyama.py,sha256=Vp8MkqNqe6YGqDmMJa46MI3COJvw6oLjMB9x4DCDVxE,4339
torchebm/integrators/heun.py,sha256=tMebtwjFa70GnPIgdpVhYqCaegC88s8BPHy5TMR8Srw,1590
torchebm/integrators/integrator_utils.py,sha256=GrDL8dEGW1GDWLBZ8bO2pqZh02k0bU0GSrtgrz9WQdQ,4480
torchebm/integrators/leapfrog.py,sha256=kqjw_xNeGE348ow01sPPZBII5nv2LdpYl5NRdH0L7TE,17517
torchebm/integrators/midpoint.py,sha256=icFAGawxOjRPNZT7-7x3ko12NgTsPeYNLTroIr4RhHg,1383
torchebm/integrators/rk4.py,sha256=RDDk5QN5H5VxS0egFPhd65ZA2OvCSbUQUVj2Ieky2LI,4404
torchebm/interpolants/__init__.py,sha256=3KB91tlHC3BaEkzaZW1tyXPNEsRKQA94BuGaL4mScJs,882
torchebm/interpolants/cosine.py,sha256=_2AV2ivMyG7tF-r7KEmZ4tM1jkRZBAjxfahNT0JLatM,2158
torchebm/interpolants/interpolant_utils.py,sha256=qk0zrdK8uMrH3crq8C4lmmABaYTu-H5d6u8YNcvrFVE,2124
torchebm/interpolants/linear.py,sha256=QzXcJPuJOiti8PhAdAz2AzxILdKo9FYAplzGWCJn53Q,1900
torchebm/interpolants/variance_preserving.py,sha256=q-sSLW0HTdj_FO1vV9xuIJAHHMHpmonwTZsh3BZzUdU,3714
torchebm/losses/__init__.py,sha256=Pqn4wTc5k4hiXGdeyiHXNP8C7t1D8-0QPWgu5p-H9dM,1505
torchebm/losses/contrastive_divergence.py,sha256=CYkMShDP2I5l4TOxLVbyARGLgbpWrcD8qVnOH7x4p_I,9377
torchebm/losses/energy_matching.py,sha256=qoNEJBnr4JZZ_duWNauprfX05vz0IjvEehn6lp4-mS8,21253
torchebm/losses/equilibrium_matching.py,sha256=Pot9NhXo1OTIvgU8hbcPkbRBSjG5YKhdku9WWscuUDY,20419
torchebm/losses/flow_matching.py,sha256=AKte2Phxlf35ahHJTz9xO1v8GCp8Jbn39xyRgjRCGio,8250
torchebm/losses/loss_utils.py,sha256=zpNhYIfqTbbCS0N239ofiyUQuPAWk7WV-Utv6aTWHVM,4135
torchebm/losses/score_matching.py,sha256=FVjEKaCadI-P8Cw0U2fusXYiiz7Hz3XGCuMXXytJHKI,21924
torchebm/models/__init__.py,sha256=8hZW6K467mgonSf2jCyj3AcnAT0rqvfU_iR2SomKdXM,2268
torchebm/models/backbone.py,sha256=1VyjHLEzMcE_q3UrhP_oBhHH5VJyRsof5HnTR33px74,4979
torchebm/models/conditional_transformer_2d.py,sha256=H_9y_VDDT3bedH3K3QK4yJpLu_p84uEmH4o7kkNo7GM,2764
torchebm/models/dit.py,sha256=lVoLR7t35Dm9jcSbTdNyZ0pwue8J8mR2EksBjN-zW68,14807
torchebm/models/mlp.py,sha256=hN5PE35DOYCo5I50oF44lnSLA3EqKpXIrknf4E52aY8,8624
torchebm/models/wrappers.py,sha256=sdVXkv0i4o3MAYGgSfaGWyLpW4ZWdK5-Tj1Nnvm7-xc,13290
torchebm/models/components/__init__.py,sha256=TXLGQnQlDOR6XvIuzz57z2ScA2u625Y5_ci8MSyLWRw,964
torchebm/models/components/embeddings.py,sha256=usNeI-w35phc_uc8qjZNVn2lqMGthCjhS12cf8y72wc,4876
torchebm/models/components/heads.py,sha256=74oiHDnRa7i88HWj2cq6y7BoCsAKLJV-KZcPrev_WXk,3258
torchebm/models/components/patch.py,sha256=MCFlyFlTB4MYMtXvkn_PKY0U5uHYQ9tAkHO4b4HMz70,2302
torchebm/models/components/positional.py,sha256=cK7mEtB7RcK_ZdTdiPMhun24oiOXAL53a-OkEeLTuUg,1824
torchebm/models/components/transformer.py,sha256=vOzIBqOcFAgYDIBfVkvPMUZ-K_MUXteNoNppcuuB9nI,5689
torchebm/samplers/__init__.py,sha256=qqvs2KTHw13Zt_Gt9UDN6E4IGPBzierUYsh2zmaoGQk,1114
torchebm/samplers/flow.py,sha256=03I6K8vJaCrGltwJLAH36_5ksS8a-AsUBLujFG0YFCU,26499
torchebm/samplers/gradient_descent.py,sha256=4sqBU5TmGq9PP_Y6xBihDWFoIKw-BVJGtoH1nvo2mrQ,10938
torchebm/samplers/hmc.py,sha256=PCHUZ4q_m1ijnf3oi0_eCFJyto00FLHopEEDcy7fznE,32325
torchebm/samplers/langevin_dynamics.py,sha256=gAtdT0vZRzIaSKV4zSq2CIvr5Sx9YpdDexUQSxxq3Ko,7977
torchebm/utils/__init__.py,sha256=1OLIbkvYyuVxBHDfwmCS_biETxfOULarrVSKZEC28N8,646
torchebm/utils/distributed.py,sha256=7YV9DbMJMvepAvRLLacgurM1YZmFA8SG0-cH4SP8SNI,5471
torchebm/utils/ema.py,sha256=3o5K56eAwUgg4d-_j7z-V2xQZqC_DpTelTwQr-sUh2Y,7879
torchebm/utils/profiling.py,sha256=fPlpTL6Vm0IsnK7PuxYaHkj5X0CDkEWaxdcjaldGq88,3086
torchebm/utils/training.py,sha256=scY160KsUif5m0_3qST3uq2c028zNrB30YmpQB_9NoQ,3800
torchebm-0.8.8.dist-info/licenses/LICENSE,sha256=HeINI_-IOwxcykODWlBkhPmNSZLYWNlKJ0JV3b9K3WQ,1078
torchebm-0.8.8.dist-info/METADATA,sha256=Nu6N0FW6_mlnnAmdPLnvPzMTXCmrsVMpIOupUd4eZJM,35462
torchebm-0.8.8.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
torchebm-0.8.8.dist-info/top_level.txt,sha256=FTDHo7yycQ1rtcscAqb-fXXaoOVdtugPVNEUnm_FYHs,9
torchebm-0.8.8.dist-info/RECORD,,
