topologies 0.0.1__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 (1726) hide show
  1. topologies-0.0.1/.gitignore +20 -0
  2. topologies-0.0.1/CHANGELOG.md +41 -0
  3. topologies-0.0.1/LICENSE +191 -0
  4. topologies-0.0.1/PKG-INFO +143 -0
  5. topologies-0.0.1/README.md +76 -0
  6. topologies-0.0.1/pyproject.toml +146 -0
  7. topologies-0.0.1/src/topologies/.training-generated-manifest.json +35 -0
  8. topologies-0.0.1/src/topologies/.training-materialized-manifest.json +35 -0
  9. topologies-0.0.1/src/topologies/__init__.py +89 -0
  10. topologies-0.0.1/src/topologies/_catalog.py +293 -0
  11. topologies-0.0.1/src/topologies/_compat.py +27 -0
  12. topologies-0.0.1/src/topologies/_execution.py +178 -0
  13. topologies-0.0.1/src/topologies/_util/__init__.py +1 -0
  14. topologies-0.0.1/src/topologies/_util/fs.py +98 -0
  15. topologies-0.0.1/src/topologies/_util/jsonsafe.py +42 -0
  16. topologies-0.0.1/src/topologies/_util/seed.py +19 -0
  17. topologies-0.0.1/src/topologies/_util/suggest.py +26 -0
  18. topologies-0.0.1/src/topologies/analysis/__init__.py +102 -0
  19. topologies-0.0.1/src/topologies/analysis/core.py +448 -0
  20. topologies-0.0.1/src/topologies/analysis/decode.py +186 -0
  21. topologies-0.0.1/src/topologies/analysis/engineering.py +878 -0
  22. topologies-0.0.1/src/topologies/analysis/integrations.py +70 -0
  23. topologies-0.0.1/src/topologies/analysis/profiling.py +122 -0
  24. topologies-0.0.1/src/topologies/analysis/records.py +123 -0
  25. topologies-0.0.1/src/topologies/analysis/statistics.py +211 -0
  26. topologies-0.0.1/src/topologies/analysis/visualization.py +135 -0
  27. topologies-0.0.1/src/topologies/artifact_store/__init__.py +31 -0
  28. topologies-0.0.1/src/topologies/artifact_store/client.py +402 -0
  29. topologies-0.0.1/src/topologies/artifact_store/exceptions.py +21 -0
  30. topologies-0.0.1/src/topologies/build/__init__.py +93 -0
  31. topologies-0.0.1/src/topologies/build/api.py +111 -0
  32. topologies-0.0.1/src/topologies/build/audit.py +103 -0
  33. topologies-0.0.1/src/topologies/build/budget.py +212 -0
  34. topologies-0.0.1/src/topologies/build/catalog.py +202 -0
  35. topologies-0.0.1/src/topologies/build/config.py +246 -0
  36. topologies-0.0.1/src/topologies/build/contract.py +85 -0
  37. topologies-0.0.1/src/topologies/build/params.py +92 -0
  38. topologies-0.0.1/src/topologies/build/presets.py +100 -0
  39. topologies-0.0.1/src/topologies/build/validation.py +75 -0
  40. topologies-0.0.1/src/topologies/catalog/losses/a2c.json +1 -0
  41. topologies-0.0.1/src/topologies/catalog/losses/adversarial-hinge.json +9 -0
  42. topologies-0.0.1/src/topologies/catalog/losses/binary-cross-entropy-logits.json +1 -0
  43. topologies-0.0.1/src/topologies/catalog/losses/bpr.json +9 -0
  44. topologies-0.0.1/src/topologies/catalog/losses/cross-entropy.json +1 -0
  45. topologies-0.0.1/src/topologies/catalog/losses/ctc.json +9 -0
  46. topologies-0.0.1/src/topologies/catalog/losses/density-nll.json +9 -0
  47. topologies-0.0.1/src/topologies/catalog/losses/dice.json +1 -0
  48. topologies-0.0.1/src/topologies/catalog/losses/gaussian-kl.json +1 -0
  49. topologies-0.0.1/src/topologies/catalog/losses/gaussian-nll.json +9 -0
  50. topologies-0.0.1/src/topologies/catalog/losses/generalized-iou.json +10 -0
  51. topologies-0.0.1/src/topologies/catalog/losses/huber.json +1 -0
  52. topologies-0.0.1/src/topologies/catalog/losses/info-nce.json +9 -0
  53. topologies-0.0.1/src/topologies/catalog/losses/mean-absolute-error.json +1 -0
  54. topologies-0.0.1/src/topologies/catalog/losses/mean-squared-error.json +1 -0
  55. topologies-0.0.1/src/topologies/catalog/losses/multi-resolution-stft.json +9 -0
  56. topologies-0.0.1/src/topologies/catalog/losses/pinball.json +9 -0
  57. topologies-0.0.1/src/topologies/catalog/losses/poly1-loss.json +1 -0
  58. topologies-0.0.1/src/topologies/catalog/losses/si-sdr.json +9 -0
  59. topologies-0.0.1/src/topologies/catalog/losses/sigmoid-focal.json +9 -0
  60. topologies-0.0.1/src/topologies/catalog/losses/smooth-l1.json +9 -0
  61. topologies-0.0.1/src/topologies/catalog/losses/softmax-focal.json +9 -0
  62. topologies-0.0.1/src/topologies/catalog/losses/triplet-margin.json +9 -0
  63. topologies-0.0.1/src/topologies/catalog/optimizers/adabelief.json +1 -0
  64. topologies-0.0.1/src/topologies/catalog/optimizers/adafactor.json +1 -0
  65. topologies-0.0.1/src/topologies/catalog/optimizers/adagrad.json +9 -0
  66. topologies-0.0.1/src/topologies/catalog/optimizers/adahessian.json +1 -0
  67. topologies-0.0.1/src/topologies/catalog/optimizers/adam-8bit.json +1 -0
  68. topologies-0.0.1/src/topologies/catalog/optimizers/adam-mini.json +1 -0
  69. topologies-0.0.1/src/topologies/catalog/optimizers/adam.json +1 -0
  70. topologies-0.0.1/src/topologies/catalog/optimizers/adamw.json +1 -0
  71. topologies-0.0.1/src/topologies/catalog/optimizers/ademamix.json +1 -0
  72. topologies-0.0.1/src/topologies/catalog/optimizers/asam.json +1 -0
  73. topologies-0.0.1/src/topologies/catalog/optimizers/ema.json +1 -0
  74. topologies-0.0.1/src/topologies/catalog/optimizers/fadam.json +1 -0
  75. topologies-0.0.1/src/topologies/catalog/optimizers/fisher-sam.json +1 -0
  76. topologies-0.0.1/src/topologies/catalog/optimizers/flora.json +1 -0
  77. topologies-0.0.1/src/topologies/catalog/optimizers/galore.json +1 -0
  78. topologies-0.0.1/src/topologies/catalog/optimizers/grokfast.json +1 -0
  79. topologies-0.0.1/src/topologies/catalog/optimizers/gsam.json +1 -0
  80. topologies-0.0.1/src/topologies/catalog/optimizers/kfac.json +1 -0
  81. topologies-0.0.1/src/topologies/catalog/optimizers/lamb.json +1 -0
  82. topologies-0.0.1/src/topologies/catalog/optimizers/lars.json +1 -0
  83. topologies-0.0.1/src/topologies/catalog/optimizers/lion.json +1 -0
  84. topologies-0.0.1/src/topologies/catalog/optimizers/lookahead.json +1 -0
  85. topologies-0.0.1/src/topologies/catalog/optimizers/model-soup.json +1 -0
  86. topologies-0.0.1/src/topologies/catalog/optimizers/muon.json +1 -0
  87. topologies-0.0.1/src/topologies/catalog/optimizers/novograd.json +1 -0
  88. topologies-0.0.1/src/topologies/catalog/optimizers/polynomial-decay.json +1 -0
  89. topologies-0.0.1/src/topologies/catalog/optimizers/qlora.json +1 -0
  90. topologies-0.0.1/src/topologies/catalog/optimizers/radam.json +1 -0
  91. topologies-0.0.1/src/topologies/catalog/optimizers/relora.json +1 -0
  92. topologies-0.0.1/src/topologies/catalog/optimizers/rmsprop.json +9 -0
  93. topologies-0.0.1/src/topologies/catalog/optimizers/sam.json +1 -0
  94. topologies-0.0.1/src/topologies/catalog/optimizers/schedule-free.json +1 -0
  95. topologies-0.0.1/src/topologies/catalog/optimizers/sgd.json +1 -0
  96. topologies-0.0.1/src/topologies/catalog/optimizers/shampoo.json +1 -0
  97. topologies-0.0.1/src/topologies/catalog/optimizers/sm3.json +1 -0
  98. topologies-0.0.1/src/topologies/catalog/optimizers/soap.json +1 -0
  99. topologies-0.0.1/src/topologies/catalog/optimizers/sophia.json +1 -0
  100. topologies-0.0.1/src/topologies/catalog/optimizers/swa.json +1 -0
  101. topologies-0.0.1/src/topologies/catalog/optimizers/swad.json +1 -0
  102. topologies-0.0.1/src/topologies/catalog/optimizers/swalp.json +1 -0
  103. topologies-0.0.1/src/topologies/catalog/optimizers/twa.json +1 -0
  104. topologies-0.0.1/src/topologies/catalog/schedulers/clr.json +1 -0
  105. topologies-0.0.1/src/topologies/catalog/schedulers/cooldown.json +1 -0
  106. topologies-0.0.1/src/topologies/catalog/schedulers/cosine-min.json +1 -0
  107. topologies-0.0.1/src/topologies/catalog/schedulers/cosine-restarts.json +1 -0
  108. topologies-0.0.1/src/topologies/catalog/schedulers/exponential.json +19 -0
  109. topologies-0.0.1/src/topologies/catalog/schedulers/grafted.json +1 -0
  110. topologies-0.0.1/src/topologies/catalog/schedulers/multistep.json +1 -0
  111. topologies-0.0.1/src/topologies/catalog/schedulers/noam.json +21 -0
  112. topologies-0.0.1/src/topologies/catalog/schedulers/onecycle.json +1 -0
  113. topologies-0.0.1/src/topologies/catalog/schedulers/reduce-on-plateau.json +1 -0
  114. topologies-0.0.1/src/topologies/catalog/schedulers/sgdr.json +1 -0
  115. topologies-0.0.1/src/topologies/catalog/schedulers/trapezoidal.json +21 -0
  116. topologies-0.0.1/src/topologies/catalog/schedulers/warmup-linear.json +1 -0
  117. topologies-0.0.1/src/topologies/catalog/schedulers/wsd.json +1 -0
  118. topologies-0.0.1/src/topologies/catalog/stability/activation-offload.json +1 -0
  119. topologies-0.0.1/src/topologies/catalog/stability/bf16-recipe.json +14 -0
  120. topologies-0.0.1/src/topologies/catalog/stability/compile-parity.json +4 -0
  121. topologies-0.0.1/src/topologies/catalog/stability/fsdp-wrap.json +17 -0
  122. topologies-0.0.1/src/topologies/catalog/stability/grad-checkpoint.json +1 -0
  123. topologies-0.0.1/src/topologies/catalog/stability/grad-clip.json +17 -0
  124. topologies-0.0.1/src/topologies/catalog/stability/llrd.json +18 -0
  125. topologies-0.0.1/src/topologies/catalog/stability/loss-scale.json +17 -0
  126. topologies-0.0.1/src/topologies/catalog/stability/nan-skip.json +1 -0
  127. topologies-0.0.1/src/topologies/catalog/stability/zero-partition.json +1 -0
  128. topologies-0.0.1/src/topologies/cli.py +294 -0
  129. topologies-0.0.1/src/topologies/data/__init__.py +85 -0
  130. topologies-0.0.1/src/topologies/data/adapters.py +162 -0
  131. topologies-0.0.1/src/topologies/data/collate.py +96 -0
  132. topologies-0.0.1/src/topologies/data/corpus.py +263 -0
  133. topologies-0.0.1/src/topologies/data/datasets.py +53 -0
  134. topologies-0.0.1/src/topologies/data/download.py +204 -0
  135. topologies-0.0.1/src/topologies/data/event_codec.py +62 -0
  136. topologies-0.0.1/src/topologies/data/integrations/__init__.py +24 -0
  137. topologies-0.0.1/src/topologies/data/integrations/karate.py +68 -0
  138. topologies-0.0.1/src/topologies/data/integrations/local.py +345 -0
  139. topologies-0.0.1/src/topologies/data/loaders.py +71 -0
  140. topologies-0.0.1/src/topologies/data/preprocessing.py +188 -0
  141. topologies-0.0.1/src/topologies/data/profiling.py +341 -0
  142. topologies-0.0.1/src/topologies/data/samplers.py +32 -0
  143. topologies-0.0.1/src/topologies/data/splits.py +158 -0
  144. topologies-0.0.1/src/topologies/data/stream.py +313 -0
  145. topologies-0.0.1/src/topologies/data/text.py +212 -0
  146. topologies-0.0.1/src/topologies/data/windows.py +90 -0
  147. topologies-0.0.1/src/topologies/datasets/.dataset-catalog-generated-manifest.json +6 -0
  148. topologies-0.0.1/src/topologies/datasets/__init__.py +25 -0
  149. topologies-0.0.1/src/topologies/datasets/api.py +101 -0
  150. topologies-0.0.1/src/topologies/datasets/artifacts.lock.json +5332 -0
  151. topologies-0.0.1/src/topologies/datasets/catalog.json +17755 -0
  152. topologies-0.0.1/src/topologies/datasets/core.py +511 -0
  153. topologies-0.0.1/src/topologies/datasets/integrations/.dataset-generated-manifest.json +21 -0
  154. topologies-0.0.1/src/topologies/datasets/integrations/__init__.py +1 -0
  155. topologies-0.0.1/src/topologies/datasets/integrations/ag_news/__init__.py +22 -0
  156. topologies-0.0.1/src/topologies/datasets/integrations/cifar10/__init__.py +55 -0
  157. topologies-0.0.1/src/topologies/datasets/integrations/cora/__init__.py +36 -0
  158. topologies-0.0.1/src/topologies/datasets/integrations/fashion_mnist/__init__.py +18 -0
  159. topologies-0.0.1/src/topologies/datasets/integrations/fsdd/__init__.py +54 -0
  160. topologies-0.0.1/src/topologies/datasets/integrations/iris/__init__.py +80 -0
  161. topologies-0.0.1/src/topologies/datasets/integrations/jena_climate/__init__.py +59 -0
  162. topologies-0.0.1/src/topologies/datasets/integrations/mnist/__init__.py +18 -0
  163. topologies-0.0.1/src/topologies/datasets/integrations/movielens_100k/__init__.py +43 -0
  164. topologies-0.0.1/src/topologies/datasets/integrations/oxford_iiit_pet/__init__.py +52 -0
  165. topologies-0.0.1/src/topologies/datasets/integrations/palmer_penguins/__init__.py +46 -0
  166. topologies-0.0.1/src/topologies/datasets/integrations/penn_fudan_pedestrians/__init__.py +52 -0
  167. topologies-0.0.1/src/topologies/datasets/integrations/speech_commands/__init__.py +41 -0
  168. topologies-0.0.1/src/topologies/datasets/integrations/uci_har/__init__.py +27 -0
  169. topologies-0.0.1/src/topologies/datasets/integrations/wikitext_2_raw/__init__.py +44 -0
  170. topologies-0.0.1/src/topologies/datasets/records/1000-genomes.json +13 -0
  171. topologies-0.0.1/src/topologies/datasets/records/2wikimultihopqa.json +12 -0
  172. topologies-0.0.1/src/topologies/datasets/records/3dpw.json +12 -0
  173. topologies-0.0.1/src/topologies/datasets/records/abo.json +12 -0
  174. topologies-0.0.1/src/topologies/datasets/records/acdc.json +13 -0
  175. topologies-0.0.1/src/topologies/datasets/records/activitynet-captions.json +13 -0
  176. topologies-0.0.1/src/topologies/datasets/records/activitynet.json +12 -0
  177. topologies-0.0.1/src/topologies/datasets/records/ade20k.json +13 -0
  178. topologies-0.0.1/src/topologies/datasets/records/adni.json +13 -0
  179. topologies-0.0.1/src/topologies/datasets/records/adroit.json +13 -0
  180. topologies-0.0.1/src/topologies/datasets/records/adult.json +13 -0
  181. topologies-0.0.1/src/topologies/datasets/records/aerorit.json +13 -0
  182. topologies-0.0.1/src/topologies/datasets/records/agora.json +12 -0
  183. topologies-0.0.1/src/topologies/datasets/records/aishell.json +12 -0
  184. topologies-0.0.1/src/topologies/datasets/records/ali-ccp.json +17 -0
  185. topologies-0.0.1/src/topologies/datasets/records/alimama.json +13 -0
  186. topologies-0.0.1/src/topologies/datasets/records/alpaca.json +13 -0
  187. topologies-0.0.1/src/topologies/datasets/records/amass.json +13 -0
  188. topologies-0.0.1/src/topologies/datasets/records/amazon-2018.json +13 -0
  189. topologies-0.0.1/src/topologies/datasets/records/amazon-2023.json +13 -0
  190. topologies-0.0.1/src/topologies/datasets/records/amazon-computers.json +13 -0
  191. topologies-0.0.1/src/topologies/datasets/records/ami.json +13 -0
  192. topologies-0.0.1/src/topologies/datasets/records/amos.json +13 -0
  193. topologies-0.0.1/src/topologies/datasets/records/amos22.json +13 -0
  194. topologies-0.0.1/src/topologies/datasets/records/anli.json +12 -0
  195. topologies-0.0.1/src/topologies/datasets/records/anuraset.json +13 -0
  196. topologies-0.0.1/src/topologies/datasets/records/apps.json +12 -0
  197. topologies-0.0.1/src/topologies/datasets/records/aptos.json +13 -0
  198. topologies-0.0.1/src/topologies/datasets/records/arc.json +13 -0
  199. topologies-0.0.1/src/topologies/datasets/records/argoverse-2.json +13 -0
  200. topologies-0.0.1/src/topologies/datasets/records/arkit-scenes.json +12 -0
  201. topologies-0.0.1/src/topologies/datasets/records/arxiv-pubmed-sum.json +12 -0
  202. topologies-0.0.1/src/topologies/datasets/records/ashrae-gepiii.json +13 -0
  203. topologies-0.0.1/src/topologies/datasets/records/asvspoof-2019.json +13 -0
  204. topologies-0.0.1/src/topologies/datasets/records/atari-100k.json +13 -0
  205. topologies-0.0.1/src/topologies/datasets/records/audiocaps.json +13 -0
  206. topologies-0.0.1/src/topologies/datasets/records/audioset.json +13 -0
  207. topologies-0.0.1/src/topologies/datasets/records/ava.json +13 -0
  208. topologies-0.0.1/src/topologies/datasets/records/avazu.json +13 -0
  209. topologies-0.0.1/src/topologies/datasets/records/avenue.json +13 -0
  210. topologies-0.0.1/src/topologies/datasets/records/babyai.json +13 -0
  211. topologies-0.0.1/src/topologies/datasets/records/bach.json +13 -0
  212. topologies-0.0.1/src/topologies/datasets/records/bank-marketing.json +13 -0
  213. topologies-0.0.1/src/topologies/datasets/records/bbbc.json +13 -0
  214. topologies-0.0.1/src/topologies/datasets/records/bbh.json +13 -0
  215. topologies-0.0.1/src/topologies/datasets/records/bbq.json +12 -0
  216. topologies-0.0.1/src/topologies/datasets/records/bdd100k.json +12 -0
  217. topologies-0.0.1/src/topologies/datasets/records/beans.json +13 -0
  218. topologies-0.0.1/src/topologies/datasets/records/bigearthnet.json +13 -0
  219. topologies-0.0.1/src/topologies/datasets/records/bijie.json +13 -0
  220. topologies-0.0.1/src/topologies/datasets/records/billion-word.json +12 -0
  221. topologies-0.0.1/src/topologies/datasets/records/bindingdb.json +13 -0
  222. topologies-0.0.1/src/topologies/datasets/records/birdclef.json +13 -0
  223. topologies-0.0.1/src/topologies/datasets/records/birdset.json +13 -0
  224. topologies-0.0.1/src/topologies/datasets/records/ble-aoa.json +13 -0
  225. topologies-0.0.1/src/topologies/datasets/records/blendedmvs.json +12 -0
  226. topologies-0.0.1/src/topologies/datasets/records/bobsl.json +13 -0
  227. topologies-0.0.1/src/topologies/datasets/records/book-crossing.json +12 -0
  228. topologies-0.0.1/src/topologies/datasets/records/brats.json +13 -0
  229. topologies-0.0.1/src/topologies/datasets/records/breakhis.json +13 -0
  230. topologies-0.0.1/src/topologies/datasets/records/btad.json +13 -0
  231. topologies-0.0.1/src/topologies/datasets/records/btcv.json +13 -0
  232. topologies-0.0.1/src/topologies/datasets/records/c4.json +13 -0
  233. topologies-0.0.1/src/topologies/datasets/records/california-housing.json +12 -0
  234. topologies-0.0.1/src/topologies/datasets/records/caltech-101.json +12 -0
  235. topologies-0.0.1/src/topologies/datasets/records/camelyon-16.json +13 -0
  236. topologies-0.0.1/src/topologies/datasets/records/camelyon-17.json +13 -0
  237. topologies-0.0.1/src/topologies/datasets/records/camelyon.json +13 -0
  238. topologies-0.0.1/src/topologies/datasets/records/camus.json +13 -0
  239. topologies-0.0.1/src/topologies/datasets/records/camvid.json +12 -0
  240. topologies-0.0.1/src/topologies/datasets/records/caselaw-hold.json +13 -0
  241. topologies-0.0.1/src/topologies/datasets/records/casia-splicing.json +13 -0
  242. topologies-0.0.1/src/topologies/datasets/records/casia-v2.json +13 -0
  243. topologies-0.0.1/src/topologies/datasets/records/casia-webface.json +12 -0
  244. topologies-0.0.1/src/topologies/datasets/records/cassava-leaf.json +13 -0
  245. topologies-0.0.1/src/topologies/datasets/records/cassava.json +13 -0
  246. topologies-0.0.1/src/topologies/datasets/records/cc3m.json +13 -0
  247. topologies-0.0.1/src/topologies/datasets/records/ccmatrix.json +12 -0
  248. topologies-0.0.1/src/topologies/datasets/records/ccpg.json +13 -0
  249. topologies-0.0.1/src/topologies/datasets/records/celeba-spurious.json +13 -0
  250. topologies-0.0.1/src/topologies/datasets/records/celeba.json +12 -0
  251. topologies-0.0.1/src/topologies/datasets/records/census.json +13 -0
  252. topologies-0.0.1/src/topologies/datasets/records/chaos.json +13 -0
  253. topologies-0.0.1/src/topologies/datasets/records/charades.json +12 -0
  254. topologies-0.0.1/src/topologies/datasets/records/chembl-33.json +13 -0
  255. topologies-0.0.1/src/topologies/datasets/records/chestxray14.json +13 -0
  256. topologies-0.0.1/src/topologies/datasets/records/chexpert.json +11 -0
  257. topologies-0.0.1/src/topologies/datasets/records/chime-4.json +13 -0
  258. topologies-0.0.1/src/topologies/datasets/records/chime-6.json +13 -0
  259. topologies-0.0.1/src/topologies/datasets/records/chime-frb.json +13 -0
  260. topologies-0.0.1/src/topologies/datasets/records/cholec-t50.json +13 -0
  261. topologies-0.0.1/src/topologies/datasets/records/cholec80.json +13 -0
  262. topologies-0.0.1/src/topologies/datasets/records/cholecseg8k.json +13 -0
  263. topologies-0.0.1/src/topologies/datasets/records/cifar-10.1.json +12 -0
  264. topologies-0.0.1/src/topologies/datasets/records/cifar-100.json +13 -0
  265. topologies-0.0.1/src/topologies/datasets/records/cinic-10.json +13 -0
  266. topologies-0.0.1/src/topologies/datasets/records/citeseer.json +13 -0
  267. topologies-0.0.1/src/topologies/datasets/records/cityscapes.json +13 -0
  268. topologies-0.0.1/src/topologies/datasets/records/climsim.json +13 -0
  269. topologies-0.0.1/src/topologies/datasets/records/clotho.json +13 -0
  270. topologies-0.0.1/src/topologies/datasets/records/cmapss.json +13 -0
  271. topologies-0.0.1/src/topologies/datasets/records/cnn-dailymail.json +13 -0
  272. topologies-0.0.1/src/topologies/datasets/records/co3d.json +12 -0
  273. topologies-0.0.1/src/topologies/datasets/records/coauthor-cs.json +12 -0
  274. topologies-0.0.1/src/topologies/datasets/records/coco-2017.json +13 -0
  275. topologies-0.0.1/src/topologies/datasets/records/coco-captions.json +13 -0
  276. topologies-0.0.1/src/topologies/datasets/records/coco-pose.json +13 -0
  277. topologies-0.0.1/src/topologies/datasets/records/codecontests.json +12 -0
  278. topologies-0.0.1/src/topologies/datasets/records/codex.json +13 -0
  279. topologies-0.0.1/src/topologies/datasets/records/common-voice.json +13 -0
  280. topologies-0.0.1/src/topologies/datasets/records/compas.json +13 -0
  281. topologies-0.0.1/src/topologies/datasets/records/compcars.json +12 -0
  282. topologies-0.0.1/src/topologies/datasets/records/conceptual-12m.json +13 -0
  283. topologies-0.0.1/src/topologies/datasets/records/coqa.json +13 -0
  284. topologies-0.0.1/src/topologies/datasets/records/cora.json +13 -0
  285. topologies-0.0.1/src/topologies/datasets/records/cord.json +13 -0
  286. topologies-0.0.1/src/topologies/datasets/records/covertype.json +13 -0
  287. topologies-0.0.1/src/topologies/datasets/records/coyo-700m.json +13 -0
  288. topologies-0.0.1/src/topologies/datasets/records/crafter.json +13 -0
  289. topologies-0.0.1/src/topologies/datasets/records/crema-d.json +12 -0
  290. topologies-0.0.1/src/topologies/datasets/records/cremi.json +13 -0
  291. topologies-0.0.1/src/topologies/datasets/records/criteo.json +13 -0
  292. topologies-0.0.1/src/topologies/datasets/records/crowdhuman.json +12 -0
  293. topologies-0.0.1/src/topologies/datasets/records/crowdpose.json +12 -0
  294. topologies-0.0.1/src/topologies/datasets/records/csl-daily.json +13 -0
  295. topologies-0.0.1/src/topologies/datasets/records/ctw1500.json +13 -0
  296. topologies-0.0.1/src/topologies/datasets/records/cuad.json +13 -0
  297. topologies-0.0.1/src/topologies/datasets/records/cub-200.json +13 -0
  298. topologies-0.0.1/src/topologies/datasets/records/cuhk03.json +12 -0
  299. topologies-0.0.1/src/topologies/datasets/records/cwru.json +13 -0
  300. topologies-0.0.1/src/topologies/datasets/records/d4rl.json +13 -0
  301. topologies-0.0.1/src/topologies/datasets/records/dagm.json +13 -0
  302. topologies-0.0.1/src/topologies/datasets/records/dailydialog.json +13 -0
  303. topologies-0.0.1/src/topologies/datasets/records/dancetrack.json +13 -0
  304. topologies-0.0.1/src/topologies/datasets/records/datacomp-1b.json +13 -0
  305. topologies-0.0.1/src/topologies/datasets/records/davis.json +13 -0
  306. topologies-0.0.1/src/topologies/datasets/records/dcase.json +13 -0
  307. topologies-0.0.1/src/topologies/datasets/records/dclde.json +13 -0
  308. topologies-0.0.1/src/topologies/datasets/records/deeperforensics.json +13 -0
  309. topologies-0.0.1/src/topologies/datasets/records/deepglobe.json +13 -0
  310. topologies-0.0.1/src/topologies/datasets/records/deeppcb.json +13 -0
  311. topologies-0.0.1/src/topologies/datasets/records/didemo.json +13 -0
  312. topologies-0.0.1/src/topologies/datasets/records/diii-d.json +13 -0
  313. topologies-0.0.1/src/topologies/datasets/records/dmc.json +13 -0
  314. topologies-0.0.1/src/topologies/datasets/records/dns-challenge.json +13 -0
  315. topologies-0.0.1/src/topologies/datasets/records/docvqa.json +13 -0
  316. topologies-0.0.1/src/topologies/datasets/records/dolly.json +13 -0
  317. topologies-0.0.1/src/topologies/datasets/records/dolma.json +13 -0
  318. topologies-0.0.1/src/topologies/datasets/records/domainnet.json +13 -0
  319. topologies-0.0.1/src/topologies/datasets/records/dota.json +13 -0
  320. topologies-0.0.1/src/topologies/datasets/records/drop.json +12 -0
  321. topologies-0.0.1/src/topologies/datasets/records/ds-1000.json +12 -0
  322. topologies-0.0.1/src/topologies/datasets/records/dsb-2018.json +13 -0
  323. topologies-0.0.1/src/topologies/datasets/records/dsec-flow.json +13 -0
  324. topologies-0.0.1/src/topologies/datasets/records/dtd.json +13 -0
  325. topologies-0.0.1/src/topologies/datasets/records/duacs-aviso.json +13 -0
  326. topologies-0.0.1/src/topologies/datasets/records/duacs.json +13 -0
  327. topologies-0.0.1/src/topologies/datasets/records/dud-e.json +13 -0
  328. topologies-0.0.1/src/topologies/datasets/records/dude.json +13 -0
  329. topologies-0.0.1/src/topologies/datasets/records/dvs-cifar10.json +13 -0
  330. topologies-0.0.1/src/topologies/datasets/records/dvs-gesture.json +13 -0
  331. topologies-0.0.1/src/topologies/datasets/records/dycheck.json +13 -0
  332. topologies-0.0.1/src/topologies/datasets/records/earnings-22.json +12 -0
  333. topologies-0.0.1/src/topologies/datasets/records/ego4d.json +12 -0
  334. topologies-0.0.1/src/topologies/datasets/records/eicu-crd.json +13 -0
  335. topologies-0.0.1/src/topologies/datasets/records/electricity.json +13 -0
  336. topologies-0.0.1/src/topologies/datasets/records/elliptic.json +12 -0
  337. topologies-0.0.1/src/topologies/datasets/records/elpv.json +13 -0
  338. topologies-0.0.1/src/topologies/datasets/records/emnist.json +13 -0
  339. topologies-0.0.1/src/topologies/datasets/records/endovis.json +13 -0
  340. topologies-0.0.1/src/topologies/datasets/records/enose.json +13 -0
  341. topologies-0.0.1/src/topologies/datasets/records/epic-kitchens-100.json +12 -0
  342. topologies-0.0.1/src/topologies/datasets/records/epinions.json +13 -0
  343. topologies-0.0.1/src/topologies/datasets/records/eqa.json +13 -0
  344. topologies-0.0.1/src/topologies/datasets/records/era5.json +13 -0
  345. topologies-0.0.1/src/topologies/datasets/records/esc-50.json +13 -0
  346. topologies-0.0.1/src/topologies/datasets/records/esd.json +12 -0
  347. topologies-0.0.1/src/topologies/datasets/records/eth3d.json +13 -0
  348. topologies-0.0.1/src/topologies/datasets/records/ett.json +13 -0
  349. topologies-0.0.1/src/topologies/datasets/records/europarl.json +12 -0
  350. topologies-0.0.1/src/topologies/datasets/records/eurosat.json +13 -0
  351. topologies-0.0.1/src/topologies/datasets/records/exchange.json +13 -0
  352. topologies-0.0.1/src/topologies/datasets/records/extrasensory.json +13 -0
  353. topologies-0.0.1/src/topologies/datasets/records/fair1m.json +13 -0
  354. topologies-0.0.1/src/topologies/datasets/records/fashioniq.json +13 -0
  355. topologies-0.0.1/src/topologies/datasets/records/fedscale.json +13 -0
  356. topologies-0.0.1/src/topologies/datasets/records/femto-pronostia.json +13 -0
  357. topologies-0.0.1/src/topologies/datasets/records/fever.json +12 -0
  358. topologies-0.0.1/src/topologies/datasets/records/ffhq.json +12 -0
  359. topologies-0.0.1/src/topologies/datasets/records/fgvc-aircraft.json +13 -0
  360. topologies-0.0.1/src/topologies/datasets/records/fineweb-edu.json +13 -0
  361. topologies-0.0.1/src/topologies/datasets/records/fineweb.json +13 -0
  362. topologies-0.0.1/src/topologies/datasets/records/fintabnet.json +13 -0
  363. topologies-0.0.1/src/topologies/datasets/records/fiqa-2018.json +13 -0
  364. topologies-0.0.1/src/topologies/datasets/records/flan.json +12 -0
  365. topologies-0.0.1/src/topologies/datasets/records/flchain.json +13 -0
  366. topologies-0.0.1/src/topologies/datasets/records/fleurs.json +12 -0
  367. topologies-0.0.1/src/topologies/datasets/records/flickr30k-entities.json +12 -0
  368. topologies-0.0.1/src/topologies/datasets/records/flickr30k.json +13 -0
  369. topologies-0.0.1/src/topologies/datasets/records/floodnet.json +13 -0
  370. topologies-0.0.1/src/topologies/datasets/records/flores-200.json +13 -0
  371. topologies-0.0.1/src/topologies/datasets/records/flyingchairs.json +13 -0
  372. topologies-0.0.1/src/topologies/datasets/records/fma.json +13 -0
  373. topologies-0.0.1/src/topologies/datasets/records/fmd.json +12 -0
  374. topologies-0.0.1/src/topologies/datasets/records/fmow.json +13 -0
  375. topologies-0.0.1/src/topologies/datasets/records/food-101.json +13 -0
  376. topologies-0.0.1/src/topologies/datasets/records/forgerynet.json +13 -0
  377. topologies-0.0.1/src/topologies/datasets/records/foursquare.json +13 -0
  378. topologies-0.0.1/src/topologies/datasets/records/franka-kitchen.json +13 -0
  379. topologies-0.0.1/src/topologies/datasets/records/fraud-ieee.json +13 -0
  380. topologies-0.0.1/src/topologies/datasets/records/freihand.json +12 -0
  381. topologies-0.0.1/src/topologies/datasets/records/fsd50k.json +12 -0
  382. topologies-0.0.1/src/topologies/datasets/records/fsdnoisy18k.json +12 -0
  383. topologies-0.0.1/src/topologies/datasets/records/funsd.json +13 -0
  384. topologies-0.0.1/src/topologies/datasets/records/gages-ii.json +13 -0
  385. topologies-0.0.1/src/topologies/datasets/records/gait3d.json +13 -0
  386. topologies-0.0.1/src/topologies/datasets/records/gbsg2.json +13 -0
  387. topologies-0.0.1/src/topologies/datasets/records/gefcom2014.json +13 -0
  388. topologies-0.0.1/src/topologies/datasets/records/gelsight-wedge.json +13 -0
  389. topologies-0.0.1/src/topologies/datasets/records/german-credit.json +13 -0
  390. topologies-0.0.1/src/topologies/datasets/records/gigaspeech.json +13 -0
  391. topologies-0.0.1/src/topologies/datasets/records/gigaword.json +12 -0
  392. topologies-0.0.1/src/topologies/datasets/records/glue.json +13 -0
  393. topologies-0.0.1/src/topologies/datasets/records/goodreads.json +13 -0
  394. topologies-0.0.1/src/topologies/datasets/records/got-10k.json +13 -0
  395. topologies-0.0.1/src/topologies/datasets/records/gowalla.json +13 -0
  396. topologies-0.0.1/src/topologies/datasets/records/gqa.json +13 -0
  397. topologies-0.0.1/src/topologies/datasets/records/grew.json +13 -0
  398. topologies-0.0.1/src/topologies/datasets/records/grf.json +13 -0
  399. topologies-0.0.1/src/topologies/datasets/records/gsm8k.json +13 -0
  400. topologies-0.0.1/src/topologies/datasets/records/gtzan.json +13 -0
  401. topologies-0.0.1/src/topologies/datasets/records/gwosc.json +13 -0
  402. topologies-0.0.1/src/topologies/datasets/records/hacs.json +12 -0
  403. topologies-0.0.1/src/topologies/datasets/records/hellaswag.json +13 -0
  404. topologies-0.0.1/src/topologies/datasets/records/hepmass.json +13 -0
  405. topologies-0.0.1/src/topologies/datasets/records/hgb.json +12 -0
  406. topologies-0.0.1/src/topologies/datasets/records/higgs.json +13 -0
  407. topologies-0.0.1/src/topologies/datasets/records/hirid.json +13 -0
  408. topologies-0.0.1/src/topologies/datasets/records/hm3d.json +12 -0
  409. topologies-0.0.1/src/topologies/datasets/records/hmdb51.json +13 -0
  410. topologies-0.0.1/src/topologies/datasets/records/home-credit-default-risk.json +13 -0
  411. topologies-0.0.1/src/topologies/datasets/records/hotpotqa.json +12 -0
  412. topologies-0.0.1/src/topologies/datasets/records/house-prices.json +13 -0
  413. topologies-0.0.1/src/topologies/datasets/records/howto100m.json +13 -0
  414. topologies-0.0.1/src/topologies/datasets/records/htru2.json +13 -0
  415. topologies-0.0.1/src/topologies/datasets/records/human36m.json +13 -0
  416. topologies-0.0.1/src/topologies/datasets/records/humaneval.json +13 -0
  417. topologies-0.0.1/src/topologies/datasets/records/iam.json +13 -0
  418. topologies-0.0.1/src/topologies/datasets/records/icdar15.json +13 -0
  419. topologies-0.0.1/src/topologies/datasets/records/icdar2017-mlt.json +13 -0
  420. topologies-0.0.1/src/topologies/datasets/records/idrid.json +13 -0
  421. topologies-0.0.1/src/topologies/datasets/records/iemocap.json +12 -0
  422. topologies-0.0.1/src/topologies/datasets/records/ifeval.json +12 -0
  423. topologies-0.0.1/src/topologies/datasets/records/ili.json +13 -0
  424. topologies-0.0.1/src/topologies/datasets/records/imagenet-1k.json +13 -0
  425. topologies-0.0.1/src/topologies/datasets/records/imagenet-21k.json +13 -0
  426. topologies-0.0.1/src/topologies/datasets/records/imagenet-9.json +12 -0
  427. topologies-0.0.1/src/topologies/datasets/records/imagenet-o.json +13 -0
  428. topologies-0.0.1/src/topologies/datasets/records/imagenet-p.json +13 -0
  429. topologies-0.0.1/src/topologies/datasets/records/imagenet-real.json +12 -0
  430. topologies-0.0.1/src/topologies/datasets/records/imagenet-v2.json +13 -0
  431. topologies-0.0.1/src/topologies/datasets/records/imdb.json +13 -0
  432. topologies-0.0.1/src/topologies/datasets/records/inaturalist-2017.json +13 -0
  433. topologies-0.0.1/src/topologies/datasets/records/inaturalist-2021.json +13 -0
  434. topologies-0.0.1/src/topologies/datasets/records/inria-aerial.json +13 -0
  435. topologies-0.0.1/src/topologies/datasets/records/ipinyou.json +13 -0
  436. topologies-0.0.1/src/topologies/datasets/records/isic.json +13 -0
  437. topologies-0.0.1/src/topologies/datasets/records/isles-2022.json +13 -0
  438. topologies-0.0.1/src/topologies/datasets/records/isprs-potsdam.json +13 -0
  439. topologies-0.0.1/src/topologies/datasets/records/iwslt.json +13 -0
  440. topologies-0.0.1/src/topologies/datasets/records/jarvis-dft.json +13 -0
  441. topologies-0.0.1/src/topologies/datasets/records/jester.json +12 -0
  442. topologies-0.0.1/src/topologies/datasets/records/jigsaw-toxic.json +12 -0
  443. topologies-0.0.1/src/topologies/datasets/records/jigsaws.json +13 -0
  444. topologies-0.0.1/src/topologies/datasets/records/jvs.json +12 -0
  445. topologies-0.0.1/src/topologies/datasets/records/kermany-oct.json +13 -0
  446. topologies-0.0.1/src/topologies/datasets/records/kilt.json +12 -0
  447. topologies-0.0.1/src/topologies/datasets/records/kinetics-400.json +13 -0
  448. topologies-0.0.1/src/topologies/datasets/records/kinetics-600.json +13 -0
  449. topologies-0.0.1/src/topologies/datasets/records/kinetics-700.json +13 -0
  450. topologies-0.0.1/src/topologies/datasets/records/kits.json +13 -0
  451. topologies-0.0.1/src/topologies/datasets/records/kitti.json +13 -0
  452. topologies-0.0.1/src/topologies/datasets/records/kkbox-wsdm-cup.json +13 -0
  453. topologies-0.0.1/src/topologies/datasets/records/kmnist.json +12 -0
  454. topologies-0.0.1/src/topologies/datasets/records/ksponspeech.json +12 -0
  455. topologies-0.0.1/src/topologies/datasets/records/kuairec.json +13 -0
  456. topologies-0.0.1/src/topologies/datasets/records/kuaisar.json +13 -0
  457. topologies-0.0.1/src/topologies/datasets/records/kvasir-seg.json +13 -0
  458. topologies-0.0.1/src/topologies/datasets/records/laion-2b.json +13 -0
  459. topologies-0.0.1/src/topologies/datasets/records/laion-400m.json +13 -0
  460. topologies-0.0.1/src/topologies/datasets/records/lakh-midi.json +12 -0
  461. topologies-0.0.1/src/topologies/datasets/records/landslide4sense.json +13 -0
  462. topologies-0.0.1/src/topologies/datasets/records/largest.json +13 -0
  463. topologies-0.0.1/src/topologies/datasets/records/lasot.json +13 -0
  464. topologies-0.0.1/src/topologies/datasets/records/lastfm-1k.json +13 -0
  465. topologies-0.0.1/src/topologies/datasets/records/leaf.json +13 -0
  466. topologies-0.0.1/src/topologies/datasets/records/levir-cd.json +13 -0
  467. topologies-0.0.1/src/topologies/datasets/records/lfm-360k.json +13 -0
  468. topologies-0.0.1/src/topologies/datasets/records/lfw.json +12 -0
  469. topologies-0.0.1/src/topologies/datasets/records/libero.json +13 -0
  470. topologies-0.0.1/src/topologies/datasets/records/libri-light.json +13 -0
  471. topologies-0.0.1/src/topologies/datasets/records/librimix.json +13 -0
  472. topologies-0.0.1/src/topologies/datasets/records/librispeech.json +13 -0
  473. topologies-0.0.1/src/topologies/datasets/records/libritts.json +13 -0
  474. topologies-0.0.1/src/topologies/datasets/records/lidc-idri.json +13 -0
  475. topologies-0.0.1/src/topologies/datasets/records/lits.json +13 -0
  476. topologies-0.0.1/src/topologies/datasets/records/livecodebench.json +12 -0
  477. topologies-0.0.1/src/topologies/datasets/records/ljspeech.json +13 -0
  478. topologies-0.0.1/src/topologies/datasets/records/llava-instruct-150k.json +13 -0
  479. topologies-0.0.1/src/topologies/datasets/records/longbench.json +12 -0
  480. topologies-0.0.1/src/topologies/datasets/records/lsst-alert.json +13 -0
  481. topologies-0.0.1/src/topologies/datasets/records/luna16.json +13 -0
  482. topologies-0.0.1/src/topologies/datasets/records/lvis.json +13 -0
  483. topologies-0.0.1/src/topologies/datasets/records/lvos.json +12 -0
  484. topologies-0.0.1/src/topologies/datasets/records/m4.json +13 -0
  485. topologies-0.0.1/src/topologies/datasets/records/m5.json +13 -0
  486. topologies-0.0.1/src/topologies/datasets/records/maestro.json +13 -0
  487. topologies-0.0.1/src/topologies/datasets/records/magnatagatune.json +13 -0
  488. topologies-0.0.1/src/topologies/datasets/records/mapillary-vistas.json +13 -0
  489. topologies-0.0.1/src/topologies/datasets/records/market-1501.json +12 -0
  490. topologies-0.0.1/src/topologies/datasets/records/matbench.json +13 -0
  491. topologies-0.0.1/src/topologies/datasets/records/materials-project.json +13 -0
  492. topologies-0.0.1/src/topologies/datasets/records/math.json +13 -0
  493. topologies-0.0.1/src/topologies/datasets/records/mathqa.json +12 -0
  494. topologies-0.0.1/src/topologies/datasets/records/matterport3d.json +12 -0
  495. topologies-0.0.1/src/topologies/datasets/records/mbpp.json +13 -0
  496. topologies-0.0.1/src/topologies/datasets/records/md17.json +13 -0
  497. topologies-0.0.1/src/topologies/datasets/records/medical-segmentation-decathlon.json +13 -0
  498. topologies-0.0.1/src/topologies/datasets/records/medleydb.json +12 -0
  499. topologies-0.0.1/src/topologies/datasets/records/metaworld.json +13 -0
  500. topologies-0.0.1/src/topologies/datasets/records/metr-la.json +12 -0
  501. topologies-0.0.1/src/topologies/datasets/records/mhealth.json +13 -0
  502. topologies-0.0.1/src/topologies/datasets/records/middlebury.json +13 -0
  503. topologies-0.0.1/src/topologies/datasets/records/million-aid.json +13 -0
  504. topologies-0.0.1/src/topologies/datasets/records/mimic-cxr.json +13 -0
  505. topologies-0.0.1/src/topologies/datasets/records/mimic-iii.json +13 -0
  506. topologies-0.0.1/src/topologies/datasets/records/mind.json +13 -0
  507. topologies-0.0.1/src/topologies/datasets/records/minerl-obtaindiamond.json +13 -0
  508. topologies-0.0.1/src/topologies/datasets/records/minigrid.json +13 -0
  509. topologies-0.0.1/src/topologies/datasets/records/mit-bih.json +13 -0
  510. topologies-0.0.1/src/topologies/datasets/records/mit-indoor67.json +12 -0
  511. topologies-0.0.1/src/topologies/datasets/records/mjsynth.json +13 -0
  512. topologies-0.0.1/src/topologies/datasets/records/mls.json +12 -0
  513. topologies-0.0.1/src/topologies/datasets/records/mmlu-pro.json +13 -0
  514. topologies-0.0.1/src/topologies/datasets/records/mmlu.json +13 -0
  515. topologies-0.0.1/src/topologies/datasets/records/mmwave-har.json +13 -0
  516. topologies-0.0.1/src/topologies/datasets/records/modelnet40.json +13 -0
  517. topologies-0.0.1/src/topologies/datasets/records/moisesdb.json +13 -0
  518. topologies-0.0.1/src/topologies/datasets/records/moleculenet-8.json +13 -0
  519. topologies-0.0.1/src/topologies/datasets/records/moments-in-time.json +12 -0
  520. topologies-0.0.1/src/topologies/datasets/records/monash-30.json +13 -0
  521. topologies-0.0.1/src/topologies/datasets/records/monuseg.json +13 -0
  522. topologies-0.0.1/src/topologies/datasets/records/mose.json +12 -0
  523. topologies-0.0.1/src/topologies/datasets/records/mot17.json +13 -0
  524. topologies-0.0.1/src/topologies/datasets/records/mot20.json +13 -0
  525. topologies-0.0.1/src/topologies/datasets/records/movielens-1m.json +13 -0
  526. topologies-0.0.1/src/topologies/datasets/records/movielens-20m.json +13 -0
  527. topologies-0.0.1/src/topologies/datasets/records/movielens-25m.json +12 -0
  528. topologies-0.0.1/src/topologies/datasets/records/mpdd.json +13 -0
  529. topologies-0.0.1/src/topologies/datasets/records/mpi-inf-3dhp.json +12 -0
  530. topologies-0.0.1/src/topologies/datasets/records/mpii.json +13 -0
  531. topologies-0.0.1/src/topologies/datasets/records/ms-asl.json +13 -0
  532. topologies-0.0.1/src/topologies/datasets/records/ms-marco.json +13 -0
  533. topologies-0.0.1/src/topologies/datasets/records/msd.json +13 -0
  534. topologies-0.0.1/src/topologies/datasets/records/msmt17.json +12 -0
  535. topologies-0.0.1/src/topologies/datasets/records/msr-vtt.json +13 -0
  536. topologies-0.0.1/src/topologies/datasets/records/mstar.json +13 -0
  537. topologies-0.0.1/src/topologies/datasets/records/msvd.json +13 -0
  538. topologies-0.0.1/src/topologies/datasets/records/mtbs.json +13 -0
  539. topologies-0.0.1/src/topologies/datasets/records/mteb.json +12 -0
  540. topologies-0.0.1/src/topologies/datasets/records/multi-news.json +12 -0
  541. topologies-0.0.1/src/topologies/datasets/records/mura.json +13 -0
  542. topologies-0.0.1/src/topologies/datasets/records/musdb18-hq.json +13 -0
  543. topologies-0.0.1/src/topologies/datasets/records/musiccaps.json +12 -0
  544. topologies-0.0.1/src/topologies/datasets/records/musicnet.json +12 -0
  545. topologies-0.0.1/src/topologies/datasets/records/mutag.json +13 -0
  546. topologies-0.0.1/src/topologies/datasets/records/mvimgnet.json +12 -0
  547. topologies-0.0.1/src/topologies/datasets/records/mvsec.json +13 -0
  548. topologies-0.0.1/src/topologies/datasets/records/mvtec-ad.json +13 -0
  549. topologies-0.0.1/src/topologies/datasets/records/mvtec-loco.json +13 -0
  550. topologies-0.0.1/src/topologies/datasets/records/mvtec.json +13 -0
  551. topologies-0.0.1/src/topologies/datasets/records/n-caltech101.json +13 -0
  552. topologies-0.0.1/src/topologies/datasets/records/n-cmapss.json +13 -0
  553. topologies-0.0.1/src/topologies/datasets/records/n2c2.json +13 -0
  554. topologies-0.0.1/src/topologies/datasets/records/nab.json +13 -0
  555. topologies-0.0.1/src/topologies/datasets/records/nabirds.json +12 -0
  556. topologies-0.0.1/src/topologies/datasets/records/nas-bench-301.json +13 -0
  557. topologies-0.0.1/src/topologies/datasets/records/natural-questions.json +13 -0
  558. topologies-0.0.1/src/topologies/datasets/records/nct-crc-he.json +13 -0
  559. topologies-0.0.1/src/topologies/datasets/records/nell-995.json +13 -0
  560. topologies-0.0.1/src/topologies/datasets/records/nerf-synthetic.json +12 -0
  561. topologies-0.0.1/src/topologies/datasets/records/netflix-prize.json +12 -0
  562. topologies-0.0.1/src/topologies/datasets/records/nethack.json +13 -0
  563. topologies-0.0.1/src/topologies/datasets/records/neu-det.json +13 -0
  564. topologies-0.0.1/src/topologies/datasets/records/nico.json +13 -0
  565. topologies-0.0.1/src/topologies/datasets/records/nih-14.json +13 -0
  566. topologies-0.0.1/src/topologies/datasets/records/nllb-seed.json +13 -0
  567. topologies-0.0.1/src/topologies/datasets/records/nmnist.json +13 -0
  568. topologies-0.0.1/src/topologies/datasets/records/noaa-oisst.json +13 -0
  569. topologies-0.0.1/src/topologies/datasets/records/nsl-kdd.json +13 -0
  570. topologies-0.0.1/src/topologies/datasets/records/nsynth.json +13 -0
  571. topologies-0.0.1/src/topologies/datasets/records/ntu-rgbd.json +12 -0
  572. topologies-0.0.1/src/topologies/datasets/records/nuscenes.json +13 -0
  573. topologies-0.0.1/src/topologies/datasets/records/nwpu-resisc45.json +13 -0
  574. topologies-0.0.1/src/topologies/datasets/records/nyc-taxi.json +13 -0
  575. topologies-0.0.1/src/topologies/datasets/records/nyu-depth-v2.json +12 -0
  576. topologies-0.0.1/src/topologies/datasets/records/objaverse.json +12 -0
  577. topologies-0.0.1/src/topologies/datasets/records/objectnet.json +12 -0
  578. topologies-0.0.1/src/topologies/datasets/records/objects365.json +13 -0
  579. topologies-0.0.1/src/topologies/datasets/records/oc20.json +13 -0
  580. topologies-0.0.1/src/topologies/datasets/records/oc22.json +13 -0
  581. topologies-0.0.1/src/topologies/datasets/records/octa-500.json +13 -0
  582. topologies-0.0.1/src/topologies/datasets/records/office-home.json +12 -0
  583. topologies-0.0.1/src/topologies/datasets/records/ogbg-molhiv.json +13 -0
  584. topologies-0.0.1/src/topologies/datasets/records/ogbg-pcba.json +13 -0
  585. topologies-0.0.1/src/topologies/datasets/records/ogbg-ppa.json +12 -0
  586. topologies-0.0.1/src/topologies/datasets/records/ogbl-collab.json +13 -0
  587. topologies-0.0.1/src/topologies/datasets/records/ogbl-ddi.json +13 -0
  588. topologies-0.0.1/src/topologies/datasets/records/ogbl-ppa.json +13 -0
  589. topologies-0.0.1/src/topologies/datasets/records/ogbn-arxiv.json +13 -0
  590. topologies-0.0.1/src/topologies/datasets/records/ogbn-mag.json +13 -0
  591. topologies-0.0.1/src/topologies/datasets/records/ogbn-papers100m.json +13 -0
  592. topologies-0.0.1/src/topologies/datasets/records/ogbn-products.json +13 -0
  593. topologies-0.0.1/src/topologies/datasets/records/ogbn-proteins.json +13 -0
  594. topologies-0.0.1/src/topologies/datasets/records/ok-vqa.json +13 -0
  595. topologies-0.0.1/src/topologies/datasets/records/once.json +12 -0
  596. topologies-0.0.1/src/topologies/datasets/records/open-bandit.json +13 -0
  597. topologies-0.0.1/src/topologies/datasets/records/openhermes.json +13 -0
  598. topologies-0.0.1/src/topologies/datasets/records/openimages-v7.json +13 -0
  599. topologies-0.0.1/src/topologies/datasets/records/openmic-2018.json +13 -0
  600. topologies-0.0.1/src/topologies/datasets/records/openml-cc18.json +13 -0
  601. topologies-0.0.1/src/topologies/datasets/records/openwebtext.json +12 -0
  602. topologies-0.0.1/src/topologies/datasets/records/opportunity.json +13 -0
  603. topologies-0.0.1/src/topologies/datasets/records/opus-100.json +13 -0
  604. topologies-0.0.1/src/topologies/datasets/records/oqmd.json +13 -0
  605. topologies-0.0.1/src/topologies/datasets/records/orcadata.json +13 -0
  606. topologies-0.0.1/src/topologies/datasets/records/otb100.json +13 -0
  607. topologies-0.0.1/src/topologies/datasets/records/otto.json +13 -0
  608. topologies-0.0.1/src/topologies/datasets/records/outbrain-click-prediction.json +13 -0
  609. topologies-0.0.1/src/topologies/datasets/records/oxford-flowers-102.json +13 -0
  610. topologies-0.0.1/src/topologies/datasets/records/pacs.json +12 -0
  611. topologies-0.0.1/src/topologies/datasets/records/padchest.json +13 -0
  612. topologies-0.0.1/src/topologies/datasets/records/pamap2.json +13 -0
  613. topologies-0.0.1/src/topologies/datasets/records/pannuke.json +13 -0
  614. topologies-0.0.1/src/topologies/datasets/records/paracrawl.json +12 -0
  615. topologies-0.0.1/src/topologies/datasets/records/pascal-context.json +12 -0
  616. topologies-0.0.1/src/topologies/datasets/records/pascal-voc-07.json +13 -0
  617. topologies-0.0.1/src/topologies/datasets/records/pascal-voc-12.json +13 -0
  618. topologies-0.0.1/src/topologies/datasets/records/pastis.json +13 -0
  619. topologies-0.0.1/src/topologies/datasets/records/pcam.json +13 -0
  620. topologies-0.0.1/src/topologies/datasets/records/pcb-defect.json +13 -0
  621. topologies-0.0.1/src/topologies/datasets/records/pcqm4mv2.json +13 -0
  622. topologies-0.0.1/src/topologies/datasets/records/pdbbind.json +13 -0
  623. topologies-0.0.1/src/topologies/datasets/records/pdebench.json +13 -0
  624. topologies-0.0.1/src/topologies/datasets/records/pems-03.json +13 -0
  625. topologies-0.0.1/src/topologies/datasets/records/pems-04.json +13 -0
  626. topologies-0.0.1/src/topologies/datasets/records/pems-07.json +13 -0
  627. topologies-0.0.1/src/topologies/datasets/records/pems-08.json +13 -0
  628. topologies-0.0.1/src/topologies/datasets/records/pems-bay.json +13 -0
  629. topologies-0.0.1/src/topologies/datasets/records/penn-treebank.json +12 -0
  630. topologies-0.0.1/src/topologies/datasets/records/peoples-speech.json +12 -0
  631. topologies-0.0.1/src/topologies/datasets/records/phakir.json +13 -0
  632. topologies-0.0.1/src/topologies/datasets/records/phm-milling.json +13 -0
  633. topologies-0.0.1/src/topologies/datasets/records/physionet-2012.json +13 -0
  634. topologies-0.0.1/src/topologies/datasets/records/places365.json +13 -0
  635. topologies-0.0.1/src/topologies/datasets/records/plant-pathology-2020.json +13 -0
  636. topologies-0.0.1/src/topologies/datasets/records/plantdoc.json +13 -0
  637. topologies-0.0.1/src/topologies/datasets/records/plantvillage.json +13 -0
  638. topologies-0.0.1/src/topologies/datasets/records/porto-seguro.json +13 -0
  639. topologies-0.0.1/src/topologies/datasets/records/posetrack21.json +12 -0
  640. topologies-0.0.1/src/topologies/datasets/records/potsdam.json +13 -0
  641. topologies-0.0.1/src/topologies/datasets/records/powergridworld.json +13 -0
  642. topologies-0.0.1/src/topologies/datasets/records/ppi.json +12 -0
  643. topologies-0.0.1/src/topologies/datasets/records/procgen.json +13 -0
  644. topologies-0.0.1/src/topologies/datasets/records/prophesee-gen1.json +13 -0
  645. topologies-0.0.1/src/topologies/datasets/records/proteinnet.json +13 -0
  646. topologies-0.0.1/src/topologies/datasets/records/proteins.json +13 -0
  647. topologies-0.0.1/src/topologies/datasets/records/publaynet.json +13 -0
  648. topologies-0.0.1/src/topologies/datasets/records/pubmed.json +13 -0
  649. topologies-0.0.1/src/topologies/datasets/records/pubmedqa.json +12 -0
  650. topologies-0.0.1/src/topologies/datasets/records/pubtabnet.json +13 -0
  651. topologies-0.0.1/src/topologies/datasets/records/pv-el.json +13 -0
  652. topologies-0.0.1/src/topologies/datasets/records/qm7b.json +13 -0
  653. topologies-0.0.1/src/topologies/datasets/records/qm9.json +13 -0
  654. topologies-0.0.1/src/topologies/datasets/records/qmnist.json +12 -0
  655. topologies-0.0.1/src/topologies/datasets/records/race.json +12 -0
  656. topologies-0.0.1/src/topologies/datasets/records/radar-doppler.json +13 -0
  657. topologies-0.0.1/src/topologies/datasets/records/real-toxicity-prompts.json +12 -0
  658. topologies-0.0.1/src/topologies/datasets/records/reddit.json +13 -0
  659. topologies-0.0.1/src/topologies/datasets/records/redpajama-v2.json +13 -0
  660. topologies-0.0.1/src/topologies/datasets/records/ref-youtube-vos.json +12 -0
  661. topologies-0.0.1/src/topologies/datasets/records/refcoco-plus.json +13 -0
  662. topologies-0.0.1/src/topologies/datasets/records/refcoco.json +13 -0
  663. topologies-0.0.1/src/topologies/datasets/records/refcocog.json +13 -0
  664. topologies-0.0.1/src/topologies/datasets/records/replica.json +12 -0
  665. topologies-0.0.1/src/topologies/datasets/records/retailrocket-ctr.json +13 -0
  666. topologies-0.0.1/src/topologies/datasets/records/retailrocket.json +13 -0
  667. topologies-0.0.1/src/topologies/datasets/records/retinal-vessels.json +13 -0
  668. topologies-0.0.1/src/topologies/datasets/records/rl-unplugged.json +13 -0
  669. topologies-0.0.1/src/topologies/datasets/records/rlbench.json +13 -0
  670. topologies-0.0.1/src/topologies/datasets/records/rmd17.json +13 -0
  671. topologies-0.0.1/src/topologies/datasets/records/robomimic.json +13 -0
  672. topologies-0.0.1/src/topologies/datasets/records/rossmann-store-sales.json +13 -0
  673. topologies-0.0.1/src/topologies/datasets/records/rsna-pneumonia.json +13 -0
  674. topologies-0.0.1/src/topologies/datasets/records/rvl-cdip.json +13 -0
  675. topologies-0.0.1/src/topologies/datasets/records/s3dis.json +13 -0
  676. topologies-0.0.1/src/topologies/datasets/records/samsum.json +12 -0
  677. topologies-0.0.1/src/topologies/datasets/records/santander.json +13 -0
  678. topologies-0.0.1/src/topologies/datasets/records/sbu-captions.json +13 -0
  679. topologies-0.0.1/src/topologies/datasets/records/scannet.json +13 -0
  680. topologies-0.0.1/src/topologies/datasets/records/scedc.json +13 -0
  681. topologies-0.0.1/src/topologies/datasets/records/sdss.json +13 -0
  682. topologies-0.0.1/src/topologies/datasets/records/sen12ms.json +13 -0
  683. topologies-0.0.1/src/topologies/datasets/records/shanghaitech.json +13 -0
  684. topologies-0.0.1/src/topologies/datasets/records/shapenet-part.json +13 -0
  685. topologies-0.0.1/src/topologies/datasets/records/shipsear.json +13 -0
  686. topologies-0.0.1/src/topologies/datasets/records/shl.json +13 -0
  687. topologies-0.0.1/src/topologies/datasets/records/siim-acr-pneumothorax.json +13 -0
  688. topologies-0.0.1/src/topologies/datasets/records/sintel.json +13 -0
  689. topologies-0.0.1/src/topologies/datasets/records/sketchy.json +13 -0
  690. topologies-0.0.1/src/topologies/datasets/records/slakh2100.json +12 -0
  691. topologies-0.0.1/src/topologies/datasets/records/sleep-edf.json +13 -0
  692. topologies-0.0.1/src/topologies/datasets/records/slimpajama.json +12 -0
  693. topologies-0.0.1/src/topologies/datasets/records/smac.json +13 -0
  694. topologies-0.0.1/src/topologies/datasets/records/smacv2.json +13 -0
  695. topologies-0.0.1/src/topologies/datasets/records/smap-msl.json +13 -0
  696. topologies-0.0.1/src/topologies/datasets/records/smd.json +13 -0
  697. topologies-0.0.1/src/topologies/datasets/records/sms-wsj.json +12 -0
  698. topologies-0.0.1/src/topologies/datasets/records/snli.json +12 -0
  699. topologies-0.0.1/src/topologies/datasets/records/solar.json +13 -0
  700. topologies-0.0.1/src/topologies/datasets/records/something-something-v2.json +13 -0
  701. topologies-0.0.1/src/topologies/datasets/records/sonyc-ust.json +12 -0
  702. topologies-0.0.1/src/topologies/datasets/records/spacenet.json +13 -0
  703. topologies-0.0.1/src/topologies/datasets/records/spgispeech.json +13 -0
  704. topologies-0.0.1/src/topologies/datasets/records/spotify-mpd.json +13 -0
  705. topologies-0.0.1/src/topologies/datasets/records/spring.json +13 -0
  706. topologies-0.0.1/src/topologies/datasets/records/squad-2.json +13 -0
  707. topologies-0.0.1/src/topologies/datasets/records/squad.json +13 -0
  708. topologies-0.0.1/src/topologies/datasets/records/sroie.json +13 -0
  709. topologies-0.0.1/src/topologies/datasets/records/ssdd.json +13 -0
  710. topologies-0.0.1/src/topologies/datasets/records/stanford-cars.json +13 -0
  711. topologies-0.0.1/src/topologies/datasets/records/stanford-dogs.json +12 -0
  712. topologies-0.0.1/src/topologies/datasets/records/steam.json +13 -0
  713. topologies-0.0.1/src/topologies/datasets/records/stl-10.json +13 -0
  714. topologies-0.0.1/src/topologies/datasets/records/sugarbeets-2016.json +13 -0
  715. topologies-0.0.1/src/topologies/datasets/records/sun-rgbd.json +12 -0
  716. topologies-0.0.1/src/topologies/datasets/records/sun397.json +12 -0
  717. topologies-0.0.1/src/topologies/datasets/records/superglue.json +13 -0
  718. topologies-0.0.1/src/topologies/datasets/records/sustainbench.json +13 -0
  719. topologies-0.0.1/src/topologies/datasets/records/susy.json +13 -0
  720. topologies-0.0.1/src/topologies/datasets/records/svamp.json +12 -0
  721. topologies-0.0.1/src/topologies/datasets/records/svhn.json +13 -0
  722. topologies-0.0.1/src/topologies/datasets/records/swat-wadi.json +13 -0
  723. topologies-0.0.1/src/topologies/datasets/records/switchboard.json +12 -0
  724. topologies-0.0.1/src/topologies/datasets/records/synthia.json +13 -0
  725. topologies-0.0.1/src/topologies/datasets/records/synthtext.json +13 -0
  726. topologies-0.0.1/src/topologies/datasets/records/tabarena.json +12 -0
  727. topologies-0.0.1/src/topologies/datasets/records/tabula-sapiens.json +13 -0
  728. topologies-0.0.1/src/topologies/datasets/records/tao.json +12 -0
  729. topologies-0.0.1/src/topologies/datasets/records/tape.json +13 -0
  730. topologies-0.0.1/src/topologies/datasets/records/tartanair.json +13 -0
  731. topologies-0.0.1/src/topologies/datasets/records/tcga.json +13 -0
  732. topologies-0.0.1/src/topologies/datasets/records/ted2020.json +12 -0
  733. topologies-0.0.1/src/topologies/datasets/records/tedlium-3.json +13 -0
  734. topologies-0.0.1/src/topologies/datasets/records/tenrec.json +13 -0
  735. topologies-0.0.1/src/topologies/datasets/records/terra-incognita.json +12 -0
  736. topologies-0.0.1/src/topologies/datasets/records/tess.json +13 -0
  737. topologies-0.0.1/src/topologies/datasets/records/textvqa.json +13 -0
  738. topologies-0.0.1/src/topologies/datasets/records/tgb.json +12 -0
  739. topologies-0.0.1/src/topologies/datasets/records/the-pile.json +13 -0
  740. topologies-0.0.1/src/topologies/datasets/records/theoremqa.json +12 -0
  741. topologies-0.0.1/src/topologies/datasets/records/things3d.json +13 -0
  742. topologies-0.0.1/src/topologies/datasets/records/timit.json +12 -0
  743. topologies-0.0.1/src/topologies/datasets/records/tiny-imagenet.json +13 -0
  744. topologies-0.0.1/src/topologies/datasets/records/titanic.json +13 -0
  745. topologies-0.0.1/src/topologies/datasets/records/totalsegmentator.json +13 -0
  746. topologies-0.0.1/src/topologies/datasets/records/touch-and-go.json +13 -0
  747. topologies-0.0.1/src/topologies/datasets/records/trackingnet.json +13 -0
  748. topologies-0.0.1/src/topologies/datasets/records/traffic.json +13 -0
  749. topologies-0.0.1/src/topologies/datasets/records/transnas-bench-101.json +13 -0
  750. topologies-0.0.1/src/topologies/datasets/records/triviaqa.json +13 -0
  751. topologies-0.0.1/src/topologies/datasets/records/truthfulqa.json +13 -0
  752. topologies-0.0.1/src/topologies/datasets/records/twitch-gamers.json +12 -0
  753. topologies-0.0.1/src/topologies/datasets/records/tydiqa.json +13 -0
  754. topologies-0.0.1/src/topologies/datasets/records/uav123.json +12 -0
  755. topologies-0.0.1/src/topologies/datasets/records/uc-merced.json +13 -0
  756. topologies-0.0.1/src/topologies/datasets/records/ucf-crime.json +13 -0
  757. topologies-0.0.1/src/topologies/datasets/records/ucf101.json +13 -0
  758. topologies-0.0.1/src/topologies/datasets/records/ucr-anomaly-archive.json +13 -0
  759. topologies-0.0.1/src/topologies/datasets/records/ucr.json +13 -0
  760. topologies-0.0.1/src/topologies/datasets/records/ucsd-ped.json +13 -0
  761. topologies-0.0.1/src/topologies/datasets/records/uea.json +13 -0
  762. topologies-0.0.1/src/topologies/datasets/records/uecfood256.json +13 -0
  763. topologies-0.0.1/src/topologies/datasets/records/ujiindoorloc.json +13 -0
  764. topologies-0.0.1/src/topologies/datasets/records/uk-biobank.json +13 -0
  765. topologies-0.0.1/src/topologies/datasets/records/uk-dale.json +13 -0
  766. topologies-0.0.1/src/topologies/datasets/records/ultrachat.json +12 -0
  767. topologies-0.0.1/src/topologies/datasets/records/ultrafeedback.json +12 -0
  768. topologies-0.0.1/src/topologies/datasets/records/umls.json +13 -0
  769. topologies-0.0.1/src/topologies/datasets/records/un-parallel.json +12 -0
  770. topologies-0.0.1/src/topologies/datasets/records/urbansound8k.json +13 -0
  771. topologies-0.0.1/src/topologies/datasets/records/uwb-nlos.json +13 -0
  772. topologies-0.0.1/src/topologies/datasets/records/vaihingen.json +13 -0
  773. topologies-0.0.1/src/topologies/datasets/records/vctk.json +13 -0
  774. topologies-0.0.1/src/topologies/datasets/records/veri-776.json +12 -0
  775. topologies-0.0.1/src/topologies/datasets/records/vggface2.json +13 -0
  776. topologies-0.0.1/src/topologies/datasets/records/vima.json +13 -0
  777. topologies-0.0.1/src/topologies/datasets/records/visa.json +13 -0
  778. topologies-0.0.1/src/topologies/datasets/records/visda-2017.json +12 -0
  779. topologies-0.0.1/src/topologies/datasets/records/visual-genome.json +13 -0
  780. topologies-0.0.1/src/topologies/datasets/records/voicebank-demand.json +12 -0
  781. topologies-0.0.1/src/topologies/datasets/records/voxceleb-1.json +13 -0
  782. topologies-0.0.1/src/topologies/datasets/records/voxceleb-2.json +13 -0
  783. topologies-0.0.1/src/topologies/datasets/records/voxlingua107.json +12 -0
  784. topologies-0.0.1/src/topologies/datasets/records/voxpopuli.json +13 -0
  785. topologies-0.0.1/src/topologies/datasets/records/vqav2.json +13 -0
  786. topologies-0.0.1/src/topologies/datasets/records/waterbirds.json +13 -0
  787. topologies-0.0.1/src/topologies/datasets/records/wavcaps.json +12 -0
  788. topologies-0.0.1/src/topologies/datasets/records/waymo-open.json +13 -0
  789. topologies-0.0.1/src/topologies/datasets/records/weather.json +13 -0
  790. topologies-0.0.1/src/topologies/datasets/records/weatherbench-2.json +13 -0
  791. topologies-0.0.1/src/topologies/datasets/records/wham.json +13 -0
  792. topologies-0.0.1/src/topologies/datasets/records/whamr.json +13 -0
  793. topologies-0.0.1/src/topologies/datasets/records/whu-hi.json +13 -0
  794. topologies-0.0.1/src/topologies/datasets/records/widar-3.json +13 -0
  795. topologies-0.0.1/src/topologies/datasets/records/widar.json +13 -0
  796. topologies-0.0.1/src/topologies/datasets/records/wider-face.json +12 -0
  797. topologies-0.0.1/src/topologies/datasets/records/wikipedia-web-traffic.json +13 -0
  798. topologies-0.0.1/src/topologies/datasets/records/wikitext-103.json +12 -0
  799. topologies-0.0.1/src/topologies/datasets/records/wilds-fmow.json +13 -0
  800. topologies-0.0.1/src/topologies/datasets/records/wilds-rxrx1.json +13 -0
  801. topologies-0.0.1/src/topologies/datasets/records/wind.json +13 -0
  802. topologies-0.0.1/src/topologies/datasets/records/wisdm.json +13 -0
  803. topologies-0.0.1/src/topologies/datasets/records/wit.json +13 -0
  804. topologies-0.0.1/src/topologies/datasets/records/wmt.json +13 -0
  805. topologies-0.0.1/src/topologies/datasets/records/wsj.json +12 -0
  806. topologies-0.0.1/src/topologies/datasets/records/wsj0-2mix.json +12 -0
  807. topologies-0.0.1/src/topologies/datasets/records/xbd.json +13 -0
  808. topologies-0.0.1/src/topologies/datasets/records/xeno-canto.json +13 -0
  809. topologies-0.0.1/src/topologies/datasets/records/xnli.json +13 -0
  810. topologies-0.0.1/src/topologies/datasets/records/xsum.json +13 -0
  811. topologies-0.0.1/src/topologies/datasets/records/xtreme.json +13 -0
  812. topologies-0.0.1/src/topologies/datasets/records/xview.json +13 -0
  813. topologies-0.0.1/src/topologies/datasets/records/yahoo-s5.json +13 -0
  814. topologies-0.0.1/src/topologies/datasets/records/yelp.json +13 -0
  815. topologies-0.0.1/src/topologies/datasets/records/yfcc100m.json +13 -0
  816. topologies-0.0.1/src/topologies/datasets/records/youtube-vis.json +12 -0
  817. topologies-0.0.1/src/topologies/datasets/records/youtube-vos.json +13 -0
  818. topologies-0.0.1/src/topologies/datasets/records/zinc-12k.json +13 -0
  819. topologies-0.0.1/src/topologies/datasets/records/zinc-250k.json +13 -0
  820. topologies-0.0.1/src/topologies/datasets/records/zinc-full.json +13 -0
  821. topologies-0.0.1/src/topologies/datasets/schema.py +278 -0
  822. topologies-0.0.1/src/topologies/enums.py +80 -0
  823. topologies-0.0.1/src/topologies/evaluation/__init__.py +58 -0
  824. topologies-0.0.1/src/topologies/evaluation/api.py +1162 -0
  825. topologies-0.0.1/src/topologies/evaluation/metrics.py +438 -0
  826. topologies-0.0.1/src/topologies/experiments/__init__.py +181 -0
  827. topologies-0.0.1/src/topologies/experiments/analysis.py +645 -0
  828. topologies-0.0.1/src/topologies/experiments/bench.py +435 -0
  829. topologies-0.0.1/src/topologies/experiments/bench_checks.py +251 -0
  830. topologies-0.0.1/src/topologies/experiments/cache.py +229 -0
  831. topologies-0.0.1/src/topologies/experiments/callbacks.py +163 -0
  832. topologies-0.0.1/src/topologies/experiments/cli.py +528 -0
  833. topologies-0.0.1/src/topologies/experiments/coordinator.py +387 -0
  834. topologies-0.0.1/src/topologies/experiments/core.py +499 -0
  835. topologies-0.0.1/src/topologies/experiments/costs.py +180 -0
  836. topologies-0.0.1/src/topologies/experiments/distributions.py +133 -0
  837. topologies-0.0.1/src/topologies/experiments/errors.py +58 -0
  838. topologies-0.0.1/src/topologies/experiments/insights.py +352 -0
  839. topologies-0.0.1/src/topologies/experiments/metrics.py +134 -0
  840. topologies-0.0.1/src/topologies/experiments/parallel.py +294 -0
  841. topologies-0.0.1/src/topologies/experiments/pipeline.py +592 -0
  842. topologies-0.0.1/src/topologies/experiments/planning.py +289 -0
  843. topologies-0.0.1/src/topologies/experiments/pruners.py +86 -0
  844. topologies-0.0.1/src/topologies/experiments/render/__init__.py +1 -0
  845. topologies-0.0.1/src/topologies/experiments/render/console.py +489 -0
  846. topologies-0.0.1/src/topologies/experiments/render/html.py +512 -0
  847. topologies-0.0.1/src/topologies/experiments/render/live.py +130 -0
  848. topologies-0.0.1/src/topologies/experiments/render/markdown.py +73 -0
  849. topologies-0.0.1/src/topologies/experiments/results.py +694 -0
  850. topologies-0.0.1/src/topologies/experiments/routing.py +510 -0
  851. topologies-0.0.1/src/topologies/experiments/runner.py +233 -0
  852. topologies-0.0.1/src/topologies/experiments/space.py +513 -0
  853. topologies-0.0.1/src/topologies/experiments/spec.py +221 -0
  854. topologies-0.0.1/src/topologies/experiments/stats.py +322 -0
  855. topologies-0.0.1/src/topologies/experiments/store.py +198 -0
  856. topologies-0.0.1/src/topologies/experiments/table.py +134 -0
  857. topologies-0.0.1/src/topologies/experiments/tasks.py +905 -0
  858. topologies-0.0.1/src/topologies/experiments/trial.py +425 -0
  859. topologies-0.0.1/src/topologies/experiments/tui.py +14 -0
  860. topologies-0.0.1/src/topologies/experiments/yaml_spec.py +221 -0
  861. topologies-0.0.1/src/topologies/generation.py +252 -0
  862. topologies-0.0.1/src/topologies/implementations/losses/a2c.py +53 -0
  863. topologies-0.0.1/src/topologies/implementations/losses/poly1_loss.py +43 -0
  864. topologies-0.0.1/src/topologies/implementations/optimizers/adabelief.py +54 -0
  865. topologies-0.0.1/src/topologies/implementations/optimizers/adafactor.py +66 -0
  866. topologies-0.0.1/src/topologies/implementations/optimizers/adahessian.py +76 -0
  867. topologies-0.0.1/src/topologies/implementations/optimizers/adam8bit.py +75 -0
  868. topologies-0.0.1/src/topologies/implementations/optimizers/adam_mini.py +78 -0
  869. topologies-0.0.1/src/topologies/implementations/optimizers/ademamix.py +59 -0
  870. topologies-0.0.1/src/topologies/implementations/optimizers/asam.py +71 -0
  871. topologies-0.0.1/src/topologies/implementations/optimizers/core.py +29 -0
  872. topologies-0.0.1/src/topologies/implementations/optimizers/ema.py +49 -0
  873. topologies-0.0.1/src/topologies/implementations/optimizers/fadam.py +58 -0
  874. topologies-0.0.1/src/topologies/implementations/optimizers/fisher_sam.py +76 -0
  875. topologies-0.0.1/src/topologies/implementations/optimizers/flora.py +41 -0
  876. topologies-0.0.1/src/topologies/implementations/optimizers/galore.py +42 -0
  877. topologies-0.0.1/src/topologies/implementations/optimizers/grokfast.py +46 -0
  878. topologies-0.0.1/src/topologies/implementations/optimizers/gsam.py +67 -0
  879. topologies-0.0.1/src/topologies/implementations/optimizers/kfac.py +43 -0
  880. topologies-0.0.1/src/topologies/implementations/optimizers/lamb.py +63 -0
  881. topologies-0.0.1/src/topologies/implementations/optimizers/lars.py +46 -0
  882. topologies-0.0.1/src/topologies/implementations/optimizers/lion.py +49 -0
  883. topologies-0.0.1/src/topologies/implementations/optimizers/lookahead.py +53 -0
  884. topologies-0.0.1/src/topologies/implementations/optimizers/model_soup.py +50 -0
  885. topologies-0.0.1/src/topologies/implementations/optimizers/muon.py +41 -0
  886. topologies-0.0.1/src/topologies/implementations/optimizers/novograd.py +62 -0
  887. topologies-0.0.1/src/topologies/implementations/optimizers/polynomial_decay.py +52 -0
  888. topologies-0.0.1/src/topologies/implementations/optimizers/qlora.py +18 -0
  889. topologies-0.0.1/src/topologies/implementations/optimizers/radam.py +71 -0
  890. topologies-0.0.1/src/topologies/implementations/optimizers/relora.py +17 -0
  891. topologies-0.0.1/src/topologies/implementations/optimizers/sam.py +78 -0
  892. topologies-0.0.1/src/topologies/implementations/optimizers/schedule_free.py +38 -0
  893. topologies-0.0.1/src/topologies/implementations/optimizers/shampoo.py +60 -0
  894. topologies-0.0.1/src/topologies/implementations/optimizers/sm3.py +44 -0
  895. topologies-0.0.1/src/topologies/implementations/optimizers/soap.py +54 -0
  896. topologies-0.0.1/src/topologies/implementations/optimizers/sophia.py +61 -0
  897. topologies-0.0.1/src/topologies/implementations/optimizers/swa.py +50 -0
  898. topologies-0.0.1/src/topologies/implementations/optimizers/swad.py +52 -0
  899. topologies-0.0.1/src/topologies/implementations/optimizers/swalp.py +43 -0
  900. topologies-0.0.1/src/topologies/implementations/optimizers/twa.py +50 -0
  901. topologies-0.0.1/src/topologies/implementations/schedulers/clr.py +21 -0
  902. topologies-0.0.1/src/topologies/implementations/schedulers/cooldown.py +37 -0
  903. topologies-0.0.1/src/topologies/implementations/schedulers/cosine_min.py +19 -0
  904. topologies-0.0.1/src/topologies/implementations/schedulers/cosine_restarts.py +24 -0
  905. topologies-0.0.1/src/topologies/implementations/schedulers/exponential.py +32 -0
  906. topologies-0.0.1/src/topologies/implementations/schedulers/grafted.py +43 -0
  907. topologies-0.0.1/src/topologies/implementations/schedulers/multistep.py +21 -0
  908. topologies-0.0.1/src/topologies/implementations/schedulers/noam.py +39 -0
  909. topologies-0.0.1/src/topologies/implementations/schedulers/onecycle.py +10 -0
  910. topologies-0.0.1/src/topologies/implementations/schedulers/reduce_on_plateau.py +22 -0
  911. topologies-0.0.1/src/topologies/implementations/schedulers/sgdr.py +24 -0
  912. topologies-0.0.1/src/topologies/implementations/schedulers/trapezoidal.py +42 -0
  913. topologies-0.0.1/src/topologies/implementations/schedulers/warmup_linear.py +32 -0
  914. topologies-0.0.1/src/topologies/implementations/schedulers/wsd.py +46 -0
  915. topologies-0.0.1/src/topologies/implementations/training/core.py +298 -0
  916. topologies-0.0.1/src/topologies/inference.py +88 -0
  917. topologies-0.0.1/src/topologies/inference_tasks/__init__.py +1 -0
  918. topologies-0.0.1/src/topologies/inference_tasks/audio.py +26 -0
  919. topologies-0.0.1/src/topologies/inference_tasks/detection.py +63 -0
  920. topologies-0.0.1/src/topologies/inference_tasks/diffusion.py +114 -0
  921. topologies-0.0.1/src/topologies/inference_tasks/music.py +34 -0
  922. topologies-0.0.1/src/topologies/inference_tasks/retrieval.py +19 -0
  923. topologies-0.0.1/src/topologies/introspection/__init__.py +84 -0
  924. topologies-0.0.1/src/topologies/introspection/_tree.py +111 -0
  925. topologies-0.0.1/src/topologies/introspection/adapters/__init__.py +159 -0
  926. topologies-0.0.1/src/topologies/introspection/adapters/builtins.py +20 -0
  927. topologies-0.0.1/src/topologies/introspection/adapters/diffusion.py +41 -0
  928. topologies-0.0.1/src/topologies/introspection/adapters/generic.py +17 -0
  929. topologies-0.0.1/src/topologies/introspection/adapters/huggingface.py +96 -0
  930. topologies-0.0.1/src/topologies/introspection/adapters/topologies_catalog.py +212 -0
  931. topologies-0.0.1/src/topologies/introspection/adapters/torch_modules.py +34 -0
  932. topologies-0.0.1/src/topologies/introspection/api.py +303 -0
  933. topologies-0.0.1/src/topologies/introspection/capture.py +580 -0
  934. topologies-0.0.1/src/topologies/introspection/cli.py +70 -0
  935. topologies-0.0.1/src/topologies/introspection/collection.py +457 -0
  936. topologies-0.0.1/src/topologies/introspection/errors.py +35 -0
  937. topologies-0.0.1/src/topologies/introspection/functional.py +339 -0
  938. topologies-0.0.1/src/topologies/introspection/hooks.py +262 -0
  939. topologies-0.0.1/src/topologies/introspection/html.py +122 -0
  940. topologies-0.0.1/src/topologies/introspection/inputs.py +180 -0
  941. topologies-0.0.1/src/topologies/introspection/integrations.py +110 -0
  942. topologies-0.0.1/src/topologies/introspection/interventions.py +388 -0
  943. topologies-0.0.1/src/topologies/introspection/objectives.py +261 -0
  944. topologies-0.0.1/src/topologies/introspection/payloads.py +116 -0
  945. topologies-0.0.1/src/topologies/introspection/reducers.py +755 -0
  946. topologies-0.0.1/src/topologies/introspection/report.py +454 -0
  947. topologies-0.0.1/src/topologies/introspection/semantics.py +218 -0
  948. topologies-0.0.1/src/topologies/introspection/session.py +386 -0
  949. topologies-0.0.1/src/topologies/introspection/sites.py +501 -0
  950. topologies-0.0.1/src/topologies/introspection/store.py +148 -0
  951. topologies-0.0.1/src/topologies/introspection/subject.py +236 -0
  952. topologies-0.0.1/src/topologies/introspection/techniques/__init__.py +220 -0
  953. topologies-0.0.1/src/topologies/introspection/techniques/_features.py +109 -0
  954. topologies-0.0.1/src/topologies/introspection/techniques/activations.py +215 -0
  955. topologies-0.0.1/src/topologies/introspection/techniques/anomalies.py +224 -0
  956. topologies-0.0.1/src/topologies/introspection/techniques/attention.py +280 -0
  957. topologies-0.0.1/src/topologies/introspection/techniques/attribution.py +450 -0
  958. topologies-0.0.1/src/topologies/introspection/techniques/dictionary.py +205 -0
  959. topologies-0.0.1/src/topologies/introspection/techniques/geometry.py +124 -0
  960. topologies-0.0.1/src/topologies/introspection/techniques/gradients.py +144 -0
  961. topologies-0.0.1/src/topologies/introspection/techniques/interventions.py +447 -0
  962. topologies-0.0.1/src/topologies/introspection/techniques/lens.py +173 -0
  963. topologies-0.0.1/src/topologies/introspection/techniques/maximization.py +191 -0
  964. topologies-0.0.1/src/topologies/introspection/techniques/neurons.py +206 -0
  965. topologies-0.0.1/src/topologies/introspection/techniques/probing.py +224 -0
  966. topologies-0.0.1/src/topologies/introspection/techniques/process.py +261 -0
  967. topologies-0.0.1/src/topologies/introspection/techniques/similarity.py +176 -0
  968. topologies-0.0.1/src/topologies/introspection/techniques/structure.py +246 -0
  969. topologies-0.0.1/src/topologies/introspection/techniques/weights.py +156 -0
  970. topologies-0.0.1/src/topologies/introspection/trace.py +324 -0
  971. topologies-0.0.1/src/topologies/introspection/views.py +168 -0
  972. topologies-0.0.1/src/topologies/layers/__init__.py +0 -0
  973. topologies-0.0.1/src/topologies/layers/alphafold.py +27 -0
  974. topologies-0.0.1/src/topologies/layers/audio_frontend.py +138 -0
  975. topologies-0.0.1/src/topologies/layers/audio_perturb.py +61 -0
  976. topologies-0.0.1/src/topologies/layers/augmix.py +62 -0
  977. topologies-0.0.1/src/topologies/layers/auprc.py +26 -0
  978. topologies-0.0.1/src/topologies/layers/autoformer.py +26 -0
  979. topologies-0.0.1/src/topologies/layers/awp.py +55 -0
  980. topologies-0.0.1/src/topologies/layers/bcq.py +40 -0
  981. topologies-0.0.1/src/topologies/layers/bevformer.py +111 -0
  982. topologies-0.0.1/src/topologies/layers/bic.py +17 -0
  983. topologies-0.0.1/src/topologies/layers/bidirectional_rnn_wrapper.py +34 -0
  984. topologies-0.0.1/src/topologies/layers/bifpn.py +28 -0
  985. topologies-0.0.1/src/topologies/layers/bigbird_attention.py +52 -0
  986. topologies-0.0.1/src/topologies/layers/bigvgan.py +23 -0
  987. topologies-0.0.1/src/topologies/layers/box_coder.py +53 -0
  988. topologies-0.0.1/src/topologies/layers/causal_temporal_block.py +22 -0
  989. topologies-0.0.1/src/topologies/layers/cbam_gate.py +43 -0
  990. topologies-0.0.1/src/topologies/layers/chemformer.py +79 -0
  991. topologies-0.0.1/src/topologies/layers/chunked_cross_attention.py +47 -0
  992. topologies-0.0.1/src/topologies/layers/cluster_gcn.py +59 -0
  993. topologies-0.0.1/src/topologies/layers/cnp.py +15 -0
  994. topologies-0.0.1/src/topologies/layers/complex.py +44 -0
  995. topologies-0.0.1/src/topologies/layers/conformer.py +59 -0
  996. topologies-0.0.1/src/topologies/layers/contrastive_search.py +45 -0
  997. topologies-0.0.1/src/topologies/layers/conv_blocks.py +26 -0
  998. topologies-0.0.1/src/topologies/layers/coverage.py +11 -0
  999. topologies-0.0.1/src/topologies/layers/cpvt.py +41 -0
  1000. topologies-0.0.1/src/topologies/layers/crepe.py +43 -0
  1001. topologies-0.0.1/src/topologies/layers/crestereo.py +88 -0
  1002. topologies-0.0.1/src/topologies/layers/cvt.py +36 -0
  1003. topologies-0.0.1/src/topologies/layers/ddim.py +21 -0
  1004. topologies-0.0.1/src/topologies/layers/deeplob.py +111 -0
  1005. topologies-0.0.1/src/topologies/layers/depth_mu_p.py +45 -0
  1006. topologies-0.0.1/src/topologies/layers/diagonal_ssm.py +33 -0
  1007. topologies-0.0.1/src/topologies/layers/diffusion_policy.py +121 -0
  1008. topologies-0.0.1/src/topologies/layers/diffusion_residual.py +27 -0
  1009. topologies-0.0.1/src/topologies/layers/distmult.py +23 -0
  1010. topologies-0.0.1/src/topologies/layers/drop_path.py +19 -0
  1011. topologies-0.0.1/src/topologies/layers/dsp_utils.py +79 -0
  1012. topologies-0.0.1/src/topologies/layers/dueling_q_head.py +16 -0
  1013. topologies-0.0.1/src/topologies/layers/edgenext.py +53 -0
  1014. topologies-0.0.1/src/topologies/layers/edm_heun.py +26 -0
  1015. topologies-0.0.1/src/topologies/layers/efficientad.py +11 -0
  1016. topologies-0.0.1/src/topologies/layers/efficientdet.py +32 -0
  1017. topologies-0.0.1/src/topologies/layers/feature_pyramid.py +22 -0
  1018. topologies-0.0.1/src/topologies/layers/feed_forward.py +27 -0
  1019. topologies-0.0.1/src/topologies/layers/film.py +42 -0
  1020. topologies-0.0.1/src/topologies/layers/fir_filter.py +22 -0
  1021. topologies-0.0.1/src/topologies/layers/flownet2.py +51 -0
  1022. topologies-0.0.1/src/topologies/layers/fm_interaction.py +11 -0
  1023. topologies-0.0.1/src/topologies/layers/fsq.py +24 -0
  1024. topologies-0.0.1/src/topologies/layers/ft_transformer.py +21 -0
  1025. topologies-0.0.1/src/topologies/layers/gail.py +33 -0
  1026. topologies-0.0.1/src/topologies/layers/galerkin_attention.py +24 -0
  1027. topologies-0.0.1/src/topologies/layers/galore.py +18 -0
  1028. topologies-0.0.1/src/topologies/layers/geglu.py +17 -0
  1029. topologies-0.0.1/src/topologies/layers/geglu_ffn.py +21 -0
  1030. topologies-0.0.1/src/topologies/layers/geneformer.py +45 -0
  1031. topologies-0.0.1/src/topologies/layers/ghostnet.py +44 -0
  1032. topologies-0.0.1/src/topologies/layers/git.py +55 -0
  1033. topologies-0.0.1/src/topologies/layers/gmt.py +44 -0
  1034. topologies-0.0.1/src/topologies/layers/gp_bo.py +39 -0
  1035. topologies-0.0.1/src/topologies/layers/graph_convolution.py +64 -0
  1036. topologies-0.0.1/src/topologies/layers/graph_readout.py +32 -0
  1037. topologies-0.0.1/src/topologies/layers/groupnorm_block.py +21 -0
  1038. topologies-0.0.1/src/topologies/layers/hard_swish.py +8 -0
  1039. topologies-0.0.1/src/topologies/layers/harmonic_oscillator.py +39 -0
  1040. topologies-0.0.1/src/topologies/layers/hole.py +37 -0
  1041. topologies-0.0.1/src/topologies/layers/hstu.py +42 -0
  1042. topologies-0.0.1/src/topologies/layers/hyena_block.py +51 -0
  1043. topologies-0.0.1/src/topologies/layers/hyena_operator.py +36 -0
  1044. topologies-0.0.1/src/topologies/layers/i3d.py +17 -0
  1045. topologies-0.0.1/src/topologies/layers/imap.py +37 -0
  1046. topologies-0.0.1/src/topologies/layers/inception_resnet.py +17 -0
  1047. topologies-0.0.1/src/topologies/layers/influence.py +20 -0
  1048. topologies-0.0.1/src/topologies/layers/informer.py +22 -0
  1049. topologies-0.0.1/src/topologies/layers/instancenorm_block.py +20 -0
  1050. topologies-0.0.1/src/topologies/layers/instant_ngp.py +54 -0
  1051. topologies-0.0.1/src/topologies/layers/integrated_gradients.py +26 -0
  1052. topologies-0.0.1/src/topologies/layers/inverted_residual.py +21 -0
  1053. topologies-0.0.1/src/topologies/layers/joint_energy.py +44 -0
  1054. topologies-0.0.1/src/topologies/layers/jukebox.py +44 -0
  1055. topologies-0.0.1/src/topologies/layers/k_planes.py +45 -0
  1056. topologies-0.0.1/src/topologies/layers/kan.py +64 -0
  1057. topologies-0.0.1/src/topologies/layers/kerple.py +34 -0
  1058. topologies-0.0.1/src/topologies/layers/kpconv.py +61 -0
  1059. topologies-0.0.1/src/topologies/layers/lars.py +28 -0
  1060. topologies-0.0.1/src/topologies/layers/layer_scale.py +20 -0
  1061. topologies-0.0.1/src/topologies/layers/layernorm.py +20 -0
  1062. topologies-0.0.1/src/topologies/layers/layernorm_affine.py +18 -0
  1063. topologies-0.0.1/src/topologies/layers/layoutlm.py +37 -0
  1064. topologies-0.0.1/src/topologies/layers/leaf.py +60 -0
  1065. topologies-0.0.1/src/topologies/layers/learned_position.py +22 -0
  1066. topologies-0.0.1/src/topologies/layers/linear_attention.py +42 -0
  1067. topologies-0.0.1/src/topologies/layers/linear_chain_crf.py +58 -0
  1068. topologies-0.0.1/src/topologies/layers/linformer.py +37 -0
  1069. topologies-0.0.1/src/topologies/layers/llava.py +19 -0
  1070. topologies-0.0.1/src/topologies/layers/mamba_block.py +39 -0
  1071. topologies-0.0.1/src/topologies/layers/masked_pooling.py +44 -0
  1072. topologies-0.0.1/src/topologies/layers/memorizing_attention.py +86 -0
  1073. topologies-0.0.1/src/topologies/layers/meshcnn.py +379 -0
  1074. topologies-0.0.1/src/topologies/layers/minigpt4.py +18 -0
  1075. topologies-0.0.1/src/topologies/layers/moirai.py +24 -0
  1076. topologies-0.0.1/src/topologies/layers/multi_head_attention.py +59 -0
  1077. topologies-0.0.1/src/topologies/layers/nearest_neighbor_retriever.py +23 -0
  1078. topologies-0.0.1/src/topologies/layers/neuralucb_bonus.py +16 -0
  1079. topologies-0.0.1/src/topologies/layers/nlu_heads.py +46 -0
  1080. topologies-0.0.1/src/topologies/layers/node_ensemble.py +71 -0
  1081. topologies-0.0.1/src/topologies/layers/noisy_nets.py +55 -0
  1082. topologies-0.0.1/src/topologies/layers/paged_attention.py +103 -0
  1083. topologies-0.0.1/src/topologies/layers/pali.py +52 -0
  1084. topologies-0.0.1/src/topologies/layers/patch_embedding.py +25 -0
  1085. topologies-0.0.1/src/topologies/layers/pcen.py +37 -0
  1086. topologies-0.0.1/src/topologies/layers/perceiver_io.py +30 -0
  1087. topologies-0.0.1/src/topologies/layers/performer.py +46 -0
  1088. topologies-0.0.1/src/topologies/layers/pointcnn.py +62 -0
  1089. topologies-0.0.1/src/topologies/layers/pointllm.py +25 -0
  1090. topologies-0.0.1/src/topologies/layers/ppo_clip.py +19 -0
  1091. topologies-0.0.1/src/topologies/layers/prelu.py +21 -0
  1092. topologies-0.0.1/src/topologies/layers/projection_head.py +16 -0
  1093. topologies-0.0.1/src/topologies/layers/qk_norm_attention.py +58 -0
  1094. topologies-0.0.1/src/topologies/layers/quartznet.py +41 -0
  1095. topologies-0.0.1/src/topologies/layers/quate.py +52 -0
  1096. topologies-0.0.1/src/topologies/layers/randaugment.py +99 -0
  1097. topologies-0.0.1/src/topologies/layers/rectified_flow.py +63 -0
  1098. topologies-0.0.1/src/topologies/layers/recurrent_backbone.py +27 -0
  1099. topologies-0.0.1/src/topologies/layers/replaced_token_detection_head.py +36 -0
  1100. topologies-0.0.1/src/topologies/layers/residual_blocks.py +19 -0
  1101. topologies-0.0.1/src/topologies/layers/resnet_v15.py +43 -0
  1102. topologies-0.0.1/src/topologies/layers/rmsnorm.py +16 -0
  1103. topologies-0.0.1/src/topologies/layers/rnn_transducer.py +42 -0
  1104. topologies-0.0.1/src/topologies/layers/rotary_embedding.py +24 -0
  1105. topologies-0.0.1/src/topologies/layers/rotate.py +46 -0
  1106. topologies-0.0.1/src/topologies/layers/rwkv_channel_mix.py +30 -0
  1107. topologies-0.0.1/src/topologies/layers/s3d.py +33 -0
  1108. topologies-0.0.1/src/topologies/layers/s4_kernel.py +93 -0
  1109. topologies-0.0.1/src/topologies/layers/saint.py +26 -0
  1110. topologies-0.0.1/src/topologies/layers/score_sde.py +46 -0
  1111. topologies-0.0.1/src/topologies/layers/seanet.py +43 -0
  1112. topologies-0.0.1/src/topologies/layers/sentence_bert.py +21 -0
  1113. topologies-0.0.1/src/topologies/layers/shufflenet_v2.py +27 -0
  1114. topologies-0.0.1/src/topologies/layers/silu_activation.py +16 -0
  1115. topologies-0.0.1/src/topologies/layers/simkd.py +28 -0
  1116. topologies-0.0.1/src/topologies/layers/simple.py +33 -0
  1117. topologies-0.0.1/src/topologies/layers/simplenet.py +47 -0
  1118. topologies-0.0.1/src/topologies/layers/sinusoidal_position.py +22 -0
  1119. topologies-0.0.1/src/topologies/layers/skip_upsample.py +20 -0
  1120. topologies-0.0.1/src/topologies/layers/skipinit.py +18 -0
  1121. topologies-0.0.1/src/topologies/layers/slot_attention.py +57 -0
  1122. topologies-0.0.1/src/topologies/layers/snake.py +13 -0
  1123. topologies-0.0.1/src/topologies/layers/snip.py +13 -0
  1124. topologies-0.0.1/src/topologies/layers/source_separation.py +15 -0
  1125. topologies-0.0.1/src/topologies/layers/speaker_embedding.py +22 -0
  1126. topologies-0.0.1/src/topologies/layers/specaugment.py +65 -0
  1127. topologies-0.0.1/src/topologies/layers/squeeze_excitation.py +18 -0
  1128. topologies-0.0.1/src/topologies/layers/stargan_v2.py +65 -0
  1129. topologies-0.0.1/src/topologies/layers/stft_frontend.py +27 -0
  1130. topologies-0.0.1/src/topologies/layers/structure_module.py +111 -0
  1131. topologies-0.0.1/src/topologies/layers/stylegan2.py +57 -0
  1132. topologies-0.0.1/src/topologies/layers/surrogate_gradient.py +57 -0
  1133. topologies-0.0.1/src/topologies/layers/tabtransformer.py +69 -0
  1134. topologies-0.0.1/src/topologies/layers/temporal_shift.py +31 -0
  1135. topologies-0.0.1/src/topologies/layers/tfn.py +60 -0
  1136. topologies-0.0.1/src/topologies/layers/tide.py +136 -0
  1137. topologies-0.0.1/src/topologies/layers/top_2_router.py +46 -0
  1138. topologies-0.0.1/src/topologies/layers/topk_moe_layer.py +63 -0
  1139. topologies-0.0.1/src/topologies/layers/transducer_asr.py +53 -0
  1140. topologies-0.0.1/src/topologies/layers/transformer_block.py +66 -0
  1141. topologies-0.0.1/src/topologies/layers/transh.py +26 -0
  1142. topologies-0.0.1/src/topologies/layers/tsmixer.py +60 -0
  1143. topologies-0.0.1/src/topologies/layers/ttt_titans_mblm.py +50 -0
  1144. topologies-0.0.1/src/topologies/layers/tucker.py +40 -0
  1145. topologies-0.0.1/src/topologies/layers/unet3plus.py +60 -0
  1146. topologies-0.0.1/src/topologies/layers/unetpp.py +42 -0
  1147. topologies-0.0.1/src/topologies/layers/v_trace.py +82 -0
  1148. topologies-0.0.1/src/topologies/layers/vector_quantizer.py +25 -0
  1149. topologies-0.0.1/src/topologies/layers/waveglow.py +45 -0
  1150. topologies-0.0.1/src/topologies/layers/whisper.py +54 -0
  1151. topologies-0.0.1/src/topologies/losses/catalog.json +529 -0
  1152. topologies-0.0.1/src/topologies/metadata/__init__.py +30 -0
  1153. topologies-0.0.1/src/topologies/metadata/schema.py +446 -0
  1154. topologies-0.0.1/src/topologies/metrics.py +206 -0
  1155. topologies-0.0.1/src/topologies/models/__init__.py +0 -0
  1156. topologies-0.0.1/src/topologies/models/actor_critic.py +33 -0
  1157. topologies-0.0.1/src/topologies/models/alexnet.py +57 -0
  1158. topologies-0.0.1/src/topologies/models/attention_nmt.py +93 -0
  1159. topologies-0.0.1/src/topologies/models/attn2d.py +68 -0
  1160. topologies-0.0.1/src/topologies/models/audio_crnn.py +79 -0
  1161. topologies-0.0.1/src/topologies/models/autoencoders.py +106 -0
  1162. topologies-0.0.1/src/topologies/models/autoint.py +50 -0
  1163. topologies-0.0.1/src/topologies/models/awr.py +46 -0
  1164. topologies-0.0.1/src/topologies/models/bart.py +121 -0
  1165. topologies-0.0.1/src/topologies/models/bert4rec.py +70 -0
  1166. topologies-0.0.1/src/topologies/models/bert_encoder.py +845 -0
  1167. topologies-0.0.1/src/topologies/models/bigbird.py +95 -0
  1168. topologies-0.0.1/src/topologies/models/biggan.py +407 -0
  1169. topologies-0.0.1/src/topologies/models/blip2.py +37 -0
  1170. topologies-0.0.1/src/topologies/models/bst.py +41 -0
  1171. topologies-0.0.1/src/topologies/models/byt5.py +204 -0
  1172. topologies-0.0.1/src/topologies/models/c3d.py +27 -0
  1173. topologies-0.0.1/src/topologies/models/canine.py +221 -0
  1174. topologies-0.0.1/src/topologies/models/cascade_rcnn.py +123 -0
  1175. topologies-0.0.1/src/topologies/models/caser.py +51 -0
  1176. topologies-0.0.1/src/topologies/models/centernet.py +26 -0
  1177. topologies-0.0.1/src/topologies/models/cgan.py +52 -0
  1178. topologies-0.0.1/src/topologies/models/charformer.py +272 -0
  1179. topologies-0.0.1/src/topologies/models/compact_convnet.py +38 -0
  1180. topologies-0.0.1/src/topologies/models/consistency_model.py +103 -0
  1181. topologies-0.0.1/src/topologies/models/contrastive_vision.py +90 -0
  1182. topologies-0.0.1/src/topologies/models/conv_asr.py +59 -0
  1183. topologies-0.0.1/src/topologies/models/convnext.py +88 -0
  1184. topologies-0.0.1/src/topologies/models/ctc_acoustic_model.py +63 -0
  1185. topologies-0.0.1/src/topologies/models/cyclegan.py +41 -0
  1186. topologies-0.0.1/src/topologies/models/dalle.py +56 -0
  1187. topologies-0.0.1/src/topologies/models/dbn.py +67 -0
  1188. topologies-0.0.1/src/topologies/models/dcgan.py +65 -0
  1189. topologies-0.0.1/src/topologies/models/dcn_v2.py +40 -0
  1190. topologies-0.0.1/src/topologies/models/dcrnn.py +160 -0
  1191. topologies-0.0.1/src/topologies/models/ddsp_synthesizer.py +51 -0
  1192. topologies-0.0.1/src/topologies/models/deberta.py +344 -0
  1193. topologies-0.0.1/src/topologies/models/decision_transformer.py +67 -0
  1194. topologies-0.0.1/src/topologies/models/deepfm.py +36 -0
  1195. topologies-0.0.1/src/topologies/models/deis.py +25 -0
  1196. topologies-0.0.1/src/topologies/models/deit.py +69 -0
  1197. topologies-0.0.1/src/topologies/models/densenet.py +85 -0
  1198. topologies-0.0.1/src/topologies/models/detect_ops.py +101 -0
  1199. topologies-0.0.1/src/topologies/models/detr.py +53 -0
  1200. topologies-0.0.1/src/topologies/models/diagonal_ssm_model.py +38 -0
  1201. topologies-0.0.1/src/topologies/models/diffusion_unet.py +94 -0
  1202. topologies-0.0.1/src/topologies/models/dinov2.py +53 -0
  1203. topologies-0.0.1/src/topologies/models/dip_vae.py +65 -0
  1204. topologies-0.0.1/src/topologies/models/distributional_q.py +119 -0
  1205. topologies-0.0.1/src/topologies/models/dit.py +109 -0
  1206. topologies-0.0.1/src/topologies/models/dlinear.py +47 -0
  1207. topologies-0.0.1/src/topologies/models/dpm_solver.py +31 -0
  1208. topologies-0.0.1/src/topologies/models/dqn_family.py +33 -0
  1209. topologies-0.0.1/src/topologies/models/dreamerv2.py +55 -0
  1210. topologies-0.0.1/src/topologies/models/dreamerv3.py +52 -0
  1211. topologies-0.0.1/src/topologies/models/droq.py +49 -0
  1212. topologies-0.0.1/src/topologies/models/dss_s5_mamba.py +156 -0
  1213. topologies-0.0.1/src/topologies/models/e_branchformer.py +51 -0
  1214. topologies-0.0.1/src/topologies/models/ease.py +48 -0
  1215. topologies-0.0.1/src/topologies/models/ecapa_tdnn.py +40 -0
  1216. topologies-0.0.1/src/topologies/models/efficient_attention.py +94 -0
  1217. topologies-0.0.1/src/topologies/models/efficient_conv.py +102 -0
  1218. topologies-0.0.1/src/topologies/models/fastpitch.py +41 -0
  1219. topologies-0.0.1/src/topologies/models/fastspeech2.py +45 -0
  1220. topologies-0.0.1/src/topologies/models/ffjord.py +48 -0
  1221. topologies-0.0.1/src/topologies/models/fibinet.py +53 -0
  1222. topologies-0.0.1/src/topologies/models/fits.py +34 -0
  1223. topologies-0.0.1/src/topologies/models/flow_matching.py +36 -0
  1224. topologies-0.0.1/src/topologies/models/fno.py +38 -0
  1225. topologies-0.0.1/src/topologies/models/fsq.py +22 -0
  1226. topologies-0.0.1/src/topologies/models/fullsubnet.py +60 -0
  1227. topologies-0.0.1/src/topologies/models/gatv2.py +50 -0
  1228. topologies-0.0.1/src/topologies/models/gemma.py +189 -0
  1229. topologies-0.0.1/src/topologies/models/glow_tts.py +109 -0
  1230. topologies-0.0.1/src/topologies/models/gmlp.py +57 -0
  1231. topologies-0.0.1/src/topologies/models/gmt.py +44 -0
  1232. topologies-0.0.1/src/topologies/models/gpt2.py +120 -0
  1233. topologies-0.0.1/src/topologies/models/gpt_decoder_family.py +297 -0
  1234. topologies-0.0.1/src/topologies/models/gpt_decoder_lm.py +71 -0
  1235. topologies-0.0.1/src/topologies/models/gpt_j_neox_pythia.py +135 -0
  1236. topologies-0.0.1/src/topologies/models/grace.py +73 -0
  1237. topologies-0.0.1/src/topologies/models/grad_tts.py +42 -0
  1238. topologies-0.0.1/src/topologies/models/graph_network.py +61 -0
  1239. topologies-0.0.1/src/topologies/models/graph_wavenet.py +34 -0
  1240. topologies-0.0.1/src/topologies/models/graphsage.py +57 -0
  1241. topologies-0.0.1/src/topologies/models/graphsaint.py +11 -0
  1242. topologies-0.0.1/src/topologies/models/han.py +10 -0
  1243. topologies-0.0.1/src/topologies/models/hnn.py +38 -0
  1244. topologies-0.0.1/src/topologies/models/hornet.py +29 -0
  1245. topologies-0.0.1/src/topologies/models/hyena.py +71 -0
  1246. topologies-0.0.1/src/topologies/models/i_jepa.py +32 -0
  1247. topologies-0.0.1/src/topologies/models/iaf.py +57 -0
  1248. topologies-0.0.1/src/topologies/models/icm.py +45 -0
  1249. topologies-0.0.1/src/topologies/models/image_text_dual_encoder.py +77 -0
  1250. topologies-0.0.1/src/topologies/models/imagegpt.py +53 -0
  1251. topologies-0.0.1/src/topologies/models/imap.py +38 -0
  1252. topologies-0.0.1/src/topologies/models/inception.py +94 -0
  1253. topologies-0.0.1/src/topologies/models/infovae.py +65 -0
  1254. topologies-0.0.1/src/topologies/models/ipa.py +69 -0
  1255. topologies-0.0.1/src/topologies/models/ipo.py +31 -0
  1256. topologies-0.0.1/src/topologies/models/iqn.py +52 -0
  1257. topologies-0.0.1/src/topologies/models/item_knn.py +26 -0
  1258. topologies-0.0.1/src/topologies/models/itransformer.py +31 -0
  1259. topologies-0.0.1/src/topologies/models/jamba_griffin_hawk.py +155 -0
  1260. topologies-0.0.1/src/topologies/models/kto.py +38 -0
  1261. topologies-0.0.1/src/topologies/models/ladder_vae.py +66 -0
  1262. topologies-0.0.1/src/topologies/models/latent_diffusion.py +303 -0
  1263. topologies-0.0.1/src/topologies/models/lenet.py +42 -0
  1264. topologies-0.0.1/src/topologies/models/lfq.py +22 -0
  1265. topologies-0.0.1/src/topologies/models/lightgcn.py +107 -0
  1266. topologies-0.0.1/src/topologies/models/line.py +45 -0
  1267. topologies-0.0.1/src/topologies/models/llama.py +171 -0
  1268. topologies-0.0.1/src/topologies/models/llama_decoder.py +84 -0
  1269. topologies-0.0.1/src/topologies/models/lnn.py +59 -0
  1270. topologies-0.0.1/src/topologies/models/longformer.py +106 -0
  1271. topologies-0.0.1/src/topologies/models/lstnet.py +45 -0
  1272. topologies-0.0.1/src/topologies/models/maddpg.py +50 -0
  1273. topologies-0.0.1/src/topologies/models/maf.py +57 -0
  1274. topologies-0.0.1/src/topologies/models/magvit.py +89 -0
  1275. topologies-0.0.1/src/topologies/models/mamba2_gated_deltanet_h3.py +122 -0
  1276. topologies-0.0.1/src/topologies/models/mappo.py +60 -0
  1277. topologies-0.0.1/src/topologies/models/marian.py +105 -0
  1278. topologies-0.0.1/src/topologies/models/mask2former.py +46 -0
  1279. topologies-0.0.1/src/topologies/models/mask_rcnn.py +26 -0
  1280. topologies-0.0.1/src/topologies/models/masked_image.py +62 -0
  1281. topologies-0.0.1/src/topologies/models/maskfeat.py +80 -0
  1282. topologies-0.0.1/src/topologies/models/maskgit.py +87 -0
  1283. topologies-0.0.1/src/topologies/models/matcha_tts.py +55 -0
  1284. topologies-0.0.1/src/topologies/models/matchboxnet.py +36 -0
  1285. topologies-0.0.1/src/topologies/models/matrix_factorization.py +31 -0
  1286. topologies-0.0.1/src/topologies/models/mistral.py +153 -0
  1287. topologies-0.0.1/src/topologies/models/mistral_decoder.py +103 -0
  1288. topologies-0.0.1/src/topologies/models/mixtral.py +180 -0
  1289. topologies-0.0.1/src/topologies/models/mlp.py +50 -0
  1290. topologies-0.0.1/src/topologies/models/mobilenet_v2.py +60 -0
  1291. topologies-0.0.1/src/topologies/models/modern_conv.py +99 -0
  1292. topologies-0.0.1/src/topologies/models/modernbert.py +182 -0
  1293. topologies-0.0.1/src/topologies/models/mpo.py +47 -0
  1294. topologies-0.0.1/src/topologies/models/mult_vae.py +48 -0
  1295. topologies-0.0.1/src/topologies/models/music_transformer.py +61 -0
  1296. topologies-0.0.1/src/topologies/models/musicgen.py +102 -0
  1297. topologies-0.0.1/src/topologies/models/muzero.py +59 -0
  1298. topologies-0.0.1/src/topologies/models/mvit.py +42 -0
  1299. topologies-0.0.1/src/topologies/models/nbeats.py +66 -0
  1300. topologies-0.0.1/src/topologies/models/neural_ode.py +26 -0
  1301. topologies-0.0.1/src/topologies/models/neural_spline_flows.py +78 -0
  1302. topologies-0.0.1/src/topologies/models/nextitnet.py +69 -0
  1303. topologies-0.0.1/src/topologies/models/ngcf.py +60 -0
  1304. topologies-0.0.1/src/topologies/models/nhits.py +68 -0
  1305. topologies-0.0.1/src/topologies/models/nice.py +33 -0
  1306. topologies-0.0.1/src/topologies/models/noam.py +32 -0
  1307. topologies-0.0.1/src/topologies/models/node.py +14 -0
  1308. topologies-0.0.1/src/topologies/models/node_ensemble.py +62 -0
  1309. topologies-0.0.1/src/topologies/models/noisy_nets.py +54 -0
  1310. topologies-0.0.1/src/topologies/models/nvae.py +50 -0
  1311. topologies-0.0.1/src/topologies/models/ot_cfm.py +103 -0
  1312. topologies-0.0.1/src/topologies/models/pairre.py +27 -0
  1313. topologies-0.0.1/src/topologies/models/panoptic.py +68 -0
  1314. topologies-0.0.1/src/topologies/models/phi.py +186 -0
  1315. topologies-0.0.1/src/topologies/models/pinn.py +29 -0
  1316. topologies-0.0.1/src/topologies/models/pinsage.py +55 -0
  1317. topologies-0.0.1/src/topologies/models/pix2pix.py +68 -0
  1318. topologies-0.0.1/src/topologies/models/pixart.py +42 -0
  1319. topologies-0.0.1/src/topologies/models/pixelcnn.py +26 -0
  1320. topologies-0.0.1/src/topologies/models/pna.py +56 -0
  1321. topologies-0.0.1/src/topologies/models/pnn.py +42 -0
  1322. topologies-0.0.1/src/topologies/models/point_cloud.py +57 -0
  1323. topologies-0.0.1/src/topologies/models/point_transformer.py +45 -0
  1324. topologies-0.0.1/src/topologies/models/point_transformer_v2.py +67 -0
  1325. topologies-0.0.1/src/topologies/models/pointpillars.py +78 -0
  1326. topologies-0.0.1/src/topologies/models/poolformer.py +49 -0
  1327. topologies-0.0.1/src/topologies/models/pos2d.py +51 -0
  1328. topologies-0.0.1/src/topologies/models/pose.py +62 -0
  1329. topologies-0.0.1/src/topologies/models/ppo_clip.py +18 -0
  1330. topologies-0.0.1/src/topologies/models/progan.py +39 -0
  1331. topologies-0.0.1/src/topologies/models/prompt_seg.py +71 -0
  1332. topologies-0.0.1/src/topologies/models/q_network.py +23 -0
  1333. topologies-0.0.1/src/topologies/models/qmix.py +58 -0
  1334. topologies-0.0.1/src/topologies/models/qr_dqn.py +56 -0
  1335. topologies-0.0.1/src/topologies/models/qwen.py +136 -0
  1336. topologies-0.0.1/src/topologies/models/rag_fid.py +90 -0
  1337. topologies-0.0.1/src/topologies/models/realnvp.py +84 -0
  1338. topologies-0.0.1/src/topologies/models/recurrent_models.py +102 -0
  1339. topologies-0.0.1/src/topologies/models/regnet.py +75 -0
  1340. topologies-0.0.1/src/topologies/models/resmlp.py +72 -0
  1341. topologies-0.0.1/src/topologies/models/resnet.py +80 -0
  1342. topologies-0.0.1/src/topologies/models/resnet_v1d.py +79 -0
  1343. topologies-0.0.1/src/topologies/models/resnet_variants.py +102 -0
  1344. topologies-0.0.1/src/topologies/models/retinanet.py +120 -0
  1345. topologies-0.0.1/src/topologies/models/retnet.py +111 -0
  1346. topologies-0.0.1/src/topologies/models/retro.py +90 -0
  1347. topologies-0.0.1/src/topologies/models/reversible_residual.py +18 -0
  1348. topologies-0.0.1/src/topologies/models/rgcn.py +139 -0
  1349. topologies-0.0.1/src/topologies/models/rloo.py +32 -0
  1350. topologies-0.0.1/src/topologies/models/rnd.py +19 -0
  1351. topologies-0.0.1/src/topologies/models/rq_transformer.py +56 -0
  1352. topologies-0.0.1/src/topologies/models/rq_vae.py +46 -0
  1353. topologies-0.0.1/src/topologies/models/rwkv.py +132 -0
  1354. topologies-0.0.1/src/topologies/models/s4.py +99 -0
  1355. topologies-0.0.1/src/topologies/models/sac.py +98 -0
  1356. topologies-0.0.1/src/topologies/models/sagan.py +37 -0
  1357. topologies-0.0.1/src/topologies/models/sagpool.py +35 -0
  1358. topologies-0.0.1/src/topologies/models/sasrec.py +59 -0
  1359. topologies-0.0.1/src/topologies/models/segment.py +77 -0
  1360. topologies-0.0.1/src/topologies/models/sentence_bert.py +21 -0
  1361. topologies-0.0.1/src/topologies/models/sentence_encoder.py +38 -0
  1362. topologies-0.0.1/src/topologies/models/sepformer.py +97 -0
  1363. topologies-0.0.1/src/topologies/models/seq2seq.py +169 -0
  1364. topologies-0.0.1/src/topologies/models/set_detect.py +72 -0
  1365. topologies-0.0.1/src/topologies/models/setr.py +36 -0
  1366. topologies-0.0.1/src/topologies/models/sgc.py +39 -0
  1367. topologies-0.0.1/src/topologies/models/sgc_precompute.py +30 -0
  1368. topologies-0.0.1/src/topologies/models/siglip.py +35 -0
  1369. topologies-0.0.1/src/topologies/models/sign.py +28 -0
  1370. topologies-0.0.1/src/topologies/models/sim.py +56 -0
  1371. topologies-0.0.1/src/topologies/models/simclr.py +32 -0
  1372. topologies-0.0.1/src/topologies/models/sincnet.py +54 -0
  1373. topologies-0.0.1/src/topologies/models/single_stage_detect.py +79 -0
  1374. topologies-0.0.1/src/topologies/models/sit.py +31 -0
  1375. topologies-0.0.1/src/topologies/models/sngan.py +30 -0
  1376. topologies-0.0.1/src/topologies/models/solov2.py +38 -0
  1377. topologies-0.0.1/src/topologies/models/sortpool.py +49 -0
  1378. topologies-0.0.1/src/topologies/models/soundstream.py +33 -0
  1379. topologies-0.0.1/src/topologies/models/spade.py +36 -0
  1380. topologies-0.0.1/src/topologies/models/spanbert.py +295 -0
  1381. topologies-0.0.1/src/topologies/models/sr_gnn.py +88 -0
  1382. topologies-0.0.1/src/topologies/models/static_embeddings.py +68 -0
  1383. topologies-0.0.1/src/topologies/models/stylegan3.py +26 -0
  1384. topologies-0.0.1/src/topologies/models/swin.py +99 -0
  1385. topologies-0.0.1/src/topologies/models/swin_unet.py +146 -0
  1386. topologies-0.0.1/src/topologies/models/t5.py +162 -0
  1387. topologies-0.0.1/src/topologies/models/tabtransformer.py +68 -0
  1388. topologies-0.0.1/src/topologies/models/tabular_mlp.py +104 -0
  1389. topologies-0.0.1/src/topologies/models/tacotron2.py +64 -0
  1390. topologies-0.0.1/src/topologies/models/tc_resnet.py +62 -0
  1391. topologies-0.0.1/src/topologies/models/tcvae.py +51 -0
  1392. topologies-0.0.1/src/topologies/models/td3.py +76 -0
  1393. topologies-0.0.1/src/topologies/models/td3_bc.py +44 -0
  1394. topologies-0.0.1/src/topologies/models/temporal_convnet.py +53 -0
  1395. topologies-0.0.1/src/topologies/models/text_cnn.py +57 -0
  1396. topologies-0.0.1/src/topologies/models/tf_gridnet.py +27 -0
  1397. topologies-0.0.1/src/topologies/models/tft.py +290 -0
  1398. topologies-0.0.1/src/topologies/models/token_mixer.py +73 -0
  1399. topologies-0.0.1/src/topologies/models/transformer_models.py +174 -0
  1400. topologies-0.0.1/src/topologies/models/transformer_xl.py +142 -0
  1401. topologies-0.0.1/src/topologies/models/transitional_text.py +97 -0
  1402. topologies-0.0.1/src/topologies/models/tsm.py +39 -0
  1403. topologies-0.0.1/src/topologies/models/two_stage_detect.py +84 -0
  1404. topologies-0.0.1/src/topologies/models/u_vit.py +52 -0
  1405. topologies-0.0.1/src/topologies/models/ul2.py +204 -0
  1406. topologies-0.0.1/src/topologies/models/unet.py +54 -0
  1407. topologies-0.0.1/src/topologies/models/unipc.py +62 -0
  1408. topologies-0.0.1/src/topologies/models/v_pred.py +24 -0
  1409. topologies-0.0.1/src/topologies/models/vae_mlp.py +67 -0
  1410. topologies-0.0.1/src/topologies/models/variational_autoencoders.py +129 -0
  1411. topologies-0.0.1/src/topologies/models/vgg.py +60 -0
  1412. topologies-0.0.1/src/topologies/models/video_conv.py +57 -0
  1413. topologies-0.0.1/src/topologies/models/video_ssl.py +51 -0
  1414. topologies-0.0.1/src/topologies/models/video_transformer.py +66 -0
  1415. topologies-0.0.1/src/topologies/models/vision_neck.py +64 -0
  1416. topologies-0.0.1/src/topologies/models/vision_norm.py +76 -0
  1417. topologies-0.0.1/src/topologies/models/vision_ssl.py +93 -0
  1418. topologies-0.0.1/src/topologies/models/vision_transformer.py +75 -0
  1419. topologies-0.0.1/src/topologies/models/vit_22b.py +23 -0
  1420. topologies-0.0.1/src/topologies/models/vit_encoder.py +88 -0
  1421. topologies-0.0.1/src/topologies/models/volume_seg.py +55 -0
  1422. topologies-0.0.1/src/topologies/models/voxel_detect.py +53 -0
  1423. topologies-0.0.1/src/topologies/models/vqvae.py +70 -0
  1424. topologies-0.0.1/src/topologies/models/wav2vec2.py +29 -0
  1425. topologies-0.0.1/src/topologies/models/wavenet.py +80 -0
  1426. topologies-0.0.1/src/topologies/models/wgan.py +32 -0
  1427. topologies-0.0.1/src/topologies/models/wgan_gp.py +58 -0
  1428. topologies-0.0.1/src/topologies/models/wide_and_deep.py +31 -0
  1429. topologies-0.0.1/src/topologies/models/world_model.py +73 -0
  1430. topologies-0.0.1/src/topologies/models/x3d.py +168 -0
  1431. topologies-0.0.1/src/topologies/models/x_vector.py +27 -0
  1432. topologies-0.0.1/src/topologies/models/xdeepfm.py +49 -0
  1433. topologies-0.0.1/src/topologies/models/xlnet.py +261 -0
  1434. topologies-0.0.1/src/topologies/models/xlstm.py +98 -0
  1435. topologies-0.0.1/src/topologies/models/xmem.py +500 -0
  1436. topologies-0.0.1/src/topologies/models/xtts.py +40 -0
  1437. topologies-0.0.1/src/topologies/models/yolov3.py +45 -0
  1438. topologies-0.0.1/src/topologies/models/yourtts.py +21 -0
  1439. topologies-0.0.1/src/topologies/nn/__init__.py +73 -0
  1440. topologies-0.0.1/src/topologies/nn/aux.py +51 -0
  1441. topologies-0.0.1/src/topologies/nn/causal.py +401 -0
  1442. topologies-0.0.1/src/topologies/nn/layers.py +317 -0
  1443. topologies-0.0.1/src/topologies/nn/mixers.py +246 -0
  1444. topologies-0.0.1/src/topologies/nn/models.py +552 -0
  1445. topologies-0.0.1/src/topologies/nn/slots.py +230 -0
  1446. topologies-0.0.1/src/topologies/optimization/__init__.py +95 -0
  1447. topologies-0.0.1/src/topologies/optimization/_torch_ao.py +50 -0
  1448. topologies-0.0.1/src/topologies/optimization/artifacts.py +396 -0
  1449. topologies-0.0.1/src/topologies/optimization/contracts.py +90 -0
  1450. topologies-0.0.1/src/topologies/optimization/distillation.py +632 -0
  1451. topologies-0.0.1/src/topologies/optimization/integrations.py +256 -0
  1452. topologies-0.0.1/src/topologies/optimization/pruning.py +416 -0
  1453. topologies-0.0.1/src/topologies/optimization/quantization.py +1261 -0
  1454. topologies-0.0.1/src/topologies/optimization/recommendations.py +165 -0
  1455. topologies-0.0.1/src/topologies/optimization/reporting.py +91 -0
  1456. topologies-0.0.1/src/topologies/optimization/torchao_pt2e.py +781 -0
  1457. topologies-0.0.1/src/topologies/optimization/workflow.py +368 -0
  1458. topologies-0.0.1/src/topologies/optimizers/catalog.json +136 -0
  1459. topologies-0.0.1/src/topologies/pretrained.py +278 -0
  1460. topologies-0.0.1/src/topologies/registry/__init__.py +6 -0
  1461. topologies-0.0.1/src/topologies/registry/api.py +59 -0
  1462. topologies-0.0.1/src/topologies/registry/index.py +165 -0
  1463. topologies-0.0.1/src/topologies/registry/loader.py +75 -0
  1464. topologies-0.0.1/src/topologies/registry/validate.py +470 -0
  1465. topologies-0.0.1/src/topologies/synth/__init__.py +178 -0
  1466. topologies-0.0.1/src/topologies/synth/attributes.py +570 -0
  1467. topologies-0.0.1/src/topologies/synth/blueprints/__init__.py +65 -0
  1468. topologies-0.0.1/src/topologies/synth/blueprints/_base.py +126 -0
  1469. topologies-0.0.1/src/topologies/synth/blueprints/algorithmic.py +312 -0
  1470. topologies-0.0.1/src/topologies/synth/blueprints/reasoning.py +42 -0
  1471. topologies-0.0.1/src/topologies/synth/blueprints/structured.py +240 -0
  1472. topologies-0.0.1/src/topologies/synth/blueprints/vision.py +88 -0
  1473. topologies-0.0.1/src/topologies/synth/cli.py +501 -0
  1474. topologies-0.0.1/src/topologies/synth/codecs.py +215 -0
  1475. topologies-0.0.1/src/topologies/synth/core.py +1331 -0
  1476. topologies-0.0.1/src/topologies/synth/counterfactuals.py +162 -0
  1477. topologies-0.0.1/src/topologies/synth/ctor.py +110 -0
  1478. topologies-0.0.1/src/topologies/synth/dataset.py +433 -0
  1479. topologies-0.0.1/src/topologies/synth/dists.py +461 -0
  1480. topologies-0.0.1/src/topologies/synth/dsp.py +352 -0
  1481. topologies-0.0.1/src/topologies/synth/errors.py +68 -0
  1482. topologies-0.0.1/src/topologies/synth/evaluate.py +659 -0
  1483. topologies-0.0.1/src/topologies/synth/export.py +150 -0
  1484. topologies-0.0.1/src/topologies/synth/fingerprint.py +124 -0
  1485. topologies-0.0.1/src/topologies/synth/generators/__init__.py +0 -0
  1486. topologies-0.0.1/src/topologies/synth/generators/augment.py +12 -0
  1487. topologies-0.0.1/src/topologies/synth/generators/base.py +200 -0
  1488. topologies-0.0.1/src/topologies/synth/generators/grammar.py +5 -0
  1489. topologies-0.0.1/src/topologies/synth/generators/program.py +9 -0
  1490. topologies-0.0.1/src/topologies/synth/introspect.py +107 -0
  1491. topologies-0.0.1/src/topologies/synth/jsonutil.py +241 -0
  1492. topologies-0.0.1/src/topologies/synth/langid.py +150 -0
  1493. topologies-0.0.1/src/topologies/synth/lib/__init__.py +105 -0
  1494. topologies-0.0.1/src/topologies/synth/lib/audio/__init__.py +768 -0
  1495. topologies-0.0.1/src/topologies/synth/lib/graph/__init__.py +340 -0
  1496. topologies-0.0.1/src/topologies/synth/lib/image/__init__.py +762 -0
  1497. topologies-0.0.1/src/topologies/synth/lib/math/__init__.py +175 -0
  1498. topologies-0.0.1/src/topologies/synth/lib/math/expr.py +191 -0
  1499. topologies-0.0.1/src/topologies/synth/lib/process/__init__.py +231 -0
  1500. topologies-0.0.1/src/topologies/synth/lib/random/__init__.py +240 -0
  1501. topologies-0.0.1/src/topologies/synth/lib/seq/__init__.py +418 -0
  1502. topologies-0.0.1/src/topologies/synth/lib/series/__init__.py +268 -0
  1503. topologies-0.0.1/src/topologies/synth/lib/tabular/__init__.py +733 -0
  1504. topologies-0.0.1/src/topologies/synth/lib/text/__init__.py +53 -0
  1505. topologies-0.0.1/src/topologies/synth/lib/text/faker.py +227 -0
  1506. topologies-0.0.1/src/topologies/synth/lib/text/formats.py +165 -0
  1507. topologies-0.0.1/src/topologies/synth/lib/text/generate.py +346 -0
  1508. topologies-0.0.1/src/topologies/synth/lib/text/grammar.py +189 -0
  1509. topologies-0.0.1/src/topologies/synth/lib/text/legacy.py +118 -0
  1510. topologies-0.0.1/src/topologies/synth/lib/text/noise.py +357 -0
  1511. topologies-0.0.1/src/topologies/synth/lib/text/numbers.py +115 -0
  1512. topologies-0.0.1/src/topologies/synth/lib/text/packs/de/faker.yaml +143 -0
  1513. topologies-0.0.1/src/topologies/synth/lib/text/packs/de/keyboard.yaml +2 -0
  1514. topologies-0.0.1/src/topologies/synth/lib/text/packs/de/translit.yaml +6 -0
  1515. topologies-0.0.1/src/topologies/synth/lib/text/packs/en/faker.yaml +174 -0
  1516. topologies-0.0.1/src/topologies/synth/lib/text/packs/en/keyboard.yaml +3 -0
  1517. topologies-0.0.1/src/topologies/synth/lib/text/packs/en/lexicon.yaml +4 -0
  1518. topologies-0.0.1/src/topologies/synth/lib/text/packs/en/numbers.yaml +34 -0
  1519. topologies-0.0.1/src/topologies/synth/lib/text/packs/en/paradigms.yaml +1 -0
  1520. topologies-0.0.1/src/topologies/synth/lib/text/packs/en/pools.yaml +7 -0
  1521. topologies-0.0.1/src/topologies/synth/lib/text/packs/en/translit.yaml +1 -0
  1522. topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/faker.yaml +121 -0
  1523. topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/keyboard.yaml +3 -0
  1524. topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/lexicon.yaml +17 -0
  1525. topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/numbers.yaml +54 -0
  1526. topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/paradigms.yaml +20 -0
  1527. topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/pools.yaml +7 -0
  1528. topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/translit.yaml +37 -0
  1529. topologies-0.0.1/src/topologies/synth/lib/text/packs.py +122 -0
  1530. topologies-0.0.1/src/topologies/synth/lib/text/pools.py +58 -0
  1531. topologies-0.0.1/src/topologies/synth/lib/text/slots.py +94 -0
  1532. topologies-0.0.1/src/topologies/synth/lib/text/template.py +357 -0
  1533. topologies-0.0.1/src/topologies/synth/limits.py +83 -0
  1534. topologies-0.0.1/src/topologies/synth/loop.py +390 -0
  1535. topologies-0.0.1/src/topologies/synth/ops/__init__.py +66 -0
  1536. topologies-0.0.1/src/topologies/synth/ops/_callable.py +29 -0
  1537. topologies-0.0.1/src/topologies/synth/ops/augment.py +171 -0
  1538. topologies-0.0.1/src/topologies/synth/ops/balance.py +154 -0
  1539. topologies-0.0.1/src/topologies/synth/ops/base.py +235 -0
  1540. topologies-0.0.1/src/topologies/synth/ops/contamination.py +191 -0
  1541. topologies-0.0.1/src/topologies/synth/ops/dedup.py +402 -0
  1542. topologies-0.0.1/src/topologies/synth/ops/filters.py +186 -0
  1543. topologies-0.0.1/src/topologies/synth/ops/label.py +230 -0
  1544. topologies-0.0.1/src/topologies/synth/ops/parse.py +118 -0
  1545. topologies-0.0.1/src/topologies/synth/ops/split.py +217 -0
  1546. topologies-0.0.1/src/topologies/synth/ops/validate.py +95 -0
  1547. topologies-0.0.1/src/topologies/synth/ops/verify.py +530 -0
  1548. topologies-0.0.1/src/topologies/synth/overrides.py +225 -0
  1549. topologies-0.0.1/src/topologies/synth/pipeline.py +1054 -0
  1550. topologies-0.0.1/src/topologies/synth/planning.py +243 -0
  1551. topologies-0.0.1/src/topologies/synth/pools.py +19 -0
  1552. topologies-0.0.1/src/topologies/synth/previews.py +187 -0
  1553. topologies-0.0.1/src/topologies/synth/quality/__init__.py +34 -0
  1554. topologies-0.0.1/src/topologies/synth/quality/embed.py +87 -0
  1555. topologies-0.0.1/src/topologies/synth/quality/fidelity.py +421 -0
  1556. topologies-0.0.1/src/topologies/synth/quality/insights.py +202 -0
  1557. topologies-0.0.1/src/topologies/synth/quality/intrinsic.py +406 -0
  1558. topologies-0.0.1/src/topologies/synth/quality/privacy.py +85 -0
  1559. topologies-0.0.1/src/topologies/synth/quality/report.py +356 -0
  1560. topologies-0.0.1/src/topologies/synth/quality/slices.py +118 -0
  1561. topologies-0.0.1/src/topologies/synth/quality/utility.py +363 -0
  1562. topologies-0.0.1/src/topologies/synth/records.py +171 -0
  1563. topologies-0.0.1/src/topologies/synth/report.py +195 -0
  1564. topologies-0.0.1/src/topologies/synth/reviewing.py +332 -0
  1565. topologies-0.0.1/src/topologies/synth/rng.py +225 -0
  1566. topologies-0.0.1/src/topologies/synth/runtime.py +33 -0
  1567. topologies-0.0.1/src/topologies/synth/sampling.py +613 -0
  1568. topologies-0.0.1/src/topologies/synth/schema.py +268 -0
  1569. topologies-0.0.1/src/topologies/synth/spec.py +366 -0
  1570. topologies-0.0.1/src/topologies/synth/store.py +374 -0
  1571. topologies-0.0.1/src/topologies/synth/streaming.py +241 -0
  1572. topologies-0.0.1/src/topologies/synth/tasks.py +717 -0
  1573. topologies-0.0.1/src/topologies/synth/textutil.py +287 -0
  1574. topologies-0.0.1/src/topologies/synth/torch_views.py +1157 -0
  1575. topologies-0.0.1/src/topologies/synth/valuetypes.py +1741 -0
  1576. topologies-0.0.1/src/topologies/synth/verdict.py +74 -0
  1577. topologies-0.0.1/src/topologies/synth/versions.py +218 -0
  1578. topologies-0.0.1/src/topologies/synth/yaml_spec.py +600 -0
  1579. topologies-0.0.1/src/topologies/taxonomy/__init__.py +8 -0
  1580. topologies-0.0.1/src/topologies/taxonomy/validate.py +74 -0
  1581. topologies-0.0.1/src/topologies/tokenizers/__init__.py +78 -0
  1582. topologies-0.0.1/src/topologies/tokenizers/base.py +350 -0
  1583. topologies-0.0.1/src/topologies/tokenizers/bpe.py +190 -0
  1584. topologies-0.0.1/src/topologies/tokenizers/hf.py +157 -0
  1585. topologies-0.0.1/src/topologies/tokenizers/simple.py +139 -0
  1586. topologies-0.0.1/src/topologies/tokenizers/stats.py +86 -0
  1587. topologies-0.0.1/src/topologies/training/__init__.py +88 -0
  1588. topologies-0.0.1/src/topologies/training/_checkpointing.py +232 -0
  1589. topologies-0.0.1/src/topologies/training/_loops.py +215 -0
  1590. topologies-0.0.1/src/topologies/training/_observability.py +148 -0
  1591. topologies-0.0.1/src/topologies/training/_optim_factory.py +341 -0
  1592. topologies-0.0.1/src/topologies/training/_run_info.py +93 -0
  1593. topologies-0.0.1/src/topologies/training/_step_metrics.py +128 -0
  1594. topologies-0.0.1/src/topologies/training/_steps.py +358 -0
  1595. topologies-0.0.1/src/topologies/training/_trainer_config.py +50 -0
  1596. topologies-0.0.1/src/topologies/training/adapters.py +207 -0
  1597. topologies-0.0.1/src/topologies/training/aux.py +62 -0
  1598. topologies-0.0.1/src/topologies/training/bart_data.py +124 -0
  1599. topologies-0.0.1/src/topologies/training/byt5_data.py +99 -0
  1600. topologies-0.0.1/src/topologies/training/callbacks.py +73 -0
  1601. topologies-0.0.1/src/topologies/training/canine_data.py +69 -0
  1602. topologies-0.0.1/src/topologies/training/charformer_data.py +65 -0
  1603. topologies-0.0.1/src/topologies/training/checkpoint.py +207 -0
  1604. topologies-0.0.1/src/topologies/training/components.py +106 -0
  1605. topologies-0.0.1/src/topologies/training/configs.py +309 -0
  1606. topologies-0.0.1/src/topologies/training/data.py +203 -0
  1607. topologies-0.0.1/src/topologies/training/device.py +103 -0
  1608. topologies-0.0.1/src/topologies/training/distributed.py +165 -0
  1609. topologies-0.0.1/src/topologies/training/engine.py +1085 -0
  1610. topologies-0.0.1/src/topologies/training/errors.py +61 -0
  1611. topologies-0.0.1/src/topologies/training/introspect.py +19 -0
  1612. topologies-0.0.1/src/topologies/training/logging/__init__.py +21 -0
  1613. topologies-0.0.1/src/topologies/training/logging/base.py +82 -0
  1614. topologies-0.0.1/src/topologies/training/logging/factory.py +69 -0
  1615. topologies-0.0.1/src/topologies/training/logging/jsonl.py +210 -0
  1616. topologies-0.0.1/src/topologies/training/logging/tabular.py +125 -0
  1617. topologies-0.0.1/src/topologies/training/logging/text.py +137 -0
  1618. topologies-0.0.1/src/topologies/training/logging_.py +23 -0
  1619. topologies-0.0.1/src/topologies/training/losses.py +218 -0
  1620. topologies-0.0.1/src/topologies/training/marian_data.py +51 -0
  1621. topologies-0.0.1/src/topologies/training/model_recipes.json +1331 -0
  1622. topologies-0.0.1/src/topologies/training/monitoring/__init__.py +15 -0
  1623. topologies-0.0.1/src/topologies/training/monitoring/base.py +55 -0
  1624. topologies-0.0.1/src/topologies/training/monitoring/dashboard/__init__.py +17 -0
  1625. topologies-0.0.1/src/topologies/training/monitoring/dashboard/app.py +87 -0
  1626. topologies-0.0.1/src/topologies/training/monitoring/dashboard/charts.py +50 -0
  1627. topologies-0.0.1/src/topologies/training/monitoring/dashboard/state.py +202 -0
  1628. topologies-0.0.1/src/topologies/training/monitoring/dashboard/views.py +36 -0
  1629. topologies-0.0.1/src/topologies/training/monitoring/dashboard/widgets.py +64 -0
  1630. topologies-0.0.1/src/topologies/training/monitoring/factory.py +88 -0
  1631. topologies-0.0.1/src/topologies/training/monitoring/standard.py +264 -0
  1632. topologies-0.0.1/src/topologies/training/monitoring/system.py +112 -0
  1633. topologies-0.0.1/src/topologies/training/recipe_base.py +447 -0
  1634. topologies-0.0.1/src/topologies/training/recipes/__init__.py +30 -0
  1635. topologies-0.0.1/src/topologies/training/recipes/audio.py +88 -0
  1636. topologies-0.0.1/src/topologies/training/recipes/autoenc.py +124 -0
  1637. topologies-0.0.1/src/topologies/training/recipes/canine.py +47 -0
  1638. topologies-0.0.1/src/topologies/training/recipes/classify.py +440 -0
  1639. topologies-0.0.1/src/topologies/training/recipes/contrastive.py +83 -0
  1640. topologies-0.0.1/src/topologies/training/recipes/density.py +50 -0
  1641. topologies-0.0.1/src/topologies/training/recipes/detection.py +118 -0
  1642. topologies-0.0.1/src/topologies/training/recipes/diffusion.py +88 -0
  1643. topologies-0.0.1/src/topologies/training/recipes/distillation.py +128 -0
  1644. topologies-0.0.1/src/topologies/training/recipes/gan.py +142 -0
  1645. topologies-0.0.1/src/topologies/training/recipes/graph.py +29 -0
  1646. topologies-0.0.1/src/topologies/training/recipes/lm.py +935 -0
  1647. topologies-0.0.1/src/topologies/training/recipes/mixtral.py +66 -0
  1648. topologies-0.0.1/src/topologies/training/recipes/permutation_lm.py +134 -0
  1649. topologies-0.0.1/src/topologies/training/recipes/ppo.py +233 -0
  1650. topologies-0.0.1/src/topologies/training/recipes/recommender.py +104 -0
  1651. topologies-0.0.1/src/topologies/training/recipes/router_probe.py +63 -0
  1652. topologies-0.0.1/src/topologies/training/recipes/segmentation.py +76 -0
  1653. topologies-0.0.1/src/topologies/training/recipes/spanbert.py +125 -0
  1654. topologies-0.0.1/src/topologies/training/recipes/speech.py +35 -0
  1655. topologies-0.0.1/src/topologies/training/recipes/static.py +28 -0
  1656. topologies-0.0.1/src/topologies/training/recipes/structured_objective.py +64 -0
  1657. topologies-0.0.1/src/topologies/training/recipes/transformer_xl.py +76 -0
  1658. topologies-0.0.1/src/topologies/training/recipes/tts.py +86 -0
  1659. topologies-0.0.1/src/topologies/training/recipes/value_learning.py +39 -0
  1660. topologies-0.0.1/src/topologies/training/recipes/vision.py +555 -0
  1661. topologies-0.0.1/src/topologies/training/registries.py +573 -0
  1662. topologies-0.0.1/src/topologies/training/reporting/__init__.py +35 -0
  1663. topologies-0.0.1/src/topologies/training/reporting/events.py +210 -0
  1664. topologies-0.0.1/src/topologies/training/reporting/reporter.py +204 -0
  1665. topologies-0.0.1/src/topologies/training/resolve.py +248 -0
  1666. topologies-0.0.1/src/topologies/training/schema.py +4 -0
  1667. topologies-0.0.1/src/topologies/training/stability/activation_offload.py +6 -0
  1668. topologies-0.0.1/src/topologies/training/stability/bf16_recipe.py +6 -0
  1669. topologies-0.0.1/src/topologies/training/stability/compile_parity.py +6 -0
  1670. topologies-0.0.1/src/topologies/training/stability/fsdp_wrap.py +6 -0
  1671. topologies-0.0.1/src/topologies/training/stability/grad_checkpoint.py +16 -0
  1672. topologies-0.0.1/src/topologies/training/stability/grad_clip.py +57 -0
  1673. topologies-0.0.1/src/topologies/training/stability/llrd.py +8 -0
  1674. topologies-0.0.1/src/topologies/training/stability/loss_scale.py +6 -0
  1675. topologies-0.0.1/src/topologies/training/stability/nan_skip.py +8 -0
  1676. topologies-0.0.1/src/topologies/training/stability/zero_partition.py +6 -0
  1677. topologies-0.0.1/src/topologies/training/t5_data.py +90 -0
  1678. topologies-0.0.1/src/topologies/training/ul2_data.py +62 -0
  1679. topologies-0.0.1/src/topologies/training-artifacts.lock.json +302 -0
  1680. topologies-0.0.1/src/topologies/training_schema.py +99 -0
  1681. topologies-0.0.1/src/topologies/utils/__init__.py +8 -0
  1682. topologies-0.0.1/src/topologies/utils/locking.py +80 -0
  1683. topologies-0.0.1/src/topologies/utils/logging.py +40 -0
  1684. topologies-0.0.1/src/topologies/validation/__init__.py +4 -0
  1685. topologies-0.0.1/src/topologies/validation/catalog_specs.json +953 -0
  1686. topologies-0.0.1/src/topologies/validation/catalog_specs.py +65 -0
  1687. topologies-0.0.1/src/topologies/validation/catalog_workflows.py +1854 -0
  1688. topologies-0.0.1/src/topologies/validation/component_workflows.py +588 -0
  1689. topologies-0.0.1/src/topologies/validation/framework.py +151 -0
  1690. topologies-0.0.1/src/topologies/validation/gate.py +87 -0
  1691. topologies-0.0.1/src/topologies/validation/implementation.py +35 -0
  1692. topologies-0.0.1/src/topologies/validation/source_style.py +23 -0
  1693. topologies-0.0.1/src/topologies/visualization/__init__.py +103 -0
  1694. topologies-0.0.1/src/topologies/visualization/adapters.py +225 -0
  1695. topologies-0.0.1/src/topologies/visualization/core.py +872 -0
  1696. topologies-0.0.1/src/topologies/visualization/experiment_views.py +765 -0
  1697. topologies-0.0.1/src/topologies/visualization/format.py +322 -0
  1698. topologies-0.0.1/src/topologies/visualization/live.py +223 -0
  1699. topologies-0.0.1/src/topologies/visualization/matplotlib/__init__.py +498 -0
  1700. topologies-0.0.1/src/topologies/visualization/matplotlib/live.py +247 -0
  1701. topologies-0.0.1/src/topologies/visualization/matplotlib/style.py +262 -0
  1702. topologies-0.0.1/src/topologies/visualization/matplotlib/views/__init__.py +84 -0
  1703. topologies-0.0.1/src/topologies/visualization/matplotlib/views/analysis.py +182 -0
  1704. topologies-0.0.1/src/topologies/visualization/matplotlib/views/engineering.py +206 -0
  1705. topologies-0.0.1/src/topologies/visualization/matplotlib/views/experiments.py +344 -0
  1706. topologies-0.0.1/src/topologies/visualization/matplotlib/views/generic.py +87 -0
  1707. topologies-0.0.1/src/topologies/visualization/matplotlib/views/hardware.py +64 -0
  1708. topologies-0.0.1/src/topologies/visualization/matplotlib/views/profile.py +77 -0
  1709. topologies-0.0.1/src/topologies/visualization/matplotlib/views/training.py +286 -0
  1710. topologies-0.0.1/src/topologies/visualization/matplotlib_views.py +325 -0
  1711. topologies-0.0.1/src/topologies/visualization/state.py +989 -0
  1712. topologies-0.0.1/src/topologies/visualization/terminal/__init__.py +133 -0
  1713. topologies-0.0.1/src/topologies/visualization/terminal/caps.py +195 -0
  1714. topologies-0.0.1/src/topologies/visualization/terminal/driver.py +356 -0
  1715. topologies-0.0.1/src/topologies/visualization/terminal/glyphs.py +102 -0
  1716. topologies-0.0.1/src/topologies/visualization/terminal/keys.py +146 -0
  1717. topologies-0.0.1/src/topologies/visualization/terminal/layout.py +162 -0
  1718. topologies-0.0.1/src/topologies/visualization/terminal/panels.py +128 -0
  1719. topologies-0.0.1/src/topologies/visualization/terminal/progress.py +76 -0
  1720. topologies-0.0.1/src/topologies/visualization/terminal/screen.py +503 -0
  1721. topologies-0.0.1/src/topologies/visualization/terminal/text.py +401 -0
  1722. topologies-0.0.1/src/topologies/visualization/terminal/widgets.py +967 -0
  1723. topologies-0.0.1/src/topologies/visualization/testing.py +827 -0
  1724. topologies-0.0.1/src/topologies/visualization/theme.py +252 -0
  1725. topologies-0.0.1/src/topologies/visualization/training_views.py +1280 -0
  1726. topologies-0.0.1/src/topologies/visualization/views.py +143 -0
@@ -0,0 +1,20 @@
1
+ __pycache__/
2
+ *.py[cod]
3
+ *.egg-info/
4
+ /dist/
5
+ /build/
6
+ .venv/
7
+ venv/
8
+ .env
9
+ .pytest_cache/
10
+ .ruff_cache/
11
+ .DS_Store
12
+ /artifacts/
13
+ /runs/
14
+ /experiments/
15
+ /data/
16
+ /docs/
17
+ /benchmarks/
18
+ /examples/
19
+ /scripts/
20
+ /tests/
@@ -0,0 +1,41 @@
1
+ # Changelog
2
+
3
+ All notable changes to `topologies` are listed here.
4
+
5
+ ## 0.0.1 — 2026-10-20
6
+
7
+ First public release: the training engine, data utilities, tokenizers, model constructor,
8
+ generation, evaluation, experiments and introspection, plus the experimental catalog of
9
+ reference implementations in `topologies.models` and `topologies.layers`.
10
+
11
+ ### Added
12
+ - **Models**: `xmem` (XMem) ([paper](https://arxiv.org/abs/2207.07115))
13
+ - **Models**: `consistency-model` (Consistency Models) ([paper](https://arxiv.org/abs/2303.01469))
14
+ - **Models**: `distributional-q` (A Distributional Perspective on Reinforcement Learning) ([paper](https://arxiv.org/abs/1707.06887))
15
+ - **Models**: `biggan` (Large Scale GAN Training for High Fidelity Natural Image Synthesis) ([paper](https://arxiv.org/abs/1809.11096))
16
+ - **Models**: `dqn-family` (Playing Atari with Deep Reinforcement Learning) ([paper](https://arxiv.org/abs/1312.5602))
17
+ - **Models**: `latent-diffusion` (High-Resolution Image Synthesis with Latent Diffusion Models) ([paper](https://arxiv.org/abs/2112.10752))
18
+ - **Models**: `rgcn` (R-GCN) ([paper](https://arxiv.org/abs/1703.06103))
19
+ - **Models**: `graphsage` (GraphSAGE Mean Aggregator) ([paper](https://arxiv.org/abs/1706.02216))
20
+ - **Models**: `qmix` (QMIX: Monotonic Value Function Factorisation for Deep Multi-Agent Reinforcement Learning) ([paper](https://arxiv.org/abs/1803.11485))
21
+ - **Models**: `lightgcn` (LightGCN) ([paper](https://arxiv.org/abs/2002.02126))
22
+ - **Models**: `tft` (TFT) ([paper](https://arxiv.org/abs/1912.09363))
23
+ - **Models**: `dcrnn` (DCRNN diffusion-GRU sequence primitive) ([paper](https://arxiv.org/abs/1707.01926))
24
+ - **Components**: `cbam-gate` (CBAM channel-and-spatial attention block) ([paper](https://arxiv.org/abs/1807.06521))
25
+ - **Components**: `topk-moe-layer` (Deterministic top-k MoE core; excludes noisy gating and balance losses) ([paper](https://arxiv.org/abs/1701.06538))
26
+ - **Components**: `meshcnn` (MeshCNN: edge convolution and collapse pooling) ([paper](https://arxiv.org/abs/1809.05910))
27
+ - **Components**: `crestereo` (CREStereo local correlation/disparity-offset primitive) ([paper](https://arxiv.org/abs/2203.11483))
28
+ - **Components**: `paged-attention` (Paged Attention) ([paper](https://arxiv.org/abs/2309.06180))
29
+ - **Components**: `conformer` (Conformer Convolution Module) ([paper](https://arxiv.org/abs/2005.08100))
30
+ - **Components**: `v-trace` (IMPALA: Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architectures) ([paper](https://arxiv.org/abs/1802.01561))
31
+ - **Components**: `cluster-gcn` (Cluster-GCN) ([paper](https://arxiv.org/abs/1905.07953))
32
+ - **Components**: `film` (FiLM) ([paper](https://arxiv.org/abs/1709.07871))
33
+ - **Losses**: `a2c` and `poly1-loss`.
34
+ - **Training/docs**: Add a guide and an end-to-end `poly1-loss` quickstart.
35
+
36
+ ### Changed
37
+ - Document how to enumerate catalog entries from Python.
38
+ - Add a licensed catalog-listing example.
39
+
40
+ ### Fixed
41
+ - Validate synthetic loader sample counts.
@@ -0,0 +1,191 @@
1
+
2
+ Apache License
3
+ Version 2.0, January 2004
4
+ http://www.apache.org/licenses/
5
+
6
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
7
+
8
+ 1. Definitions.
9
+
10
+ "License" shall mean the terms and conditions for use, reproduction,
11
+ and distribution as defined by Sections 1 through 9 of this document.
12
+
13
+ "Licensor" shall mean the copyright owner or entity authorized by
14
+ the copyright owner that is granting the License.
15
+
16
+ "Legal Entity" shall mean the union of the acting entity and all
17
+ other entities that control, are controlled by, or are under common
18
+ control with that entity. For the purposes of this definition,
19
+ "control" means (i) the power, direct or indirect, to cause the
20
+ direction or management of such entity, whether by contract or
21
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
22
+ outstanding shares, or (iii) beneficial ownership of such entity.
23
+
24
+ "You" (or "Your") shall mean an individual or Legal Entity
25
+ exercising permissions granted by this License.
26
+
27
+ "Source" form shall mean the preferred form for making modifications,
28
+ including but not limited to software source code, documentation
29
+ source, and configuration files.
30
+
31
+ "Object" form shall mean any form resulting from mechanical
32
+ transformation or translation of a Source form, including but
33
+ not limited to compiled object code, generated documentation,
34
+ and conversions to other media types.
35
+
36
+ "Work" shall mean the work of authorship, whether in Source or
37
+ Object form, made available under the License, as indicated by a
38
+ copyright notice that is included in or attached to the work
39
+ (an example is provided in the Appendix below).
40
+
41
+ "Derivative Works" shall mean any work, whether in Source or Object
42
+ form, that is based on (or derived from) the Work and for which the
43
+ editorial revisions, annotations, elaborations, or other modifications
44
+ represent, as a whole, an original work of authorship. For the purposes
45
+ of this License, Derivative Works shall not include works that remain
46
+ separable from, or merely link (or bind by name) to the interfaces of,
47
+ the Work and Derivative Works thereof.
48
+
49
+ "Contribution" shall mean any work of authorship, including
50
+ the original version of the Work and any modifications or additions
51
+ to that Work or Derivative Works thereof, that is intentionally
52
+ submitted to Licensor for inclusion in the Work by the copyright owner
53
+ or by an individual or Legal Entity authorized to submit on behalf of
54
+ the copyright owner. For the purposes of this definition, "submitted"
55
+ means any form of electronic, verbal, or written communication sent
56
+ to the Licensor or its representatives, including but not limited to
57
+ communication on electronic mailing lists, source code control systems,
58
+ and issue tracking systems that are managed by, or on behalf of, the
59
+ Licensor for the purpose of discussing and improving the Work, but
60
+ excluding communication that is conspicuously marked or otherwise
61
+ designated in writing by the copyright owner as "Not a Contribution."
62
+
63
+ "Contributor" shall mean Licensor and any individual or Legal Entity
64
+ on behalf of whom a Contribution has been received by Licensor and
65
+ subsequently incorporated within the Work.
66
+
67
+ 2. Grant of Copyright License. Subject to the terms and conditions of
68
+ this License, each Contributor hereby grants to You a perpetual,
69
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
70
+ copyright license to reproduce, prepare Derivative Works of,
71
+ publicly display, publicly perform, sublicense, and distribute the
72
+ Work and such Derivative Works in Source or Object form.
73
+
74
+ 3. Grant of Patent License. Subject to the terms and conditions of
75
+ this License, each Contributor hereby grants to You a perpetual,
76
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
77
+ (except as stated in this section) patent license to make, have made,
78
+ use, offer to sell, sell, import, and otherwise transfer the Work,
79
+ where such license applies only to those patent claims licensable
80
+ by such Contributor that are necessarily infringed by their
81
+ Contribution(s) alone or by combination of their Contribution(s)
82
+ with the Work to which such Contribution(s) was submitted. If You
83
+ institute patent litigation against any entity (including a
84
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
85
+ or a Contribution incorporated within the Work constitutes direct
86
+ or contributory patent infringement, then any patent licenses
87
+ granted to You under this License for that Work shall terminate
88
+ as of the date such litigation is filed.
89
+
90
+ 4. Redistribution. You may reproduce and distribute copies of the
91
+ Work or Derivative Works thereof in any medium, with or without
92
+ modifications, and in Source or Object form, provided that You
93
+ meet the following conditions:
94
+
95
+ (a) You must give any other recipients of the Work or
96
+ Derivative Works a copy of this License; and
97
+
98
+ (b) You must cause any modified files to carry prominent notices
99
+ stating that You changed the files; and
100
+
101
+ (c) You must retain, in the Source form of any Derivative Works
102
+ that You distribute, all copyright, patent, trademark, and
103
+ attribution notices from the Source form of the Work,
104
+ excluding those notices that do not pertain to any part of
105
+ the Derivative Works; and
106
+
107
+ (d) If the Work includes a "NOTICE" text file as part of its
108
+ distribution, then any Derivative Works that You distribute must
109
+ include a readable copy of the attribution notices contained
110
+ within such NOTICE file, excluding those notices that do not
111
+ pertain to any part of the Derivative Works, in at least one
112
+ of the following places: within a NOTICE text file distributed
113
+ as part of the Derivative Works; within the Source form or
114
+ documentation, if provided along with the Derivative Works; or,
115
+ within a display generated by the Derivative Works, if and
116
+ wherever such third-party notices normally appear. The contents
117
+ of the NOTICE file are for informational purposes only and
118
+ do not modify the License. You may add Your own attribution
119
+ notices within Derivative Works that You distribute, alongside
120
+ or as an addendum to the NOTICE text from the Work, provided
121
+ that such additional attribution notices cannot be construed
122
+ as modifying the License.
123
+
124
+ You may add Your own copyright statement to Your modifications and
125
+ may provide additional or different license terms and conditions
126
+ for use, reproduction, or distribution of Your modifications, or
127
+ for any such Derivative Works as a whole, provided Your use,
128
+ reproduction, and distribution of the Work otherwise complies with
129
+ the conditions stated in this License.
130
+
131
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
132
+ any Contribution intentionally submitted for inclusion in the Work
133
+ by You to the Licensor shall be under the terms and conditions of
134
+ this License, without any additional terms or conditions.
135
+ Notwithstanding the above, nothing herein shall supersede or modify
136
+ the terms of any separate license agreement you may have executed
137
+ with Licensor regarding such Contributions.
138
+
139
+ 6. Trademarks. This License does not grant permission to use the trade
140
+ names, trademarks, service marks, or product names of the Licensor,
141
+ except as required for reasonable and customary use in describing the
142
+ origin of the Work and reproducing the content of the NOTICE file.
143
+
144
+ 7. Disclaimer of Warranty. Unless required by applicable law or
145
+ agreed to in writing, Licensor provides the Work (and each
146
+ Contributor provides its Contributions) on an "AS IS" BASIS,
147
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
148
+ implied, including, without limitation, any warranties or conditions
149
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
150
+ PARTICULAR PURPOSE. You are solely responsible for determining the
151
+ appropriateness of using or redistributing the Work and assume any
152
+ risks associated with Your exercise of permissions under this License.
153
+
154
+ 8. Limitation of Liability. In no event and under no legal theory,
155
+ whether in tort (including negligence), contract, or otherwise,
156
+ unless required by applicable law (such as deliberate and grossly
157
+ negligent acts) or agreed to in writing, shall any Contributor be
158
+ liable to You for damages, including any direct, indirect, special,
159
+ incidental, or consequential damages of any character arising as a
160
+ result of this License or out of the use or inability to use the
161
+ Work (including but not limited to damages for loss of goodwill,
162
+ work stoppage, computer failure or malfunction, or any and all
163
+ other commercial damages or losses), even if such Contributor
164
+ has been advised of the possibility of such damages.
165
+
166
+ 9. Accepting Warranty or Additional Liability. While redistributing
167
+ the Work or Derivative Works thereof, You may choose to offer,
168
+ and charge a fee for, acceptance of support, warranty, indemnity,
169
+ or other liability obligations and/or rights consistent with this
170
+ License. However, in accepting such obligations, You may act only
171
+ on Your own behalf and on Your sole responsibility, not on behalf
172
+ of any other Contributor, and only if You agree to indemnify,
173
+ defend, and hold each Contributor harmless for any liability
174
+ incurred by, or claims asserted against, such Contributor by reason
175
+ of your accepting any such warranty or additional liability.
176
+
177
+ END OF TERMS AND CONDITIONS
178
+
179
+ Copyright 2026 Topologies Contributors
180
+
181
+ Licensed under the Apache License, Version 2.0 (the "License");
182
+ you may not use this file except in compliance with the License.
183
+ You may obtain a copy of the License at
184
+
185
+ http://www.apache.org/licenses/LICENSE-2.0
186
+
187
+ Unless required by applicable law or agreed to in writing, software
188
+ distributed under the License is distributed on an "AS IS" BASIS,
189
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
190
+ See the License for the specific language governing permissions and
191
+ limitations under the License.
@@ -0,0 +1,143 @@
1
+ Metadata-Version: 2.5
2
+ Name: topologies
3
+ Version: 0.0.1
4
+ Summary: A continuously updated library of modern neural-network architectures on PyTorch
5
+ Project-URL: Homepage, https://topologies.hackwhiz.net
6
+ Project-URL: Documentation, https://topologies.hackwhiz.net
7
+ Project-URL: Repository, https://github.com/yvsazh/topologies
8
+ Project-URL: Issues, https://github.com/yvsazh/topologies/issues
9
+ Author: Yaroslav Sazhchenko
10
+ License: Apache-2.0
11
+ License-File: LICENSE
12
+ Keywords: architectures,deep-learning,neural-networks,pytorch,transformer
13
+ Classifier: Development Status :: 3 - Alpha
14
+ Classifier: Intended Audience :: Developers
15
+ Classifier: Intended Audience :: Science/Research
16
+ Classifier: License :: OSI Approved :: Apache Software License
17
+ Classifier: Programming Language :: Python :: 3
18
+ Classifier: Programming Language :: Python :: 3.10
19
+ Classifier: Programming Language :: Python :: 3.11
20
+ Classifier: Programming Language :: Python :: 3.12
21
+ Classifier: Programming Language :: Python :: 3.13
22
+ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
23
+ Requires-Python: >=3.10
24
+ Requires-Dist: httpx>=0.27
25
+ Requires-Dist: numpy>=1.23
26
+ Requires-Dist: pydantic>=2.0
27
+ Requires-Dist: pyyaml>=6.0
28
+ Requires-Dist: tomli>=2.0; python_version < '3.11'
29
+ Requires-Dist: torch<2.13,>=2.2
30
+ Provides-Extra: all
31
+ Requires-Dist: datasets>=2.14; extra == 'all'
32
+ Requires-Dist: matplotlib>=3.7; extra == 'all'
33
+ Requires-Dist: pillow>=9; extra == 'all'
34
+ Requires-Dist: psutil>=5.9; extra == 'all'
35
+ Requires-Dist: pyarrow>=12; extra == 'all'
36
+ Requires-Dist: tensorboard>=2.12; extra == 'all'
37
+ Requires-Dist: tokenizers>=0.15; extra == 'all'
38
+ Requires-Dist: torchvision>=0.17; extra == 'all'
39
+ Provides-Extra: dev
40
+ Requires-Dist: build; extra == 'dev'
41
+ Requires-Dist: matplotlib>=3.7; extra == 'dev'
42
+ Requires-Dist: pytest-timeout>=2.3; extra == 'dev'
43
+ Requires-Dist: pytest>=8; extra == 'dev'
44
+ Requires-Dist: ruff==0.16.9; extra == 'dev'
45
+ Requires-Dist: twine; extra == 'dev'
46
+ Provides-Extra: introspection
47
+ Requires-Dist: matplotlib>=3.7; extra == 'introspection'
48
+ Provides-Extra: nlp
49
+ Requires-Dist: datasets>=2.14; extra == 'nlp'
50
+ Requires-Dist: tokenizers>=0.15; extra == 'nlp'
51
+ Provides-Extra: optimization
52
+ Requires-Dist: torchao==0.18.0; extra == 'optimization'
53
+ Provides-Extra: qt
54
+ Requires-Dist: pyside6>=6.5; extra == 'qt'
55
+ Provides-Extra: synth-parquet
56
+ Requires-Dist: pyarrow>=12; extra == 'synth-parquet'
57
+ Provides-Extra: system
58
+ Requires-Dist: psutil>=5.9; extra == 'system'
59
+ Provides-Extra: tensorboard
60
+ Requires-Dist: tensorboard>=2.12; extra == 'tensorboard'
61
+ Provides-Extra: vision
62
+ Requires-Dist: pillow>=9; extra == 'vision'
63
+ Requires-Dist: torchvision>=0.17; extra == 'vision'
64
+ Provides-Extra: visualization
65
+ Requires-Dist: matplotlib>=3.7; extra == 'visualization'
66
+ Description-Content-Type: text/markdown
67
+
68
+ # topologies
69
+
70
+ A PyTorch toolbox for building, training, comparing and inspecting neural-network
71
+ architectures — plus a catalog of small reference implementations (GPT, LLaMA, ViT,
72
+ ResNet, BERT, Mamba, ...).
73
+
74
+ > **Early alpha (v0.0.1).** The toolbox — training engine, data utilities, tokenizers,
75
+ > model constructor, generation, evaluation, experiments, introspection — is tested.
76
+ > Catalog models in `topologies.models` / `topologies.layers` are experimental
77
+ > references: importable, trainable and smoke-tested, but not validated against the papers.
78
+ > APIs may change between releases.
79
+
80
+ ## Install
81
+
82
+ ```bash
83
+ pip install topologies # core
84
+ pip install "topologies[all]" # + vision, nlp, tensorboard, visualization, ...
85
+ ```
86
+
87
+ Python 3.10–3.13, PyTorch `>=2.2,<2.13`.
88
+
89
+ ## Quickstart
90
+
91
+ Train a tiny language model on any text corpus:
92
+
93
+ ```python
94
+ from topologies import Trainer, build, data, generate, train_tokenizer
95
+ from topologies.nn import TransformerLM
96
+
97
+ docs = data.read_documents("corpus.txt")
98
+ train_docs, val_docs = data.train_val_split(docs, 0.1)
99
+ tok = train_tokenizer(train_docs, kind="bpe", vocab_size=2048)
100
+ model = build(TransformerLM, "tiny", vocab_size=tok.vocab_size, max_seq=128)
101
+
102
+ trainer = Trainer(model, data.lm_dataset(train_docs, tok, 128),
103
+ data.lm_dataset(val_docs, tok, 128),
104
+ tokenizer=tok, batch_size=32, max_epochs=20, lr=1e-3)
105
+ trainer.fit()
106
+ trainer.save_pretrained("artifacts/tiny-lm")
107
+ print(generate(model, "Once upon a time", tokenizer=tok, max_new_tokens=100, top_k=40))
108
+ ```
109
+
110
+ Compare many configurations with one declaration — the library runs every trial
111
+ (resumably), then ranks and explains the results:
112
+
113
+ ```python
114
+ from topologies import Experiment
115
+ from topologies.experiments import LanguageModeling, Variant
116
+
117
+ exp = Experiment("arch-x-tokenizer",
118
+ task=LanguageModeling(docs, val_fraction=0.1, block_size=128),
119
+ axes={"model": [TransformerLM,
120
+ Variant("llama-ish", TransformerLM, pos="rope",
121
+ norm="rms", activation="swiglu")],
122
+ "tokenizer": ["char", "bpe:1k", "bpe:4k"]},
123
+ target_params="1M", train_for="2 epochs", seeds=2)
124
+ exp.run().show()
125
+ ```
126
+
127
+ ## What's inside
128
+
129
+ - `topologies.nn` — composable stacks, presets and parameter budgets
130
+ - `Trainer`, `generate`, evaluation, optimization and quantization helpers
131
+ - `topologies.data`, dataset catalog, tokenizers
132
+ - `Experiment` — sweeps, leaderboards, pivots, importance analysis
133
+ - `topologies.introspection` / `topologies.visualization` — look inside trained models
134
+ - `topologies` CLI — `topologies --help`
135
+
136
+ ## Links
137
+
138
+ - Documentation: <https://topologies.hackwhiz.net>
139
+ - Repository and issues: <https://github.com/yvsazh/topologies>
140
+
141
+ ## License
142
+
143
+ Apache-2.0
@@ -0,0 +1,76 @@
1
+ # topologies
2
+
3
+ A PyTorch toolbox for building, training, comparing and inspecting neural-network
4
+ architectures — plus a catalog of small reference implementations (GPT, LLaMA, ViT,
5
+ ResNet, BERT, Mamba, ...).
6
+
7
+ > **Early alpha (v0.0.1).** The toolbox — training engine, data utilities, tokenizers,
8
+ > model constructor, generation, evaluation, experiments, introspection — is tested.
9
+ > Catalog models in `topologies.models` / `topologies.layers` are experimental
10
+ > references: importable, trainable and smoke-tested, but not validated against the papers.
11
+ > APIs may change between releases.
12
+
13
+ ## Install
14
+
15
+ ```bash
16
+ pip install topologies # core
17
+ pip install "topologies[all]" # + vision, nlp, tensorboard, visualization, ...
18
+ ```
19
+
20
+ Python 3.10–3.13, PyTorch `>=2.2,<2.13`.
21
+
22
+ ## Quickstart
23
+
24
+ Train a tiny language model on any text corpus:
25
+
26
+ ```python
27
+ from topologies import Trainer, build, data, generate, train_tokenizer
28
+ from topologies.nn import TransformerLM
29
+
30
+ docs = data.read_documents("corpus.txt")
31
+ train_docs, val_docs = data.train_val_split(docs, 0.1)
32
+ tok = train_tokenizer(train_docs, kind="bpe", vocab_size=2048)
33
+ model = build(TransformerLM, "tiny", vocab_size=tok.vocab_size, max_seq=128)
34
+
35
+ trainer = Trainer(model, data.lm_dataset(train_docs, tok, 128),
36
+ data.lm_dataset(val_docs, tok, 128),
37
+ tokenizer=tok, batch_size=32, max_epochs=20, lr=1e-3)
38
+ trainer.fit()
39
+ trainer.save_pretrained("artifacts/tiny-lm")
40
+ print(generate(model, "Once upon a time", tokenizer=tok, max_new_tokens=100, top_k=40))
41
+ ```
42
+
43
+ Compare many configurations with one declaration — the library runs every trial
44
+ (resumably), then ranks and explains the results:
45
+
46
+ ```python
47
+ from topologies import Experiment
48
+ from topologies.experiments import LanguageModeling, Variant
49
+
50
+ exp = Experiment("arch-x-tokenizer",
51
+ task=LanguageModeling(docs, val_fraction=0.1, block_size=128),
52
+ axes={"model": [TransformerLM,
53
+ Variant("llama-ish", TransformerLM, pos="rope",
54
+ norm="rms", activation="swiglu")],
55
+ "tokenizer": ["char", "bpe:1k", "bpe:4k"]},
56
+ target_params="1M", train_for="2 epochs", seeds=2)
57
+ exp.run().show()
58
+ ```
59
+
60
+ ## What's inside
61
+
62
+ - `topologies.nn` — composable stacks, presets and parameter budgets
63
+ - `Trainer`, `generate`, evaluation, optimization and quantization helpers
64
+ - `topologies.data`, dataset catalog, tokenizers
65
+ - `Experiment` — sweeps, leaderboards, pivots, importance analysis
66
+ - `topologies.introspection` / `topologies.visualization` — look inside trained models
67
+ - `topologies` CLI — `topologies --help`
68
+
69
+ ## Links
70
+
71
+ - Documentation: <https://topologies.hackwhiz.net>
72
+ - Repository and issues: <https://github.com/yvsazh/topologies>
73
+
74
+ ## License
75
+
76
+ Apache-2.0
@@ -0,0 +1,146 @@
1
+ [build-system]
2
+ requires = ["hatchling"]
3
+ build-backend = "hatchling.build"
4
+
5
+ [project]
6
+ name = "topologies"
7
+ dynamic = ["version"]
8
+ description = "A continuously updated library of modern neural-network architectures on PyTorch"
9
+ readme = "README.md"
10
+ license = { text = "Apache-2.0" }
11
+ requires-python = ">=3.10"
12
+ authors = [{ name = "Yaroslav Sazhchenko" }]
13
+ keywords = ["pytorch", "deep-learning", "neural-networks", "architectures", "transformer"]
14
+ classifiers = [
15
+ "Development Status :: 3 - Alpha",
16
+ "Intended Audience :: Developers",
17
+ "Intended Audience :: Science/Research",
18
+ "License :: OSI Approved :: Apache Software License",
19
+ "Programming Language :: Python :: 3",
20
+ "Programming Language :: Python :: 3.10",
21
+ "Programming Language :: Python :: 3.11",
22
+ "Programming Language :: Python :: 3.12",
23
+ "Programming Language :: Python :: 3.13",
24
+ "Topic :: Scientific/Engineering :: Artificial Intelligence",
25
+ ]
26
+ dependencies = [
27
+ "torch>=2.2,<2.13", # upper bound = newest torch the local suite passes on; raise after testing
28
+ "numpy>=1.23",
29
+ "pydantic>=2.0",
30
+ "PyYAML>=6.0",
31
+ "httpx>=0.27", # artifact-store HTTP integration (only via topologies.artifact_store.client)
32
+ "tomli>=2.0; python_version < '3.11'", # tomllib backport (utils.config, artifact_store)
33
+ ]
34
+ urls = { Homepage = "https://topologies.hackwhiz.net", Documentation = "https://topologies.hackwhiz.net", Repository = "https://github.com/yvsazh/topologies", Issues = "https://github.com/yvsazh/topologies/issues" }
35
+
36
+ [project.optional-dependencies]
37
+ dev = ["pytest>=8", "pytest-timeout>=2.3", "ruff==0.16.9", "build", "twine", "matplotlib>=3.7"]
38
+ vision = ["torchvision>=0.17", "pillow>=9"]
39
+ nlp = ["tokenizers>=0.15", "datasets>=2.14"]
40
+ tensorboard = ["tensorboard>=2.12"]
41
+ system = ["psutil>=5.9"]
42
+ optimization = ["torchao==0.18.0"]
43
+ visualization = ["matplotlib>=3.7"]
44
+ introspection = ["topologies[visualization]"] # alias: figures for topologies.introspection
45
+ qt = ["PySide6>=6.5"] # optional GUI reference example only; core visualization stays Qt-free
46
+ synth-parquet = ["pyarrow>=12"] # optional: parquet shards and exports
47
+ all = ["topologies[vision,nlp,tensorboard,system,synth-parquet,visualization]"]
48
+
49
+ [project.scripts]
50
+ topologies = "topologies.cli:main"
51
+
52
+ [tool.hatch.version]
53
+ path = "src/topologies/__init__.py"
54
+
55
+ [tool.hatch.build.targets.wheel]
56
+ packages = ["src/topologies"]
57
+ include = ["src/topologies/training/model_recipes.json"]
58
+ ignore-vcs = true
59
+
60
+ [tool.hatch.build.targets.sdist]
61
+ # Explicit allow-list: package only the library and its public metadata.
62
+ include = [
63
+ "/src/topologies",
64
+ "/README.md",
65
+ "/CHANGELOG.md",
66
+ "/LICENSE",
67
+ "/pyproject.toml",
68
+ ]
69
+ exclude = ["__pycache__"]
70
+
71
+ [tool.ruff]
72
+ target-version = "py310"
73
+ line-length = 100
74
+ extend-exclude = ["src/topologies/models/[!_]*.py", "src/topologies/layers/[!_]*.py"]
75
+
76
+ [tool.ruff.lint]
77
+ # Explicit rule set (H-P07): the rules ruff 0.16.9 enabled by default when this was
78
+ # written, frozen so a ruff upgrade cannot silently change what CI enforces.
79
+ select = [
80
+ "ASYNC100", "ASYNC105", "ASYNC115", "ASYNC116", "ASYNC210", "ASYNC220", "ASYNC221",
81
+ "ASYNC222", "ASYNC230", "ASYNC251", "B002", "B003", "B004", "B005", "B006", "B008",
82
+ "B009", "B010", "B012", "B013", "B014", "B015", "B016", "B017", "B018", "B019", "B020",
83
+ "B021", "B022", "B023", "B025", "B026", "B029", "B030", "B031", "B032", "B033", "B035",
84
+ "B039", "BLE001", "C400", "C401", "C402", "C403", "C404", "C405", "C406", "C408", "C409",
85
+ "C410", "C411", "C413", "C414", "C415", "C417", "C418", "C419", "D419", "DTZ001", "DTZ002",
86
+ "DTZ004", "DTZ005", "DTZ006", "DTZ007", "DTZ011", "DTZ012", "DTZ901", "E722", "E902",
87
+ "EXE002", "EXE004", "EXE005", "F401", "F402", "F404", "F407", "F501", "F502", "F503",
88
+ "F504", "F505", "F506", "F507", "F508", "F509", "F521", "F522", "F523", "F524", "F525",
89
+ "F541", "F601", "F602", "F621", "F622", "F631", "F632", "F633", "F634", "F701", "F702",
90
+ "F704", "F706", "F707", "F811", "F821", "F822", "F823", "F841", "F842", "F901", "FA100",
91
+ "FA102", "FLY002", "FURB105", "FURB122", "FURB129", "FURB132", "FURB136", "FURB157",
92
+ "FURB161", "FURB162", "FURB163", "FURB166", "FURB167", "FURB168", "FURB169", "FURB177",
93
+ "FURB181", "FURB188", "FURB192", "G010", "G101", "G201", "G202", "I001", "INT001",
94
+ "INT002", "INT003", "ISC004", "LOG001", "LOG002", "LOG009", "LOG014", "LOG015", "N999",
95
+ "PERF101", "PERF102", "PERF402", "PGH005", "PIE790", "PIE794", "PIE796", "PIE800",
96
+ "PIE804", "PIE807", "PIE808", "PIE810", "PLC0105", "PLC0131", "PLC0132", "PLC0205",
97
+ "PLC0206", "PLC0208", "PLC0414", "PLC3002", "PLE0100", "PLE0101", "PLE0115", "PLE0116",
98
+ "PLE0117", "PLE0118", "PLE0303", "PLE0304", "PLE0305", "PLE0307", "PLE0308", "PLE0309",
99
+ "PLE0604", "PLE0605", "PLE0643", "PLE0704", "PLE1132", "PLE1142", "PLE1205", "PLE1206",
100
+ "PLE1300", "PLE1307", "PLE1310", "PLE1507", "PLE1519", "PLE1520", "PLE1700", "PLE2502",
101
+ "PLE2510", "PLE2512", "PLE2513", "PLE2514", "PLE2515", "PLR0124", "PLR0133", "PLR0206",
102
+ "PLR1704", "PLR1708", "PLR1711", "PLR1716", "PLR1722", "PLR1730", "PLR1733", "PLR1736",
103
+ "PLR2044", "PLW0120", "PLW0127", "PLW0128", "PLW0129", "PLW0131", "PLW0133", "PLW0177",
104
+ "PLW0211", "PLW0245", "PLW0406", "PLW0602", "PLW0604", "PLW0642", "PLW0711", "PLW1501",
105
+ "PLW1507", "PLW1508", "PLW1509", "PLW1510", "PLW2101", "PT010", "PT014", "PT020", "PT025",
106
+ "PT026", "PT031", "PTH124", "PTH210", "PYI001", "PYI002", "PYI003", "PYI004", "PYI005",
107
+ "PYI006", "PYI007", "PYI008", "PYI009", "PYI010", "PYI012", "PYI013", "PYI015", "PYI016",
108
+ "PYI017", "PYI018", "PYI019", "PYI020", "PYI025", "PYI026", "PYI029", "PYI030", "PYI032",
109
+ "PYI033", "PYI034", "PYI035", "PYI036", "PYI041", "PYI042", "PYI043", "PYI044", "PYI045",
110
+ "PYI046", "PYI047", "PYI048", "PYI049", "PYI050", "PYI052", "PYI055", "PYI057", "PYI058",
111
+ "PYI059", "PYI061", "PYI062", "PYI063", "PYI064", "PYI066", "RET501", "RUF007", "RUF008",
112
+ "RUF009", "RUF010", "RUF013", "RUF015", "RUF016", "RUF017", "RUF018", "RUF019", "RUF020",
113
+ "RUF022", "RUF023", "RUF024", "RUF026", "RUF028", "RUF030", "RUF032", "RUF033", "RUF034",
114
+ "RUF040", "RUF041", "RUF046", "RUF048", "RUF049", "RUF051", "RUF053", "RUF057", "RUF058",
115
+ "RUF059", "RUF063", "RUF068", "RUF100", "RUF101", "RUF200", "S102", "S110", "S112",
116
+ "SIM101", "SIM102", "SIM103", "SIM107", "SIM113", "SIM114", "SIM115", "SIM117", "SIM118",
117
+ "SIM201", "SIM202", "SIM208", "SIM210", "SIM211", "SIM220", "SIM221", "SIM222", "SIM223",
118
+ "SIM401", "SIM905", "SIM911", "T100", "TC004", "TC005", "TC007", "TC010", "TRY002",
119
+ "TRY004", "TRY201", "TRY203", "TRY401", "UP001", "UP003", "UP004", "UP005", "UP006",
120
+ "UP007", "UP008", "UP009", "UP010", "UP011", "UP012", "UP014", "UP017", "UP018", "UP019",
121
+ "UP020", "UP021", "UP022", "UP023", "UP024", "UP025", "UP026", "UP028", "UP029", "UP030",
122
+ "UP031", "UP032", "UP033", "UP034", "UP035", "UP036", "UP037", "UP039", "UP040", "UP041",
123
+ "UP043", "UP044", "UP045", "UP046", "UP047", "UP049", "UP050", "W605", "YTT101", "YTT102",
124
+ "YTT103", "YTT201", "YTT202", "YTT203", "YTT204", "YTT301", "YTT302", "YTT303",
125
+ ]
126
+ # DTZ003: utcnow()+"Z" is the deliberate ISO convention used across registry/runs.
127
+ # PLR0402: `import torch.nn as nn` is the universal PyTorch idiom.
128
+ ignore = ["RUF012", "DTZ003", "PLR0402", "EXE001"]
129
+
130
+ [tool.ruff.lint.per-file-ignores]
131
+ "src/topologies/training/charformer_data.py" = ["I001"]
132
+ "src/topologies/training/recipes/__init__.py" = ["I001"]
133
+ "src/topologies/training/recipes/permutation_lm.py" = ["I001", "RUF059", "TRY004"]
134
+ "src/topologies/training/recipes/spanbert.py" = ["I001", "TRY004"]
135
+ "src/topologies/validation/catalog_workflows.py" = ["F401", "I001"]
136
+ "tests/test_catalog_workflows.py" = ["S102"]
137
+ "tests/test_component_workflows.py" = ["I001", "S102", "SIM102"]
138
+ "tests/test_modernbert_usage_projection.py" = ["S102"]
139
+ "tests/test_xlnet_models.py" = ["S102"]
140
+
141
+ [tool.pytest.ini_options]
142
+ testpaths = ["tests"]
143
+ pythonpath = ["src"]
144
+ markers = ["slow: long-running tests (deselect with -m 'not slow')"]
145
+ # New deprecations raised from our own modules fail the suite; third-party ones stay warnings.
146
+ filterwarnings = ["error::DeprecationWarning:topologies.*"]