Wheelodex » Projects » generax

generax

View on PyPI — Reverse Dependencies (0)

0.1.2 generax-0.1.2-py3-none-any.whl

Wheel Details

Project: generax
Version: 0.1.2
Filename: generax-0.1.2-py3-none-any.whl
Download: [link]
Size: 123787
MD5: c07bf2558ed3900dbf881468de8b8f8c
SHA256: c5b123455276d412909bd65f2c372f4d4f98aa94ac294b9ae09017c7a27ae936
Uploaded: 2024-01-17 18:41:24 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.1
Name: generax
Version: 0.1.2
Summary: Generative Models using Jax
Author: Edmond Cunningham
Author-Email: edmondcunnin[at]cs.umass.edu
Home-Page: https://github.com/EddieCunningham/generax
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.7
Requires-Dist: tqdm
Requires-Dist: optax
Requires-Dist: matplotlib
Requires-Dist: jax
Requires-Dist: einops
Requires-Dist: scikit-learn
Requires-Dist: equinox
Requires-Dist: diffrax
Requires-Dist: ott-jax
Requires-Dist: optimistix
Description-Content-Type: text/markdown
[Description omitted; length: 2119 characters]

WHEEL

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

RECORD

Path Digest Size
examples/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
examples/celeba.py sha256=O0N6YbqhxrX0H2Imt4RGhQ4CIQOmx0WzVP5is0phJ7g 6145
examples/cifar10.py sha256=WavWBb1QH8aYNHQNVmVchl4pU-pL_Gh50bW1U0AyZwo 5582
examples/cifar10_rect_flow.py sha256=gQksM-XaVbUSwX6M34nxA2DCYEFJQ_owS1bzHBnnv0w 5909
examples/cnf_max_likelihood.py sha256=jwKermMJd1ALCeXrrChl-I7tON_zK0u1CNPSUGM1dUg 4453
examples/debug.py sha256=uEE1LEWuD9rb5eWrlvek95ZQtiehGD72r6gHD_Rvbis 832
examples/flow_matching.py sha256=4Wmd9xrYV7ffGuh4o3Lh69fCJZOmxfE-8YhK-No3ovw 4700
examples/mnist.py sha256=vLutyb00uKoHimlxOTyUf_avKVBCD-7LVIRPyIQt93k 5572
examples/mnist_rect_flow.py sha256=wYkWLdw13OpyK3aa23U004eK3kIgjfA8z8wiFhatZGU 6975
examples/multisample_flow_matching.py sha256=42JGY7SgiCOGAE4ULa9wukamXLJ_pru4e1SbBcJ_8dA 4627
examples/score_matching.py sha256=jq2qX4w8MLhooxlt_KIVDyZryu3odoJOqu6EZb0rnSo 4713
examples/spline_flow_matching.py sha256=CF0AltICu45y2RXPhZcmWxLFz1C8LnjNmu-_VLeUAqY 6533
examples/spline_max_likelihood.py sha256=NgTo6eukaP-WfMjOu_3Uy5Mdq7mwrRIyP1cz2zVhc7U 3167
generax/__init__.py sha256=-6cMDc_oUcVMsnU5Js7DTaSr8lZoqHFZzOUxAwsmg3A 119
generax/trainer.py sha256=KVCaXyjtkXtRQ8EU4QO-8ZAbbGa66So5_5AMW0_w8Wg 11351
generax/trainer2.py sha256=a1wYxW-plDQThE70bKssLG9AGGAKj116O0ALoP4DT5k 8591
generax/distributions/__init__.py sha256=QeVFy-NsCDVANYhdyIg4o7ASpPawvd0Ui-KyzyM4qMk 133
generax/distributions/base.py sha256=pC55NJrvbtNkxhI57I979YDZmvOSV3sXRIpAXpr2yts 18751
generax/distributions/coupling.py sha256=5Mh7CTxGIo4JMQ645RCAn9pgaaNEaurdIXvaOW0ACeo 2859
generax/distributions/flow_models.py sha256=XkPrWE4biFRLEOi-kWjbaUCnTdwSRaLCOb9c2Rc92TA 18498
generax/flows/__init__.py sha256=D1eHobs2p_npXURU3psOS-81pUKH7pMQ-YkvpoQWTPw 414
generax/flows/affine.py sha256=-mR8OnbZt3C0Yr5N9JkCiLLAFvJKLAXIaxweqeNyZFw 20733
generax/flows/base.py sha256=QKXBdRoiHaISeuL3n9BvDL3PYKcrURmJRNgH72PKNG0 28682
generax/flows/conv.py sha256=wHhZIstMWrmkrsFjHsJDojnpcrmKDfyRIYUpkFt4Kig 12370
generax/flows/coupling.py sha256=Bi-rVOvlRtrVcKQJWxL9vgRn6MDSK9IWb33R2pO4o-U 15532
generax/flows/ffjord.py sha256=ADlCsQVPVhLAgjxZaA_yAFYBtZqNH2dm5on4R0_rt_M 6197
generax/flows/logistic_cdf_mixture_logit.py sha256=9vvJrCHOl-s4hWR3dmrIwzz8HtsJ81485eety-3_jvM 5423
generax/flows/models.py sha256=Xfmw1Z_UeBuSO7sInpTsbfTqQxRpdW5M5WhPcSRZbRc 9197
generax/flows/nonlinearities.py sha256=nSs2oPFwiycCLtp_H3c9cozhz8640HRLONHuL2jaXYU 16542
generax/flows/pac_flow.py sha256=DqcjTPuC6Nu96kZYTOHELHfnOizOAVlZ_aZAd2ABDgk 9980
generax/flows/reshape.py sha256=EAtVvRzi6CixaelMW7uxi4wT5QWcTjypAur2Mqrl0_w 7605
generax/flows/spline.py sha256=gUCUXfAXYF3nWNACTMATf-iJLy52KLXmzLL-q7-hC6s 8798
generax/flows/vector_field.py sha256=KUCrGuiwm_VwNZaRJsSKyTwd3Q1TUN7YziVgEsWnX7Q 2351
generax/nn/__init__.py sha256=SGQ4N0v3v16mK9F4bT21JOeCi51tVKPQJIwyf754H0A 279
generax/nn/grad_wrapper.py sha256=rT_XupR2QGOkkClTfSpZWJZfSGeNsBgk0urlYUv91bE 3489
generax/nn/input_convex.py sha256=-Y49Yo-NjZeMjXTbMSP_v3Rgtd29p596NJspULsBuoo 5619
generax/nn/layers.py sha256=dYFpG_6EK7RzZ_5DUhJojCMzjiHl8FsAejQ7orhY0jY 21832
generax/nn/lipschitz.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
generax/nn/neural_ode.py sha256=aYeeqPPVztu1GCiIm--KW4mpn10lB-nvLb5T3hpkSO0 8702
generax/nn/resnet.py sha256=Z8B7ra7O4bpQh43GIQJ2XZjdxfyVM0oBhnqVeqYFge8 10408
generax/nn/resnet_blocks.py sha256=TI9NbycKP4_r3YX3NyKA2zJSwpj4CIu_GO8ARXo1p-I 12493
generax/nn/time_condition.py sha256=AmxDfL7Vpbv-z60IR5X_P0vvhOjL8xjsDbXZQyTmLag 2831
generax/nn/unet.py sha256=RW3fR6c-S1XvgkjRJ3WDvWaCPyI8T2i9yZmho-8ImP8 16167
generax/tests/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
generax/tests/flow_tests.py sha256=HzLsBIaH2AIpebFqJX6i6Hi5hhnUVZVTy_Z8S1yVWvE 2879
generax/util/__init__.py sha256=PeNM4zZQxSwsG05ViSSRWNLLDHy8nQ9-TkUOiLAgnc8 142
generax/util/iterative.py sha256=_GVVUiPlvHUz2RjIkR_5WnFl5EG79XaCbA8sVebjuH8 10035
generax/util/misc.py sha256=vChkktOj7dtfcpEwx6Fv0wtQRGUyf8MVg7wGNSm8Q6o 5075
generax/util/save.py sha256=i6m3DiHQNqW9V44d0miTmlaGkceRSYRcXLL_TqGM-8s 2620
generax/util/spectral_norm.py sha256=8JzEO0m5s5oTaSwy5EYVXCZX6VTr-9PIJxoTmIbBa1A 1364
generax/util/vol_change.py sha256=bZIHA_pOvXPZ-OgrY5Kin479kQqxsuV_7JaDzyfCK24 2224
misc/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
misc/cnf_neals_funnel.py sha256=t885fhFXLozFh9F6SYFt1FCrjeLd15LMJqC72H7rwqg 8503
misc/cnf_vi.py sha256=jzwV8-zYhmfci8vhnHDuFBvev9blLSeNZJmMib9DlZw 9559
misc/cnf_vi_bare_bones.py sha256=JJuKeHxj1EqRq8EQquh9yecQiOrtxLmLrI4-ort4U9w 8548
misc/ode_jvp.py sha256=GUqCucdvGNqPkD8MHqZznrTtsue325PaptyT4ahpotc 1864
misc/realnvp_vi.py sha256=MTKOZILCBWFghPPyQjiJVkHsircPxEO_Ja_T2iIMlgU 3963
misc/vi/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
misc/vi/cnf.py sha256=Scx0-UpEhyDPmt2HLkJ6oxcF-2eV4m85pzTWRkMI2TE 14971
misc/vi/cnf_gmm.py sha256=MCWuKjYyrBF8Hd9S68TljQJqOSfAZONOJ0MQAM7Txt8 15928
misc/vi/polynomial.py sha256=x_V-XFJLSffh6YCjIHSkq_LuapBs_nDkKELgZeich4w 2043
generax-0.1.2.dist-info/METADATA sha256=yO6ktkrum4WHKIGICayYK1QrjR9zXmpdDE8AlqxQSKY 2768
generax-0.1.2.dist-info/WHEEL sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A 92
generax-0.1.2.dist-info/top_level.txt sha256=8_Ca9lZri3Xxadw6bASGK0DrIwlKL_P3lfcQX1PsiGc 22
generax-0.1.2.dist-info/RECORD — —

top_level.txt

examples
generax
misc