dewml 0.1.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (341) hide show
  1. dewml-0.1.0/LICENSE +21 -0
  2. dewml-0.1.0/MANIFEST.in +3 -0
  3. dewml-0.1.0/PKG-INFO +1594 -0
  4. dewml-0.1.0/README.md +1471 -0
  5. dewml-0.1.0/pyproject.toml +310 -0
  6. dewml-0.1.0/setup.cfg +4 -0
  7. dewml-0.1.0/src/dew/__init__.py +143 -0
  8. dewml-0.1.0/src/dew/_model_types.py +9 -0
  9. dewml-0.1.0/src/dew/artifacts.py +117 -0
  10. dewml-0.1.0/src/dew/cache.py +65 -0
  11. dewml-0.1.0/src/dew/checkpoints/__init__.py +1740 -0
  12. dewml-0.1.0/src/dew/cli/__init__.py +1 -0
  13. dewml-0.1.0/src/dew/cli/config.py +180 -0
  14. dewml-0.1.0/src/dew/cli/gcloud.py +241 -0
  15. dewml-0.1.0/src/dew/cli/launch.py +599 -0
  16. dewml-0.1.0/src/dew/cli/main.py +136 -0
  17. dewml-0.1.0/src/dew/cli/ssh_config.py +69 -0
  18. dewml-0.1.0/src/dew/cli/tpu.py +912 -0
  19. dewml-0.1.0/src/dew/cli/tpu_setup.py +232 -0
  20. dewml-0.1.0/src/dew/config/__init__.py +1220 -0
  21. dewml-0.1.0/src/dew/config/sweep.py +114 -0
  22. dewml-0.1.0/src/dew/coordination.py +438 -0
  23. dewml-0.1.0/src/dew/data/__init__.py +57 -0
  24. dewml-0.1.0/src/dew/data/chat.py +733 -0
  25. dewml-0.1.0/src/dew/data/dataset.py +1772 -0
  26. dewml-0.1.0/src/dew/data/image_augmentation.py +154 -0
  27. dewml-0.1.0/src/dew/data/images.py +666 -0
  28. dewml-0.1.0/src/dew/data/online_loader.py +438 -0
  29. dewml-0.1.0/src/dew/data/preferences.py +165 -0
  30. dewml-0.1.0/src/dew/data/processors.py +58 -0
  31. dewml-0.1.0/src/dew/data/prompts.py +271 -0
  32. dewml-0.1.0/src/dew/data/providers.py +605 -0
  33. dewml-0.1.0/src/dew/data/rows.py +99 -0
  34. dewml-0.1.0/src/dew/data/sources/av_utils.py +118 -0
  35. dewml-0.1.0/src/dew/data/sources/hf.py +241 -0
  36. dewml-0.1.0/src/dew/data/sources/hf_stream.py +298 -0
  37. dewml-0.1.0/src/dew/data/sources/pytorch.py +60 -0
  38. dewml-0.1.0/src/dew/data/sources/text.py +574 -0
  39. dewml-0.1.0/src/dew/data/sources/tfds.py +281 -0
  40. dewml-0.1.0/src/dew/data/streaming.py +93 -0
  41. dewml-0.1.0/src/dew/data/text.py +199 -0
  42. dewml-0.1.0/src/dew/data/tokens.py +406 -0
  43. dewml-0.1.0/src/dew/data/video.py +172 -0
  44. dewml-0.1.0/src/dew/decision/__init__.py +115 -0
  45. dewml-0.1.0/src/dew/decision/calibration.py +248 -0
  46. dewml-0.1.0/src/dew/decision/clef.py +162 -0
  47. dewml-0.1.0/src/dew/decision/config.py +191 -0
  48. dewml-0.1.0/src/dew/decision/data.py +251 -0
  49. dewml-0.1.0/src/dew/decision/head.py +333 -0
  50. dewml-0.1.0/src/dew/decision/images.py +53 -0
  51. dewml-0.1.0/src/dew/decision/laya.py +239 -0
  52. dewml-0.1.0/src/dew/decision/layout.py +505 -0
  53. dewml-0.1.0/src/dew/decision/metrics.py +129 -0
  54. dewml-0.1.0/src/dew/decision/model.py +107 -0
  55. dewml-0.1.0/src/dew/decision/objective.py +488 -0
  56. dewml-0.1.0/src/dew/decision/questions.py +343 -0
  57. dewml-0.1.0/src/dew/decision/scoring.py +155 -0
  58. dewml-0.1.0/src/dew/decision/task.py +627 -0
  59. dewml-0.1.0/src/dew/diffusion/__init__.py +72 -0
  60. dewml-0.1.0/src/dew/diffusion/block.py +392 -0
  61. dewml-0.1.0/src/dew/diffusion/discrete.py +377 -0
  62. dewml-0.1.0/src/dew/diffusion/presets.py +331 -0
  63. dewml-0.1.0/src/dew/diffusion/process.py +224 -0
  64. dewml-0.1.0/src/dew/diffusion/schedules/__init__.py +14 -0
  65. dewml-0.1.0/src/dew/diffusion/schedules/common.py +132 -0
  66. dewml-0.1.0/src/dew/diffusion/schedules/cosine.py +42 -0
  67. dewml-0.1.0/src/dew/diffusion/schedules/discrete.py +58 -0
  68. dewml-0.1.0/src/dew/diffusion/schedules/flow.py +87 -0
  69. dewml-0.1.0/src/dew/diffusion/schedules/karras.py +57 -0
  70. dewml-0.1.0/src/dew/diffusion/schedules/linear.py +30 -0
  71. dewml-0.1.0/src/dew/diffusion/schedules/source.py +100 -0
  72. dewml-0.1.0/src/dew/diffusion/schedules/source_grids.py +623 -0
  73. dewml-0.1.0/src/dew/diffusion/schedules/source_policy.py +626 -0
  74. dewml-0.1.0/src/dew/diffusion/schedules/sqrt.py +36 -0
  75. dewml-0.1.0/src/dew/diffusion/transforms.py +347 -0
  76. dewml-0.1.0/src/dew/eval/__init__.py +31 -0
  77. dewml-0.1.0/src/dew/eval/__main__.py +26 -0
  78. dewml-0.1.0/src/dew/eval/common.py +113 -0
  79. dewml-0.1.0/src/dew/eval/fid.py +282 -0
  80. dewml-0.1.0/src/dew/eval/harness.py +288 -0
  81. dewml-0.1.0/src/dew/eval/images.py +147 -0
  82. dewml-0.1.0/src/dew/eval/inception.py +207 -0
  83. dewml-0.1.0/src/dew/eval/lpips.py +144 -0
  84. dewml-0.1.0/src/dew/eval/psnr.py +60 -0
  85. dewml-0.1.0/src/dew/eval/ssim.py +128 -0
  86. dewml-0.1.0/src/dew/files.py +89 -0
  87. dewml-0.1.0/src/dew/inference/__init__.py +48 -0
  88. dewml-0.1.0/src/dew/inference/banks.py +769 -0
  89. dewml-0.1.0/src/dew/inference/clients.py +628 -0
  90. dewml-0.1.0/src/dew/inference/nccl.py +320 -0
  91. dewml-0.1.0/src/dew/inference/pages.py +137 -0
  92. dewml-0.1.0/src/dew/inference/pipeline.py +209 -0
  93. dewml-0.1.0/src/dew/inference/projections.py +86 -0
  94. dewml-0.1.0/src/dew/inference/rollouts.py +595 -0
  95. dewml-0.1.0/src/dew/inference/serving.py +896 -0
  96. dewml-0.1.0/src/dew/inference/serving_kernel.py +513 -0
  97. dewml-0.1.0/src/dew/inference/tasks.py +748 -0
  98. dewml-0.1.0/src/dew/inputs/__init__.py +162 -0
  99. dewml-0.1.0/src/dew/inputs/diffusion.py +690 -0
  100. dewml-0.1.0/src/dew/inputs/encoders.py +471 -0
  101. dewml-0.1.0/src/dew/interop/__init__.py +59 -0
  102. dewml-0.1.0/src/dew/interop/codecs.py +1579 -0
  103. dewml-0.1.0/src/dew/interop/components.py +56 -0
  104. dewml-0.1.0/src/dew/interop/config_records.py +45 -0
  105. dewml-0.1.0/src/dew/interop/dduf.py +72 -0
  106. dewml-0.1.0/src/dew/interop/decoder_families.py +123 -0
  107. dewml-0.1.0/src/dew/interop/decoder_parts.py +1342 -0
  108. dewml-0.1.0/src/dew/interop/diffusion.py +951 -0
  109. dewml-0.1.0/src/dew/interop/diffusion_gemma.py +300 -0
  110. dewml-0.1.0/src/dew/interop/families/__init__.py +4 -0
  111. dewml-0.1.0/src/dew/interop/families/bloom.py +88 -0
  112. dewml-0.1.0/src/dew/interop/families/deepseek.py +1192 -0
  113. dewml-0.1.0/src/dew/interop/families/deepseek_v41.py +393 -0
  114. dewml-0.1.0/src/dew/interop/families/falcon.py +105 -0
  115. dewml-0.1.0/src/dew/interop/families/gemma.py +650 -0
  116. dewml-0.1.0/src/dew/interop/families/glm.py +553 -0
  117. dewml-0.1.0/src/dew/interop/families/gpt2.py +331 -0
  118. dewml-0.1.0/src/dew/interop/families/gpt_bigcode.py +88 -0
  119. dewml-0.1.0/src/dew/interop/families/gpt_neox.py +179 -0
  120. dewml-0.1.0/src/dew/interop/families/gpt_oss.py +102 -0
  121. dewml-0.1.0/src/dew/interop/families/kimi.py +473 -0
  122. dewml-0.1.0/src/dew/interop/families/llama.py +155 -0
  123. dewml-0.1.0/src/dew/interop/families/llama4.py +170 -0
  124. dewml-0.1.0/src/dew/interop/families/masked_diffusion.py +371 -0
  125. dewml-0.1.0/src/dew/interop/families/modernbert.py +222 -0
  126. dewml-0.1.0/src/dew/interop/families/nemotron_h.py +204 -0
  127. dewml-0.1.0/src/dew/interop/families/olmo.py +69 -0
  128. dewml-0.1.0/src/dew/interop/families/opt.py +98 -0
  129. dewml-0.1.0/src/dew/interop/families/phi.py +72 -0
  130. dewml-0.1.0/src/dew/interop/families/phi3.py +128 -0
  131. dewml-0.1.0/src/dew/interop/families/qwen.py +481 -0
  132. dewml-0.1.0/src/dew/interop/families/starcoder2.py +87 -0
  133. dewml-0.1.0/src/dew/interop/flaxdiff.py +290 -0
  134. dewml-0.1.0/src/dew/interop/generation_config.py +565 -0
  135. dewml-0.1.0/src/dew/interop/gguf.py +224 -0
  136. dewml-0.1.0/src/dew/interop/harbor.py +704 -0
  137. dewml-0.1.0/src/dew/interop/hf_decoders.py +927 -0
  138. dewml-0.1.0/src/dew/interop/hub.py +24 -0
  139. dewml-0.1.0/src/dew/interop/inception_fid.py +248 -0
  140. dewml-0.1.0/src/dew/interop/mamba2.py +242 -0
  141. dewml-0.1.0/src/dew/interop/pickles.py +117 -0
  142. dewml-0.1.0/src/dew/interop/pipeline_assembly.py +985 -0
  143. dewml-0.1.0/src/dew/interop/pretrained.py +1312 -0
  144. dewml-0.1.0/src/dew/interop/processors.py +598 -0
  145. dewml-0.1.0/src/dew/interop/safetensors_io.py +434 -0
  146. dewml-0.1.0/src/dew/interop/single_file.py +311 -0
  147. dewml-0.1.0/src/dew/interop/sources.py +195 -0
  148. dewml-0.1.0/src/dew/interop/streaming.py +249 -0
  149. dewml-0.1.0/src/dew/interop/torchax_fallback.py +366 -0
  150. dewml-0.1.0/src/dew/interop/verify.py +420 -0
  151. dewml-0.1.0/src/dew/interop/weights.py +189 -0
  152. dewml-0.1.0/src/dew/io.py +66 -0
  153. dewml-0.1.0/src/dew/logging.py +101 -0
  154. dewml-0.1.0/src/dew/lora.py +907 -0
  155. dewml-0.1.0/src/dew/nn/__init__.py +0 -0
  156. dewml-0.1.0/src/dew/nn/activations.py +89 -0
  157. dewml-0.1.0/src/dew/nn/attention.py +2225 -0
  158. dewml-0.1.0/src/dew/nn/attention_residuals.py +90 -0
  159. dewml-0.1.0/src/dew/nn/attention_sinks.py +53 -0
  160. dewml-0.1.0/src/dew/nn/audio.py +770 -0
  161. dewml-0.1.0/src/dew/nn/autoencoders/__init__.py +5 -0
  162. dewml-0.1.0/src/dew/nn/autoencoders/api.py +154 -0
  163. dewml-0.1.0/src/dew/nn/autoencoders/dc_ae.py +513 -0
  164. dewml-0.1.0/src/dew/nn/autoencoders/flux2.py +100 -0
  165. dewml-0.1.0/src/dew/nn/autoencoders/kl.py +108 -0
  166. dewml-0.1.0/src/dew/nn/autoencoders/pretrained.py +94 -0
  167. dewml-0.1.0/src/dew/nn/autoencoders/qwen_image.py +389 -0
  168. dewml-0.1.0/src/dew/nn/autoencoders/rae.py +516 -0
  169. dewml-0.1.0/src/dew/nn/autoencoders/sd_vae.py +64 -0
  170. dewml-0.1.0/src/dew/nn/autoencoders/vae.py +434 -0
  171. dewml-0.1.0/src/dew/nn/autoencoders/wan.py +509 -0
  172. dewml-0.1.0/src/dew/nn/backbones/__init__.py +30 -0
  173. dewml-0.1.0/src/dew/nn/backbones/causal_transformer.py +1942 -0
  174. dewml-0.1.0/src/dew/nn/backbones/decoder_block.py +989 -0
  175. dewml-0.1.0/src/dew/nn/backbones/decoder_stack.py +537 -0
  176. dewml-0.1.0/src/dew/nn/backbones/dit.py +119 -0
  177. dewml-0.1.0/src/dew/nn/backbones/edm2.py +162 -0
  178. dewml-0.1.0/src/dew/nn/backbones/flux.py +191 -0
  179. dewml-0.1.0/src/dew/nn/backbones/flux2.py +199 -0
  180. dewml-0.1.0/src/dew/nn/backbones/jepa.py +220 -0
  181. dewml-0.1.0/src/dew/nn/backbones/joint.py +296 -0
  182. dewml-0.1.0/src/dew/nn/backbones/layer_plan.py +192 -0
  183. dewml-0.1.0/src/dew/nn/backbones/mmdit.py +303 -0
  184. dewml-0.1.0/src/dew/nn/backbones/qwen_image.py +261 -0
  185. dewml-0.1.0/src/dew/nn/backbones/sd3.py +165 -0
  186. dewml-0.1.0/src/dew/nn/backbones/ssm_dit.py +83 -0
  187. dewml-0.1.0/src/dew/nn/backbones/unet.py +135 -0
  188. dewml-0.1.0/src/dew/nn/backbones/unet3d.py +105 -0
  189. dewml-0.1.0/src/dew/nn/backbones/unet_condition.py +312 -0
  190. dewml-0.1.0/src/dew/nn/backbones/uvit.py +261 -0
  191. dewml-0.1.0/src/dew/nn/backbones/video_dit.py +63 -0
  192. dewml-0.1.0/src/dew/nn/backbones/wan.py +250 -0
  193. dewml-0.1.0/src/dew/nn/backbones/z_image.py +235 -0
  194. dewml-0.1.0/src/dew/nn/blocks.py +285 -0
  195. dewml-0.1.0/src/dew/nn/conv.py +251 -0
  196. dewml-0.1.0/src/dew/nn/deepseek_v4.py +957 -0
  197. dewml-0.1.0/src/dew/nn/diffusion_gemma.py +357 -0
  198. dewml-0.1.0/src/dew/nn/dit.py +666 -0
  199. dewml-0.1.0/src/dew/nn/dsa_kpool.py +450 -0
  200. dewml-0.1.0/src/dew/nn/dspark.py +190 -0
  201. dewml-0.1.0/src/dew/nn/engram.py +326 -0
  202. dewml-0.1.0/src/dew/nn/fake_quant.py +117 -0
  203. dewml-0.1.0/src/dew/nn/gemma3n.py +225 -0
  204. dewml-0.1.0/src/dew/nn/gemma4_moe.py +99 -0
  205. dewml-0.1.0/src/dew/nn/gpt_oss.py +122 -0
  206. dewml-0.1.0/src/dew/nn/hyper_connections.py +206 -0
  207. dewml-0.1.0/src/dew/nn/inputs.py +720 -0
  208. dewml-0.1.0/src/dew/nn/kda.py +283 -0
  209. dewml-0.1.0/src/dew/nn/kernels/__init__.py +16 -0
  210. dewml-0.1.0/src/dew/nn/kernels/decode_attention.py +159 -0
  211. dewml-0.1.0/src/dew/nn/kernels/delta_rule.py +114 -0
  212. dewml-0.1.0/src/dew/nn/kernels/generation.py +96 -0
  213. dewml-0.1.0/src/dew/nn/kernels/grouped_matmul.py +181 -0
  214. dewml-0.1.0/src/dew/nn/kernels/ragged_dot.py +444 -0
  215. dewml-0.1.0/src/dew/nn/kernels/ssd.py +350 -0
  216. dewml-0.1.0/src/dew/nn/kv_cache.py +569 -0
  217. dewml-0.1.0/src/dew/nn/linear.py +671 -0
  218. dewml-0.1.0/src/dew/nn/llama4.py +104 -0
  219. dewml-0.1.0/src/dew/nn/mixer_base.py +125 -0
  220. dewml-0.1.0/src/dew/nn/mixers/__init__.py +4 -0
  221. dewml-0.1.0/src/dew/nn/mixers/attention.py +980 -0
  222. dewml-0.1.0/src/dew/nn/mixers/gated_delta_net.py +50 -0
  223. dewml-0.1.0/src/dew/nn/mixers/mamba2.py +592 -0
  224. dewml-0.1.0/src/dew/nn/mixers/mlp.py +88 -0
  225. dewml-0.1.0/src/dew/nn/mla.py +783 -0
  226. dewml-0.1.0/src/dew/nn/mobilenet.py +431 -0
  227. dewml-0.1.0/src/dew/nn/moe.py +1150 -0
  228. dewml-0.1.0/src/dew/nn/mp.py +171 -0
  229. dewml-0.1.0/src/dew/nn/multimodal.py +494 -0
  230. dewml-0.1.0/src/dew/nn/precision.py +252 -0
  231. dewml-0.1.0/src/dew/nn/protocols.py +304 -0
  232. dewml-0.1.0/src/dew/nn/rope.py +384 -0
  233. dewml-0.1.0/src/dew/nn/safety.py +56 -0
  234. dewml-0.1.0/src/dew/nn/scan_orders.py +148 -0
  235. dewml-0.1.0/src/dew/nn/scatter.py +13 -0
  236. dewml-0.1.0/src/dew/nn/sharding.py +639 -0
  237. dewml-0.1.0/src/dew/nn/sparse_selection.py +140 -0
  238. dewml-0.1.0/src/dew/nn/ssm.py +276 -0
  239. dewml-0.1.0/src/dew/nn/text_encoders.py +858 -0
  240. dewml-0.1.0/src/dew/nn/vision/__init__.py +137 -0
  241. dewml-0.1.0/src/dew/nn/vision/common.py +118 -0
  242. dewml-0.1.0/src/dew/nn/vision/deepseek_v41.py +212 -0
  243. dewml-0.1.0/src/dew/nn/vision/gemma3n.py +235 -0
  244. dewml-0.1.0/src/dew/nn/vision/gemma4.py +394 -0
  245. dewml-0.1.0/src/dew/nn/vision/llama4.py +296 -0
  246. dewml-0.1.0/src/dew/nn/vision/qwen35.py +342 -0
  247. dewml-0.1.0/src/dew/nn/vision/siglip.py +258 -0
  248. dewml-0.1.0/src/dew/objectives/__init__.py +4 -0
  249. dewml-0.1.0/src/dew/objectives/base.py +1055 -0
  250. dewml-0.1.0/src/dew/objectives/diffusion/__init__.py +28 -0
  251. dewml-0.1.0/src/dew/objectives/diffusion/adversarial.py +330 -0
  252. dewml-0.1.0/src/dew/objectives/diffusion/alignment.py +192 -0
  253. dewml-0.1.0/src/dew/objectives/diffusion/block.py +439 -0
  254. dewml-0.1.0/src/dew/objectives/diffusion/config.py +511 -0
  255. dewml-0.1.0/src/dew/objectives/diffusion/consistency.py +446 -0
  256. dewml-0.1.0/src/dew/objectives/diffusion/end_to_end.py +272 -0
  257. dewml-0.1.0/src/dew/objectives/diffusion/few_step.py +317 -0
  258. dewml-0.1.0/src/dew/objectives/diffusion/guidance_distillation.py +134 -0
  259. dewml-0.1.0/src/dew/objectives/diffusion/masked.py +242 -0
  260. dewml-0.1.0/src/dew/objectives/diffusion/objective.py +867 -0
  261. dewml-0.1.0/src/dew/objectives/distillation.py +256 -0
  262. dewml-0.1.0/src/dew/objectives/jepa/__init__.py +16 -0
  263. dewml-0.1.0/src/dew/objectives/jepa/config.py +86 -0
  264. dewml-0.1.0/src/dew/objectives/jepa/masking.py +148 -0
  265. dewml-0.1.0/src/dew/objectives/jepa/objective.py +258 -0
  266. dewml-0.1.0/src/dew/objectives/jepa/probes.py +149 -0
  267. dewml-0.1.0/src/dew/objectives/lm/__init__.py +4 -0
  268. dewml-0.1.0/src/dew/objectives/lm/chunked.py +862 -0
  269. dewml-0.1.0/src/dew/objectives/lm/config.py +288 -0
  270. dewml-0.1.0/src/dew/objectives/lm/objective.py +1276 -0
  271. dewml-0.1.0/src/dew/objectives/rl/__init__.py +93 -0
  272. dewml-0.1.0/src/dew/objectives/rl/episodes.py +737 -0
  273. dewml-0.1.0/src/dew/objectives/rl/flow.py +442 -0
  274. dewml-0.1.0/src/dew/objectives/rl/grpo.py +363 -0
  275. dewml-0.1.0/src/dew/objectives/rl/journal.py +158 -0
  276. dewml-0.1.0/src/dew/objectives/rl/ppo.py +325 -0
  277. dewml-0.1.0/src/dew/objectives/rl/preference.py +117 -0
  278. dewml-0.1.0/src/dew/objectives/rl/records.py +70 -0
  279. dewml-0.1.0/src/dew/objectives/rl/rollout.py +154 -0
  280. dewml-0.1.0/src/dew/objectives/rl/scheduler.py +572 -0
  281. dewml-0.1.0/src/dew/objectives/rl/sessions.py +674 -0
  282. dewml-0.1.0/src/dew/objectives/rl/sources.py +372 -0
  283. dewml-0.1.0/src/dew/objectives/rl/verl.py +314 -0
  284. dewml-0.1.0/src/dew/objectives/supervised.py +138 -0
  285. dewml-0.1.0/src/dew/pool.py +123 -0
  286. dewml-0.1.0/src/dew/position.py +98 -0
  287. dewml-0.1.0/src/dew/py.typed +0 -0
  288. dewml-0.1.0/src/dew/records.py +138 -0
  289. dewml-0.1.0/src/dew/registry.py +1250 -0
  290. dewml-0.1.0/src/dew/rl/__init__.py +46 -0
  291. dewml-0.1.0/src/dew/rl/_sandbox_exec.py +28 -0
  292. dewml-0.1.0/src/dew/rl/advantage.py +170 -0
  293. dewml-0.1.0/src/dew/rl/sandbox.py +670 -0
  294. dewml-0.1.0/src/dew/rl/surrogate.py +363 -0
  295. dewml-0.1.0/src/dew/sampling/__init__.py +72 -0
  296. dewml-0.1.0/src/dew/sampling/decoding.py +968 -0
  297. dewml-0.1.0/src/dew/sampling/flow.py +216 -0
  298. dewml-0.1.0/src/dew/sampling/guidance.py +379 -0
  299. dewml-0.1.0/src/dew/sampling/guided.py +164 -0
  300. dewml-0.1.0/src/dew/sampling/pipelines.py +854 -0
  301. dewml-0.1.0/src/dew/sampling/sample.py +144 -0
  302. dewml-0.1.0/src/dew/sampling/solvers/__init__.py +25 -0
  303. dewml-0.1.0/src/dew/sampling/solvers/brownian.py +176 -0
  304. dewml-0.1.0/src/dew/sampling/solvers/common.py +144 -0
  305. dewml-0.1.0/src/dew/sampling/solvers/dpm.py +388 -0
  306. dewml-0.1.0/src/dew/sampling/solvers/gaussian.py +230 -0
  307. dewml-0.1.0/src/dew/sampling/solvers/sigma.py +300 -0
  308. dewml-0.1.0/src/dew/sampling/solvers/unipc.py +213 -0
  309. dewml-0.1.0/src/dew/sampling/strategies.py +899 -0
  310. dewml-0.1.0/src/dew/sampling/text.py +953 -0
  311. dewml-0.1.0/src/dew/sampling/vocabulary.py +188 -0
  312. dewml-0.1.0/src/dew/telemetry/__init__.py +0 -0
  313. dewml-0.1.0/src/dew/telemetry/devices.py +160 -0
  314. dewml-0.1.0/src/dew/telemetry/instrumentation.py +453 -0
  315. dewml-0.1.0/src/dew/telemetry/peaks.py +51 -0
  316. dewml-0.1.0/src/dew/telemetry/profile.py +341 -0
  317. dewml-0.1.0/src/dew/telemetry/records.py +149 -0
  318. dewml-0.1.0/src/dew/training/__init__.py +25 -0
  319. dewml-0.1.0/src/dew/training/display.py +540 -0
  320. dewml-0.1.0/src/dew/training/distributed.py +905 -0
  321. dewml-0.1.0/src/dew/training/evaluation.py +493 -0
  322. dewml-0.1.0/src/dew/training/execution.py +429 -0
  323. dewml-0.1.0/src/dew/training/host.py +255 -0
  324. dewml-0.1.0/src/dew/training/memory.py +341 -0
  325. dewml-0.1.0/src/dew/training/narrow.py +99 -0
  326. dewml-0.1.0/src/dew/training/optim.py +896 -0
  327. dewml-0.1.0/src/dew/training/posthoc.py +93 -0
  328. dewml-0.1.0/src/dew/training/quantization.py +870 -0
  329. dewml-0.1.0/src/dew/training/rungs.py +70 -0
  330. dewml-0.1.0/src/dew/training/runtime.py +323 -0
  331. dewml-0.1.0/src/dew/training/selection.py +50 -0
  332. dewml-0.1.0/src/dew/training/state.py +82 -0
  333. dewml-0.1.0/src/dew/training/tracker.py +661 -0
  334. dewml-0.1.0/src/dew/training/trainer.py +2266 -0
  335. dewml-0.1.0/src/dew/training/transaction.py +486 -0
  336. dewml-0.1.0/src/dewml.egg-info/PKG-INFO +1594 -0
  337. dewml-0.1.0/src/dewml.egg-info/SOURCES.txt +339 -0
  338. dewml-0.1.0/src/dewml.egg-info/dependency_links.txt +1 -0
  339. dewml-0.1.0/src/dewml.egg-info/entry_points.txt +2 -0
  340. dewml-0.1.0/src/dewml.egg-info/requires.txt +125 -0
  341. dewml-0.1.0/src/dewml.egg-info/top_level.txt +1 -0
dewml-0.1.0/LICENSE ADDED
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2024-2026 Ashish Kumar Singh
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
dewml-0.1.0/MANIFEST.in ADDED
@@ -0,0 +1,3 @@
1
+ # The sdist is the package's source: the tests run from a checkout, where
2
+ # their fixtures and support modules live, so none ship in it.
3
+ prune tests