FlaxDiff

View on PyPIReverse Dependencies (0)

0.1.35.6 flaxdiff-0.1.35.6-py3-none-any.whl

Wheel Details

Project: FlaxDiff
Version: 0.1.35.6
Filename: flaxdiff-0.1.35.6-py3-none-any.whl
Download: [link]
Size: 54510
MD5: d7dc3bdec56d4281bd61e328782017a6
SHA256: ed360d12b704621369c4be1d5a265dfec20de446f6e14117924f6dcb281b443d
Uploaded: 2024-10-02 21:13:18 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: flaxdiff
Version: 0.1.35.6
Summary: A versatile and easy to understand Diffusion library
Author: Ashish Kumar Singh
Author-Email: ashishkmr472[at]gmail.com
Requires-Dist: flax (>=0.8.4)
Requires-Dist: optax (>=0.2.2)
Requires-Dist: jax (>=0.4.28)
Requires-Dist: orbax
Requires-Dist: clu
Description-Content-Type: text/markdown
[Description omitted; length: 21736 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.1.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
flaxdiff/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
flaxdiff/utils.py sha256=B0GcHlzlVYDNEIdh2v5qmP4u0neIT-FqexNohuyuCvg 2452
flaxdiff/data/__init__.py sha256=PM3PkHihyohT5SHVYKc8vQ4IeVfGPpCktkSVwvqMjQ4 52
flaxdiff/data/online_loader.py sha256=DoHrMZCi5gMd9tmkCpZIUU9lGxvfYtuaz58943_lCRc 11315
flaxdiff/models/__init__.py sha256=FAivVYXxM2JrCFIXf-C3374RB2Hth25dBrzOeNFhH1U 26
flaxdiff/models/attention.py sha256=JvrP7-09MV6IfRLRBhqjPmNUU-lkEMk9TOnJSBKcar8 13289
flaxdiff/models/common.py sha256=hWsSs2BP2J-JN1s4qLRr-h-KYkcVyl2hOp1Wsm_L-h8 10994
flaxdiff/models/favor_fastattn.py sha256=79Ew1nqarsNLPzZaBSd1ILORzJr74CupYeqGiCQK5E4 27689
flaxdiff/models/simple_unet.py sha256=L5m2j5580QP7pJ5VIme7U5xYA22PZiGP7qdvcKUnB38 11463
flaxdiff/models/simple_vit.py sha256=UCDDr0XVnpf6tbJWKFtEt3_nAqMqOoakXf5amyVWZNo 7929
flaxdiff/models/autoencoder/__init__.py sha256=qY-7MldZpsfkF-_T2LqlRK7VHbqfmosz0NmvzDlBkOk 78
flaxdiff/models/autoencoder/autoencoder.py sha256=27_hYl0yXAdH9Mx4Xu9J79mSNo-FEKr9SxhVaS3ffn4 591
flaxdiff/models/autoencoder/diffusers.py sha256=JHeFLCxiHhu-QHwhKiCuKsQJn4AZumquiuxgZkiYGQ0 3643
flaxdiff/models/autoencoder/simple_autoenc.py sha256=UXHPgDmwGTnv3Uts6Zj3p9R9nJXnEiEXbllgarwDfXM 805
flaxdiff/predictors/__init__.py sha256=SKkYYRF9Wfgk2zhtZw4vCXOdOeRlrm2Mk6cvuaEvAzc 4403
flaxdiff/samplers/__init__.py sha256=_S-9TwDeshrI0VmapV-J2hqjTByOa0-oOeUs_IdovjU 285
flaxdiff/samplers/common.py sha256=_an5h5Niz9Joz_-ppridLrGHpu8X0VVvhNGknPu6AUY 5272
flaxdiff/samplers/ddim.py sha256=XHMBX06S5hMTnKMaGh6fmq189pQcaGkA6fnX6YPbHP0 511
flaxdiff/samplers/ddpm.py sha256=d_58hfVShJHsRPQf5h1-4YKrD43-HGjWz9Vd8hltZBg 2627
flaxdiff/samplers/euler.py sha256=Epf7LBKUky7B8b-1ZyIlLWdRMgjmP08BQraGSKmr_3I 2726
flaxdiff/samplers/heun_sampler.py sha256=hhWnSM26OfOIFAcsuWYa1z-2QPjASuoYTop2byLWqzE 1388
flaxdiff/samplers/multistep_dpm.py sha256=ocmEq2sCvsULy6oTFaD5BhTU4c8VHsge4bdg6tfxW80 2724
flaxdiff/samplers/rk4_sampler.py sha256=BF-dMV1KauO-SYShqrCfm3U3V-1n4clqQXBeoG8RWQo 1728
flaxdiff/schedulers/__init__.py sha256=3id390WEfdf-MN-oLSPAhlRFIXrFWr6ioAHPAwURJyE 375
flaxdiff/schedulers/common.py sha256=b-W4iI-aqScpVE8VZbBpiYvAVI6rqDkUP-C_hEVBwCI 4151
flaxdiff/schedulers/continuous.py sha256=5c_niOA20fxJ5oJDi09FfayIRogBGwtfG0XThW2IUZk 334
flaxdiff/schedulers/cosine.py sha256=9ban0dFHLMm35wQvaBT4nCQwPGmzNsXwQ1xI0oppmJI 2005
flaxdiff/schedulers/discrete.py sha256=O54wH2HVu3olJA71NxgAXFW9cr6B6Gl-DR_uZeytpds 3319
flaxdiff/schedulers/exp.py sha256=cPTnUJpYdzJRRZqMLYQz0rRUCpEmaP2tXhRumLx94jA 605
flaxdiff/schedulers/karras.py sha256=4GN120kGwdxxU-h2mVdhBVy9IORkUMm_vvz3XjthBcI 3355
flaxdiff/schedulers/linear.py sha256=6003F5ISq1Wc0h6UAzY95MJgsDIKGMhBzbiVALpea0k 581
flaxdiff/schedulers/sqrt.py sha256=1F84ZgQPuoNMhe6yxGTR2G0h7dPOZtm4UDQOakbSsEU 445
flaxdiff/trainer/__init__.py sha256=T-vUVq4zHcMK6kpCsG4Gu8vn71q6lZD-lg-Ul7yKfEk 128
flaxdiff/trainer/autoencoder_trainer.py sha256=al7AsZ7yeDMEiDD-gbcXf0ADq_xfk1VMxvg24GfA-XQ 7008
flaxdiff/trainer/diffusion_trainer.py sha256=wKkg63DWZjx2MoM3VQNCDIr40rWN8fUGxH9jWWxfZao 9373
flaxdiff/trainer/simple_trainer.py sha256=cawm6fZNQoLLATMneAU2gQ9j7kefqHnBPHuaIj3i_a4 18237
flaxdiff-0.1.35.6.dist-info/METADATA sha256=NVCk5V7Zc3iq-nrWTivzO17dQa1fIjYgjJb800ZrZhQ 22085
flaxdiff-0.1.35.6.dist-info/WHEEL sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ 91
flaxdiff-0.1.35.6.dist-info/top_level.txt sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ 9
flaxdiff-0.1.35.6.dist-info/RECORD

top_level.txt

flaxdiff