torchebm/__init__.py,sha256=JVE1JhbhGDnJrhRVbYijgDokVeDiWcDcTXSh7yfymwo,1077
torchebm/_deprecation.py,sha256=AdKsPu4kJHg7ILFDggzEC9VQpoQLU_460J4Ajq0ctO4,11655
torchebm/_version.py,sha256=aR8h0I1h0v-d6F6j1xkcbOoAX4TR1lvZD30yMpSxds0,528
torchebm/core/__init__.py,sha256=l6k-uZs4xejHNXqOZ-s89X9PpU-lpVyPPTZbzV7Dy-g,2034
torchebm/core/base_coupling.py,sha256=LNgrVZ-gHR0XJO-ayfGFKSNcXXpHposuWEdiIciYcdM,13018
torchebm/core/base_integrator.py,sha256=D92FVi0y1h2SPUxcGbuBct4M52kZqr0m96l8ZBmrR_4,33359
torchebm/core/base_interpolant.py,sha256=cNz49afzS_cvqi0ZQfRfohDD7L4rEuPLT3iPnXFSGjk,8172
torchebm/core/base_loss.py,sha256=jPeQa5Frc6DRJAhv_7PDOOqvDKld-04k99HsPZFOPB8,49768
torchebm/core/base_model.py,sha256=8Hcidwm5eM9_mMOdqmu_9GAiZbRNr-lTNANPU_cK-q0,11704
torchebm/core/base_module.py,sha256=sL3ZQOywnWHjliMlfLGmGs2arrM59XnE82dOikZX6yc,8946
torchebm/core/base_sampler.py,sha256=iiEnJPzQQjXx9q4cXcTONtEigSUb3w_kwuZYf6CsnNo,6572
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=C-D_aiX70YgY7JBJmXWD1SaAohFV_xykUHuXhYTDs9M,4013
torchebm/integrators/heun.py,sha256=tMebtwjFa70GnPIgdpVhYqCaegC88s8BPHy5TMR8Srw,1590
torchebm/integrators/integrator_utils.py,sha256=GrDL8dEGW1GDWLBZ8bO2pqZh02k0bU0GSrtgrz9WQdQ,4480
torchebm/integrators/leapfrog.py,sha256=F8GT3bzCUAtYgFQv6Im6BOM2gnrP1fNPZVpThit3cVk,17253
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=7s0YtuVJk5PKr1i0J_SXnsFIRA2AgVtpLSqyUcTylBM,1586
torchebm/losses/contrastive_divergence.py,sha256=SJZ10CLQdFeXpuCtYPXjDlVCTX6b1YEDsDYS3I73Q4w,10805
torchebm/losses/energy_matching.py,sha256=qoNEJBnr4JZZ_duWNauprfX05vz0IjvEehn6lp4-mS8,21253
torchebm/losses/equilibrium_matching.py,sha256=czaE8hVnplt5jHahlWd4r6GSup9lW8KEJxzCTLGgvII,18478
torchebm/losses/flow_matching.py,sha256=3GmkZXzDwyeQ2Ncirw-ArzDuxhsmNigYcFDQ2AD7xcw,7563
torchebm/losses/loss_utils.py,sha256=knJf7vOxnD81eQBBOYlLT5ivOHbm7R9vnLaut8X5PtY,4103
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=_LGMykssgZ5LeQz2CRG-W5En1aw6H7n-hqcVGBwCWw0,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=BKHwrVw2PQ30fkK08alln4XBiC6ZNrAxRp0UDyF0w8I,3957
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=U-aFqrlGluqjZmEBtQhi3ERgbhtmaY6syIkcc06W_wI,26122
torchebm/samplers/gradient_descent.py,sha256=8XSHVFdJcJXNogn4xCYCsVJdKVFivQ3EaPuZz4RlDUU,10461
torchebm/samplers/hmc.py,sha256=7TmxLnf893QHXCHffxg8SLRgCkKRj52FBFeiBtsg1nQ,31289
torchebm/samplers/langevin_dynamics.py,sha256=8xuYaxdYs6M03kiemGHB8I57IiWBtpTvASVZ-LRaRA8,7627
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.5.dist-info/licenses/LICENSE,sha256=HeINI_-IOwxcykODWlBkhPmNSZLYWNlKJ0JV3b9K3WQ,1078
torchebm-0.8.5.dist-info/METADATA,sha256=ePEyTDazZIlcqE5i3xgp9tUWvuCMHOpICDE2shFm6c8,35729
torchebm-0.8.5.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
torchebm-0.8.5.dist-info/top_level.txt,sha256=FTDHo7yycQ1rtcscAqb-fXXaoOVdtugPVNEUnm_FYHs,9
torchebm-0.8.5.dist-info/RECORD,,
