mlfab

View on PyPIReverse Dependencies (0)

0.1.14 mlfab-0.1.14-py3-none-any.whl

Wheel Details

Project: mlfab
Version: 0.1.14
Filename: mlfab-0.1.14-py3-none-any.whl
Download: [link]
Size: 202565
MD5: 78e81f147acb2dd20e2dbca8f6e53160
SHA256: 8182041f8cad4b9b9df386541c575c7807809dda3ac69776311197d521f8156c
Uploaded: 2024-05-19 23:46:16 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: mlfab
Version: 0.1.14
Summary: A collection of core machine learning tools
Author: Benjamin Bolte
Home-Page: https://github.com/dpshai/mlfab
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Requires-Python: >=3.11
Requires-Dist: dpshdl
Requires-Dist: GitPython
Requires-Dist: numpy
Requires-Dist: omegaconf
Requires-Dist: psutil
Requires-Dist: sarfile
Requires-Dist: six
Requires-Dist: smart-open
Requires-Dist: tensorboard
Requires-Dist: torch
Requires-Dist: torchaudio
Requires-Dist: torchvision
Requires-Dist: wcwidth
Requires-Dist: black; extra == "dev"
Requires-Dist: darglint; extra == "dev"
Requires-Dist: mypy; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-timeout; extra == "dev"
Requires-Dist: ruff; extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 8642 characters]

WHEEL

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

RECORD

Path Digest Size
mlfab/__init__.py sha256=lJIwGhNgjO3IITmn5agSzr_6KF0KK_JZUzxoEisASe8 24822
mlfab/py.typed sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/requirements.txt sha256=py4OZGMgeVKYTDdCORRA1dvb6tmFFDY4fbMvt6BkLbY 132
mlfab/core/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/core/conf.py sha256=rNyJfGClevUFJwJHF0ANYTUbpNMyJFADriwju81liLc 7429
mlfab/core/state.py sha256=ipzYT3glF-5WCczS-6onK2Fpc814lLAbnG5WvuiLepI 2264
mlfab/nn/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/nn/activations.py sha256=mhBhKLwzEvA-JZjIzriL72R1CPn6EndZZoS_3ZzbJiY 5761
mlfab/nn/embeddings.py sha256=B2_EXB6jwn9zB1ymsfHCKCpqjl1vdUmRzAwi1cDHM7Q 13865
mlfab/nn/functions.py sha256=lZlLEgAXgfCaWZMRUR-C6Illzhls6QES2u5ad2RVhok 19486
mlfab/nn/init.py sha256=U732kuIDkxqKM-gh46PTVfFPR-WiIjZVNbt7kOmU7DA 5167
mlfab/nn/kmeans.py sha256=cNMTpR7Ij2hPhEpnbwpO96L0VSdav1NbS5R_3G5lVr0 2669
mlfab/nn/lora.py sha256=1EOLsqm3sWlNP2Dj6gx3BxlkiA0v0rQAsSqFlUWsJ4c 56505
mlfab/nn/losses.py sha256=aNwLt4CU89HkcXksdwKNnDjrBHblGV-0Bsyvz784FZM 25992
mlfab/nn/lr_schedulers.py sha256=69lHDg4qpAGBdcAWaKU1xapy5UWYv1TDrVgwWyywSlU 9621
mlfab/nn/norms.py sha256=NcqTg9o4yvA1oGGusXM1Sq7A6yFHf6PZoNF4lk85lrk 16259
mlfab/nn/optimizers.py sha256=EjAlxHZdpKpwFt5TBnRlMdMJyA2uxst0fxCHdQ1TdAQ 14960
mlfab/nn/parallel.py sha256=EgMcrfpv1UHeMSUJ6Pe01vZvekiwLZbvAO3mzt9Vaes 43775
mlfab/nn/architectures/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/nn/architectures/attention.py sha256=qs7YFy5smESMI0vv6Ih4_IICXf7pkPugEv54wJSCMOM 43310
mlfab/nn/architectures/bifpn.py sha256=9aa2ZhKqiVQeSq6hzy3fxPWHkxK462QwpAmlXu6-G9k 6508
mlfab/nn/architectures/monotonic_attention.py sha256=jxRA-UF5_t0hSqZbuZQU9YiEhS6E0Za3UGR0rML1Ek4 13555
mlfab/nn/architectures/next_token.py sha256=FRLrXOi4c9IF_ff3p7I2jyjpUek52C_-bqkonbUZVsI 5566
mlfab/nn/architectures/rnn.py sha256=rAZF-pCmIhbQ95ZcbuYq8VQdJQOEBySQbwwwCmYUylk 4896
mlfab/nn/architectures/rwkv.py sha256=a024KWhST4BgHyhnAcU04Daf48RXof-RcXkuF_15XOw 26111
mlfab/nn/architectures/unet.py sha256=9xDNt4S7ny5qU9q9dBAdK_WKZNQYIGRNe6HS0oVorfw 9015
mlfab/nn/device/__init__.py sha256=ornnawDpI5mTaa48J50s3KlkKO_GBAgsibLJd-2ne4g 239
mlfab/nn/device/auto.py sha256=2cJTbZn2MLL5az9rq5I_iSa06u_WqoZBcpIaDyZtUfw 688
mlfab/nn/device/base.py sha256=UuV0EmKcNiCgQSGJcXLteE5J_nR_3FnjDK3I4j4wsIs 4186
mlfab/nn/device/cpu.py sha256=M-3KjBLK-n4RbsxpTSfHBgP0vsYLdC6Z8dKVvD5ORAE 519
mlfab/nn/device/gpu.py sha256=HWcuqI0a9wT98FSEgFI6lwz14HyZpEa3bFNeR_KkBeY 1167
mlfab/nn/device/metal.py sha256=aSlYJ219ZVF9Np0h6GX3f1qSD2DjeQjPVfH8jPKEtro 842
mlfab/nn/diffusion/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/nn/diffusion/consistency.py sha256=ppBpwZ4CG9UTCaV3yamNf7ecqRrwBUcocCa4xntiV1c 11894
mlfab/nn/diffusion/gaussian.py sha256=Zh4sSquUxu7OiZGyyBnIIKILfAst2YgZbpw-mmDU6rE 20251
mlfab/nn/diffusion/ode.py sha256=MQTdRcTPvxtzlvWSL4nX-icRUko6ig2fbBY0CuGbdaM 3823
mlfab/nn/quantization/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/nn/quantization/fsq.py sha256=ifqNXYZcWgt6zyaJIY3atYDoRw5tSV7l8PC-mDAJeGM 4295
mlfab/nn/quantization/lfq.py sha256=VipcUoKEpoS-NcfL5LHPsLEcd5eZoocj5gpeyw_XsgA 6897
mlfab/nn/quantization/vq.py sha256=x1GLdr_f6P6-UN-3ovfJ8XJjjDmwq-EPEeZr7nhL6fU 13683
mlfab/nn/triton/__init__.py sha256=ZHMYdP9e-Zviicn-4Rsvtvbp19O8LjSuJVwE8J_Tpuw 772
mlfab/nn/triton/kmeans.py sha256=cdEsCWP6xoIvQckYqUsbOYh-56LXo5SlDw3l1Fnkx9E 1969
mlfab/nn/triton/lion.py sha256=K8QasYk_E8mJ_QPyVfkic4cWPlGFSbEitJ5KngPwKE8 1688
mlfab/nn/triton/monotonic_attention.py sha256=1WCl2u4sCNG8ZtzaBSAMbxsh7R-b_ytQ8JgPLFtbCjs 10435
mlfab/nn/triton/rwkv.py sha256=4BUi5nRvaUMUjrd6rslrPpTzieshF4P7n2GgCH44nic 29357
mlfab/task/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/task/base.py sha256=WkJKfPwrHzgdR0st8zNw9Y0xtNsXxmrrNTY7eaRzwtU 7960
mlfab/task/logger.py sha256=3gl0M_LP12t7pe2tVflZgiWHLO0vaatKL2yu5hfLDm0 54922
mlfab/task/script.py sha256=QL5VHM7ZR4lstk8PED8-X3uBgKtsvu6KfT7k-eOkNK0 1375
mlfab/task/task.py sha256=CeBZ2gpmCtuhatElBzNLiMUbqkK0utNB7Spm3gAgPm0 1750
mlfab/task/launchers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/task/launchers/base.py sha256=OJ9lEQkNUn_5yq0k0hKw-vQUo665YpsdMHLxZUgXDn0 735
mlfab/task/launchers/cli.py sha256=MME6Sr8mJKNvEoH56NIjVDeRz2d2x02JE9o4UOW1FSw 2311
mlfab/task/launchers/multi_process.py sha256=wMcx2sCukI9sL9fkYxFHF959CYbVLOx1UiOGWKbei-8 1753
mlfab/task/launchers/single_process.py sha256=dKnqxl8roRoco9z176lxRQTkO887FDer0CMhO7hfgGQ 854
mlfab/task/launchers/slurm.py sha256=MAEfLTIVa6l0qXAtFNVHBI70V9F8oseUXdqzyGRmxO8 16581
mlfab/task/launchers/staged.py sha256=Pv8KdQwIO5FxIRzMv7dar3QRoUYmKXADCl_NC0Axxyk 1106
mlfab/task/loggers/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/task/loggers/json.py sha256=U-Gi26xIrmLiNnM4HDc7cY5ZiuNwV2u8oO_RTfN_v58 3967
mlfab/task/loggers/state.py sha256=0kF-bapTRlLBVRjiSST0jNwt7WAxbHplwk8us1HMybA 1578
mlfab/task/loggers/stdout.py sha256=Rn0G-HMgcvrBxXtmUgdKPN9L_PPLhgHOnSKeuYr2wLg 6783
mlfab/task/loggers/tensorboard.py sha256=1VVSyeKxa_FUevB1TyeFndhHHXNPL3x-sJDt-6_cYJU 9598
mlfab/task/mixins/__init__.py sha256=XOOXCUJASdEXBUpRy5cXVD0_M5i2GrMN1JJHXtjrCNQ 1123
mlfab/task/mixins/artifacts.py sha256=fDzd2uSGw3ljsNeDmp0Srlw8Xr08NHLGFb5VQ5NvGZg 3762
mlfab/task/mixins/checkpointing.py sha256=SCIp7Rwf9Xrl_z886uHEeYJddjB8kU-6UTbxh0HFUy4 6800
mlfab/task/mixins/compile.py sha256=jz3YeVATYX7PyGRbHqAfsxlBaXmHTc8VePAVU_iVGGk 2803
mlfab/task/mixins/cpu_stats.py sha256=-U0GM0b82J5tWXHGC-rUC67wXDGXKSP3dmfo25HjEac 9327
mlfab/task/mixins/data_loader.py sha256=1hH_-nni_eE8FSNqaTa5XBB_KZ173PQc4TPZKLuM5ek 4222
mlfab/task/mixins/device.py sha256=sBGM37Wt4QGewrltjmtV94o_Wj54ii04YgE8rZ5XGLg 1202
mlfab/task/mixins/gpu_stats.py sha256=IMIY9BuL4e0LY3VQLvEln3ho3hzdvNaTo_WG17eEWEs 8737
mlfab/task/mixins/logger.py sha256=K02gT7W9ZZHD_QyR4IkFIoVw7fWMs2Iyi4feYQOZTFQ 9394
mlfab/task/mixins/mixed_precision.py sha256=DmlEm_dlgA99id5cJ9AdzDnmDz68Y7kA-sCcUXIX--o 7762
mlfab/task/mixins/optimizer.py sha256=uo_07K_-C4x-HZ3cfqda5hYyu0uDa1DTBgqteEdHgQo 4997
mlfab/task/mixins/pretrained.py sha256=8sDlvQ8S_SnCvSfLdwLDOPMDAMMUyXqXcW706NfXKQA 2407
mlfab/task/mixins/process.py sha256=4_uwEkrSYL7KtmtxwJ9AxTVtGbl7h0IBFQh1PkvfIx8 1452
mlfab/task/mixins/profiler.py sha256=XStPDQ8DbYdtVm8JQWuKm0OmLNZLUsMCgGSd6aqtCdA 6174
mlfab/task/mixins/runnable.py sha256=mxSvd21EEG-IOiyUUIYfWLCQZn_mGZvTPHce5tCItcI 1852
mlfab/task/mixins/step_wrapper.py sha256=M14imA9BMmFBxQZaVZPvdj4HhcvPop2kJCbw772VN5I 1327
mlfab/task/mixins/train.py sha256=iTjjarIN30toG4OwE0MSCMrsY5uGRZviCZpDnyS6NaQ 24131
mlfab/utils/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/utils/experiments.py sha256=Aq25ywj7AtuRdhDvT38AKxb2fKO9I5mhVOc-xeMrXqg 25822
mlfab/utils/io.py sha256=N_BmXxzJLbT1Aw6w-Z0rAC--ymzBEOvzubYxMwg3Lr4 5233
mlfab/utils/logging.py sha256=XeOsILF4nChrA39nt_Ea_NwZDecopbGbGAKDXmsqw4Q 5863
mlfab/utils/text.py sha256=cu-bOcLJUcQ-JfjGhhTjhe_xbftcdsgamEWLJET9wO8 10673
mlfab/utils/tokens.py sha256=nXyN0r17U8KCAD7R9aMYUy49YI3UkfsmM88PKrlcJwg 15009
mlfab/utils/data/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mlfab/utils/data/collate.py sha256=aa3jMxTN3JdGEWbqXPkvUwYkWCw4U2Do0GZlsxk6my8 7467
mlfab/utils/data/dataset.py sha256=MSoA4qMC9IDVvnK26AxgSpp57p5eEkwzcPeIbwjshfw 2761
mlfab/utils/data/error_handling.py sha256=j1QzkPt46cmUoJ7593MfPCmhoPZaDyhZiSQBQIlsV-A 9453
mlfab/utils/data/transforms.py sha256=OwN5djZTLH5Y4UrSy4J4ZJL8wHhUk4Bvht2x4zyoRok 7342
mlfab-0.1.14.dist-info/LICENSE sha256=HCN2bImAzUOXldAZZI7JZ9PYq6OwMlDAP_PpX1HnuN0 1071
mlfab-0.1.14.dist-info/METADATA sha256=LPn6xSr-iq-e73E9ZkKZleTI7QFNGE6du7rqC_yz3e8 9721
mlfab-0.1.14.dist-info/WHEEL sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ 92
mlfab-0.1.14.dist-info/top_level.txt sha256=GDRyQnCPZfSbSFeroxXXqDxwT4mZpO0OwQxPNOsy6RM 6
mlfab-0.1.14.dist-info/RECORD

top_level.txt

mlfab