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.
- topologies-0.0.1/.gitignore +20 -0
- topologies-0.0.1/CHANGELOG.md +41 -0
- topologies-0.0.1/LICENSE +191 -0
- topologies-0.0.1/PKG-INFO +143 -0
- topologies-0.0.1/README.md +76 -0
- topologies-0.0.1/pyproject.toml +146 -0
- topologies-0.0.1/src/topologies/.training-generated-manifest.json +35 -0
- topologies-0.0.1/src/topologies/.training-materialized-manifest.json +35 -0
- topologies-0.0.1/src/topologies/__init__.py +89 -0
- topologies-0.0.1/src/topologies/_catalog.py +293 -0
- topologies-0.0.1/src/topologies/_compat.py +27 -0
- topologies-0.0.1/src/topologies/_execution.py +178 -0
- topologies-0.0.1/src/topologies/_util/__init__.py +1 -0
- topologies-0.0.1/src/topologies/_util/fs.py +98 -0
- topologies-0.0.1/src/topologies/_util/jsonsafe.py +42 -0
- topologies-0.0.1/src/topologies/_util/seed.py +19 -0
- topologies-0.0.1/src/topologies/_util/suggest.py +26 -0
- topologies-0.0.1/src/topologies/analysis/__init__.py +102 -0
- topologies-0.0.1/src/topologies/analysis/core.py +448 -0
- topologies-0.0.1/src/topologies/analysis/decode.py +186 -0
- topologies-0.0.1/src/topologies/analysis/engineering.py +878 -0
- topologies-0.0.1/src/topologies/analysis/integrations.py +70 -0
- topologies-0.0.1/src/topologies/analysis/profiling.py +122 -0
- topologies-0.0.1/src/topologies/analysis/records.py +123 -0
- topologies-0.0.1/src/topologies/analysis/statistics.py +211 -0
- topologies-0.0.1/src/topologies/analysis/visualization.py +135 -0
- topologies-0.0.1/src/topologies/artifact_store/__init__.py +31 -0
- topologies-0.0.1/src/topologies/artifact_store/client.py +402 -0
- topologies-0.0.1/src/topologies/artifact_store/exceptions.py +21 -0
- topologies-0.0.1/src/topologies/build/__init__.py +93 -0
- topologies-0.0.1/src/topologies/build/api.py +111 -0
- topologies-0.0.1/src/topologies/build/audit.py +103 -0
- topologies-0.0.1/src/topologies/build/budget.py +212 -0
- topologies-0.0.1/src/topologies/build/catalog.py +202 -0
- topologies-0.0.1/src/topologies/build/config.py +246 -0
- topologies-0.0.1/src/topologies/build/contract.py +85 -0
- topologies-0.0.1/src/topologies/build/params.py +92 -0
- topologies-0.0.1/src/topologies/build/presets.py +100 -0
- topologies-0.0.1/src/topologies/build/validation.py +75 -0
- topologies-0.0.1/src/topologies/catalog/losses/a2c.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/adversarial-hinge.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/binary-cross-entropy-logits.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/bpr.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/cross-entropy.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/ctc.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/density-nll.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/dice.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/gaussian-kl.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/gaussian-nll.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/generalized-iou.json +10 -0
- topologies-0.0.1/src/topologies/catalog/losses/huber.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/info-nce.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/mean-absolute-error.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/mean-squared-error.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/multi-resolution-stft.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/pinball.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/poly1-loss.json +1 -0
- topologies-0.0.1/src/topologies/catalog/losses/si-sdr.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/sigmoid-focal.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/smooth-l1.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/softmax-focal.json +9 -0
- topologies-0.0.1/src/topologies/catalog/losses/triplet-margin.json +9 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/adabelief.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/adafactor.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/adagrad.json +9 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/adahessian.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/adam-8bit.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/adam-mini.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/adam.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/adamw.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/ademamix.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/asam.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/ema.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/fadam.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/fisher-sam.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/flora.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/galore.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/grokfast.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/gsam.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/kfac.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/lamb.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/lars.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/lion.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/lookahead.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/model-soup.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/muon.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/novograd.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/polynomial-decay.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/qlora.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/radam.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/relora.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/rmsprop.json +9 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/sam.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/schedule-free.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/sgd.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/shampoo.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/sm3.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/soap.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/sophia.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/swa.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/swad.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/swalp.json +1 -0
- topologies-0.0.1/src/topologies/catalog/optimizers/twa.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/clr.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/cooldown.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/cosine-min.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/cosine-restarts.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/exponential.json +19 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/grafted.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/multistep.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/noam.json +21 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/onecycle.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/reduce-on-plateau.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/sgdr.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/trapezoidal.json +21 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/warmup-linear.json +1 -0
- topologies-0.0.1/src/topologies/catalog/schedulers/wsd.json +1 -0
- topologies-0.0.1/src/topologies/catalog/stability/activation-offload.json +1 -0
- topologies-0.0.1/src/topologies/catalog/stability/bf16-recipe.json +14 -0
- topologies-0.0.1/src/topologies/catalog/stability/compile-parity.json +4 -0
- topologies-0.0.1/src/topologies/catalog/stability/fsdp-wrap.json +17 -0
- topologies-0.0.1/src/topologies/catalog/stability/grad-checkpoint.json +1 -0
- topologies-0.0.1/src/topologies/catalog/stability/grad-clip.json +17 -0
- topologies-0.0.1/src/topologies/catalog/stability/llrd.json +18 -0
- topologies-0.0.1/src/topologies/catalog/stability/loss-scale.json +17 -0
- topologies-0.0.1/src/topologies/catalog/stability/nan-skip.json +1 -0
- topologies-0.0.1/src/topologies/catalog/stability/zero-partition.json +1 -0
- topologies-0.0.1/src/topologies/cli.py +294 -0
- topologies-0.0.1/src/topologies/data/__init__.py +85 -0
- topologies-0.0.1/src/topologies/data/adapters.py +162 -0
- topologies-0.0.1/src/topologies/data/collate.py +96 -0
- topologies-0.0.1/src/topologies/data/corpus.py +263 -0
- topologies-0.0.1/src/topologies/data/datasets.py +53 -0
- topologies-0.0.1/src/topologies/data/download.py +204 -0
- topologies-0.0.1/src/topologies/data/event_codec.py +62 -0
- topologies-0.0.1/src/topologies/data/integrations/__init__.py +24 -0
- topologies-0.0.1/src/topologies/data/integrations/karate.py +68 -0
- topologies-0.0.1/src/topologies/data/integrations/local.py +345 -0
- topologies-0.0.1/src/topologies/data/loaders.py +71 -0
- topologies-0.0.1/src/topologies/data/preprocessing.py +188 -0
- topologies-0.0.1/src/topologies/data/profiling.py +341 -0
- topologies-0.0.1/src/topologies/data/samplers.py +32 -0
- topologies-0.0.1/src/topologies/data/splits.py +158 -0
- topologies-0.0.1/src/topologies/data/stream.py +313 -0
- topologies-0.0.1/src/topologies/data/text.py +212 -0
- topologies-0.0.1/src/topologies/data/windows.py +90 -0
- topologies-0.0.1/src/topologies/datasets/.dataset-catalog-generated-manifest.json +6 -0
- topologies-0.0.1/src/topologies/datasets/__init__.py +25 -0
- topologies-0.0.1/src/topologies/datasets/api.py +101 -0
- topologies-0.0.1/src/topologies/datasets/artifacts.lock.json +5332 -0
- topologies-0.0.1/src/topologies/datasets/catalog.json +17755 -0
- topologies-0.0.1/src/topologies/datasets/core.py +511 -0
- topologies-0.0.1/src/topologies/datasets/integrations/.dataset-generated-manifest.json +21 -0
- topologies-0.0.1/src/topologies/datasets/integrations/__init__.py +1 -0
- topologies-0.0.1/src/topologies/datasets/integrations/ag_news/__init__.py +22 -0
- topologies-0.0.1/src/topologies/datasets/integrations/cifar10/__init__.py +55 -0
- topologies-0.0.1/src/topologies/datasets/integrations/cora/__init__.py +36 -0
- topologies-0.0.1/src/topologies/datasets/integrations/fashion_mnist/__init__.py +18 -0
- topologies-0.0.1/src/topologies/datasets/integrations/fsdd/__init__.py +54 -0
- topologies-0.0.1/src/topologies/datasets/integrations/iris/__init__.py +80 -0
- topologies-0.0.1/src/topologies/datasets/integrations/jena_climate/__init__.py +59 -0
- topologies-0.0.1/src/topologies/datasets/integrations/mnist/__init__.py +18 -0
- topologies-0.0.1/src/topologies/datasets/integrations/movielens_100k/__init__.py +43 -0
- topologies-0.0.1/src/topologies/datasets/integrations/oxford_iiit_pet/__init__.py +52 -0
- topologies-0.0.1/src/topologies/datasets/integrations/palmer_penguins/__init__.py +46 -0
- topologies-0.0.1/src/topologies/datasets/integrations/penn_fudan_pedestrians/__init__.py +52 -0
- topologies-0.0.1/src/topologies/datasets/integrations/speech_commands/__init__.py +41 -0
- topologies-0.0.1/src/topologies/datasets/integrations/uci_har/__init__.py +27 -0
- topologies-0.0.1/src/topologies/datasets/integrations/wikitext_2_raw/__init__.py +44 -0
- topologies-0.0.1/src/topologies/datasets/records/1000-genomes.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/2wikimultihopqa.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/3dpw.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/abo.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/acdc.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/activitynet-captions.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/activitynet.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ade20k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/adni.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/adroit.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/adult.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/aerorit.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/agora.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/aishell.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ali-ccp.json +17 -0
- topologies-0.0.1/src/topologies/datasets/records/alimama.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/alpaca.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/amass.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/amazon-2018.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/amazon-2023.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/amazon-computers.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ami.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/amos.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/amos22.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/anli.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/anuraset.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/apps.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/aptos.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/arc.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/argoverse-2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/arkit-scenes.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/arxiv-pubmed-sum.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ashrae-gepiii.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/asvspoof-2019.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/atari-100k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/audiocaps.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/audioset.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ava.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/avazu.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/avenue.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/babyai.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/bach.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/bank-marketing.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/bbbc.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/bbh.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/bbq.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/bdd100k.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/beans.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/bigearthnet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/bijie.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/billion-word.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/bindingdb.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/birdclef.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/birdset.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ble-aoa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/blendedmvs.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/bobsl.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/book-crossing.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/brats.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/breakhis.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/btad.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/btcv.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/c4.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/california-housing.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/caltech-101.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/camelyon-16.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/camelyon-17.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/camelyon.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/camus.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/camvid.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/caselaw-hold.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/casia-splicing.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/casia-v2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/casia-webface.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/cassava-leaf.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cassava.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cc3m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ccmatrix.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ccpg.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/celeba-spurious.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/celeba.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/census.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/chaos.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/charades.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/chembl-33.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/chestxray14.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/chexpert.json +11 -0
- topologies-0.0.1/src/topologies/datasets/records/chime-4.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/chime-6.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/chime-frb.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cholec-t50.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cholec80.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cholecseg8k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cifar-10.1.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/cifar-100.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cinic-10.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/citeseer.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cityscapes.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/climsim.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/clotho.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cmapss.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cnn-dailymail.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/co3d.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/coauthor-cs.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/coco-2017.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/coco-captions.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/coco-pose.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/codecontests.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/codex.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/common-voice.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/compas.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/compcars.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/conceptual-12m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/coqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cora.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cord.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/covertype.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/coyo-700m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/crafter.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/crema-d.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/cremi.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/criteo.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/crowdhuman.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/crowdpose.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/csl-daily.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ctw1500.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cuad.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cub-200.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/cuhk03.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/cwru.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/d4rl.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dagm.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dailydialog.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dancetrack.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/datacomp-1b.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/davis.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dcase.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dclde.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/deeperforensics.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/deepglobe.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/deeppcb.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/didemo.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/diii-d.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dmc.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dns-challenge.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/docvqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dolly.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dolma.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/domainnet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dota.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/drop.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ds-1000.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/dsb-2018.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dsec-flow.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dtd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/duacs-aviso.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/duacs.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dud-e.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dude.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dvs-cifar10.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dvs-gesture.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/dycheck.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/earnings-22.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ego4d.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/eicu-crd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/electricity.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/elliptic.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/elpv.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/emnist.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/endovis.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/enose.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/epic-kitchens-100.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/epinions.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/eqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/era5.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/esc-50.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/esd.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/eth3d.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ett.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/europarl.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/eurosat.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/exchange.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/extrasensory.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fair1m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fashioniq.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fedscale.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/femto-pronostia.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fever.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ffhq.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/fgvc-aircraft.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fineweb-edu.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fineweb.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fintabnet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fiqa-2018.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/flan.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/flchain.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fleurs.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/flickr30k-entities.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/flickr30k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/floodnet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/flores-200.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/flyingchairs.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fma.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fmd.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/fmow.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/food-101.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/forgerynet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/foursquare.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/franka-kitchen.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/fraud-ieee.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/freihand.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/fsd50k.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/fsdnoisy18k.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/funsd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gages-ii.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gait3d.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gbsg2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gefcom2014.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gelsight-wedge.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/german-credit.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gigaspeech.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gigaword.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/glue.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/goodreads.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/got-10k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gowalla.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/grew.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/grf.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gsm8k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gtzan.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/gwosc.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/hacs.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/hellaswag.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/hepmass.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/hgb.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/higgs.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/hirid.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/hm3d.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/hmdb51.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/home-credit-default-risk.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/hotpotqa.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/house-prices.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/howto100m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/htru2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/human36m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/humaneval.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/iam.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/icdar15.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/icdar2017-mlt.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/idrid.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/iemocap.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ifeval.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ili.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/imagenet-1k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/imagenet-21k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/imagenet-9.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/imagenet-o.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/imagenet-p.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/imagenet-real.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/imagenet-v2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/imdb.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/inaturalist-2017.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/inaturalist-2021.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/inria-aerial.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ipinyou.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/isic.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/isles-2022.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/isprs-potsdam.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/iwslt.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/jarvis-dft.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/jester.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/jigsaw-toxic.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/jigsaws.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/jvs.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/kermany-oct.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kilt.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/kinetics-400.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kinetics-600.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kinetics-700.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kits.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kitti.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kkbox-wsdm-cup.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kmnist.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ksponspeech.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/kuairec.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kuaisar.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/kvasir-seg.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/laion-2b.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/laion-400m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lakh-midi.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/landslide4sense.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/largest.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lasot.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lastfm-1k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/leaf.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/levir-cd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lfm-360k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lfw.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/libero.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/libri-light.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/librimix.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/librispeech.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/libritts.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lidc-idri.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lits.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/livecodebench.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ljspeech.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/llava-instruct-150k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/longbench.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/lsst-alert.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/luna16.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lvis.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/lvos.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/m4.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/m5.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/maestro.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/magnatagatune.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mapillary-vistas.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/market-1501.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/matbench.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/materials-project.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/math.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mathqa.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/matterport3d.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mbpp.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/md17.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/medical-segmentation-decathlon.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/medleydb.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/metaworld.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/metr-la.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mhealth.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/middlebury.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/million-aid.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mimic-cxr.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mimic-iii.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mind.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/minerl-obtaindiamond.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/minigrid.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mit-bih.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mit-indoor67.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mjsynth.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mls.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mmlu-pro.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mmlu.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mmwave-har.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/modelnet40.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/moisesdb.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/moleculenet-8.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/moments-in-time.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/monash-30.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/monuseg.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mose.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mot17.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mot20.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/movielens-1m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/movielens-20m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/movielens-25m.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mpdd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mpi-inf-3dhp.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mpii.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ms-asl.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ms-marco.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/msd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/msmt17.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/msr-vtt.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mstar.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/msvd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mtbs.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mteb.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/multi-news.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mura.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/musdb18-hq.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/musiccaps.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/musicnet.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mutag.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mvimgnet.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/mvsec.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mvtec-ad.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mvtec-loco.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/mvtec.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/n-caltech101.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/n-cmapss.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/n2c2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nab.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nabirds.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/nas-bench-301.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/natural-questions.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nct-crc-he.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nell-995.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nerf-synthetic.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/netflix-prize.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/nethack.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/neu-det.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nico.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nih-14.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nllb-seed.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nmnist.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/noaa-oisst.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nsl-kdd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nsynth.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ntu-rgbd.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/nuscenes.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nwpu-resisc45.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nyc-taxi.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/nyu-depth-v2.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/objaverse.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/objectnet.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/objects365.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/oc20.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/oc22.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/octa-500.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/office-home.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbg-molhiv.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbg-pcba.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbg-ppa.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbl-collab.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbl-ddi.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbl-ppa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbn-arxiv.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbn-mag.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbn-papers100m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbn-products.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ogbn-proteins.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ok-vqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/once.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/open-bandit.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/openhermes.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/openimages-v7.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/openmic-2018.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/openml-cc18.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/openwebtext.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/opportunity.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/opus-100.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/oqmd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/orcadata.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/otb100.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/otto.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/outbrain-click-prediction.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/oxford-flowers-102.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pacs.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/padchest.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pamap2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pannuke.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/paracrawl.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/pascal-context.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/pascal-voc-07.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pascal-voc-12.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pastis.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pcam.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pcb-defect.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pcqm4mv2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pdbbind.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pdebench.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pems-03.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pems-04.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pems-07.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pems-08.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pems-bay.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/penn-treebank.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/peoples-speech.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/phakir.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/phm-milling.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/physionet-2012.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/places365.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/plant-pathology-2020.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/plantdoc.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/plantvillage.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/porto-seguro.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/posetrack21.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/potsdam.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/powergridworld.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ppi.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/procgen.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/prophesee-gen1.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/proteinnet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/proteins.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/publaynet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pubmed.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pubmedqa.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/pubtabnet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/pv-el.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/qm7b.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/qm9.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/qmnist.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/race.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/radar-doppler.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/real-toxicity-prompts.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/reddit.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/redpajama-v2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ref-youtube-vos.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/refcoco-plus.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/refcoco.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/refcocog.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/replica.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/retailrocket-ctr.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/retailrocket.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/retinal-vessels.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/rl-unplugged.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/rlbench.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/rmd17.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/robomimic.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/rossmann-store-sales.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/rsna-pneumonia.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/rvl-cdip.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/s3dis.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/samsum.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/santander.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sbu-captions.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/scannet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/scedc.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sdss.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sen12ms.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/shanghaitech.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/shapenet-part.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/shipsear.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/shl.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/siim-acr-pneumothorax.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sintel.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sketchy.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/slakh2100.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/sleep-edf.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/slimpajama.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/smac.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/smacv2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/smap-msl.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/smd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sms-wsj.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/snli.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/solar.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/something-something-v2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sonyc-ust.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/spacenet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/spgispeech.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/spotify-mpd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/spring.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/squad-2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/squad.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sroie.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ssdd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/stanford-cars.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/stanford-dogs.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/steam.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/stl-10.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sugarbeets-2016.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sun-rgbd.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/sun397.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/superglue.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/sustainbench.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/susy.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/svamp.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/svhn.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/swat-wadi.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/switchboard.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/synthia.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/synthtext.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/tabarena.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/tabula-sapiens.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/tao.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/tape.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/tartanair.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/tcga.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ted2020.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/tedlium-3.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/tenrec.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/terra-incognita.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/tess.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/textvqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/tgb.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/the-pile.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/theoremqa.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/things3d.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/timit.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/tiny-imagenet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/titanic.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/totalsegmentator.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/touch-and-go.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/trackingnet.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/traffic.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/transnas-bench-101.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/triviaqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/truthfulqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/twitch-gamers.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/tydiqa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/uav123.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/uc-merced.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ucf-crime.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ucf101.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ucr-anomaly-archive.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ucr.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ucsd-ped.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/uea.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/uecfood256.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ujiindoorloc.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/uk-biobank.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/uk-dale.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/ultrachat.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/ultrafeedback.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/umls.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/un-parallel.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/urbansound8k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/uwb-nlos.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/vaihingen.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/vctk.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/veri-776.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/vggface2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/vima.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/visa.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/visda-2017.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/visual-genome.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/voicebank-demand.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/voxceleb-1.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/voxceleb-2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/voxlingua107.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/voxpopuli.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/vqav2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/waterbirds.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wavcaps.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/waymo-open.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/weather.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/weatherbench-2.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wham.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/whamr.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/whu-hi.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/widar-3.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/widar.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wider-face.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/wikipedia-web-traffic.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wikitext-103.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/wilds-fmow.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wilds-rxrx1.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wind.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wisdm.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wit.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wmt.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/wsj.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/wsj0-2mix.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/xbd.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/xeno-canto.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/xnli.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/xsum.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/xtreme.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/xview.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/yahoo-s5.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/yelp.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/yfcc100m.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/youtube-vis.json +12 -0
- topologies-0.0.1/src/topologies/datasets/records/youtube-vos.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/zinc-12k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/zinc-250k.json +13 -0
- topologies-0.0.1/src/topologies/datasets/records/zinc-full.json +13 -0
- topologies-0.0.1/src/topologies/datasets/schema.py +278 -0
- topologies-0.0.1/src/topologies/enums.py +80 -0
- topologies-0.0.1/src/topologies/evaluation/__init__.py +58 -0
- topologies-0.0.1/src/topologies/evaluation/api.py +1162 -0
- topologies-0.0.1/src/topologies/evaluation/metrics.py +438 -0
- topologies-0.0.1/src/topologies/experiments/__init__.py +181 -0
- topologies-0.0.1/src/topologies/experiments/analysis.py +645 -0
- topologies-0.0.1/src/topologies/experiments/bench.py +435 -0
- topologies-0.0.1/src/topologies/experiments/bench_checks.py +251 -0
- topologies-0.0.1/src/topologies/experiments/cache.py +229 -0
- topologies-0.0.1/src/topologies/experiments/callbacks.py +163 -0
- topologies-0.0.1/src/topologies/experiments/cli.py +528 -0
- topologies-0.0.1/src/topologies/experiments/coordinator.py +387 -0
- topologies-0.0.1/src/topologies/experiments/core.py +499 -0
- topologies-0.0.1/src/topologies/experiments/costs.py +180 -0
- topologies-0.0.1/src/topologies/experiments/distributions.py +133 -0
- topologies-0.0.1/src/topologies/experiments/errors.py +58 -0
- topologies-0.0.1/src/topologies/experiments/insights.py +352 -0
- topologies-0.0.1/src/topologies/experiments/metrics.py +134 -0
- topologies-0.0.1/src/topologies/experiments/parallel.py +294 -0
- topologies-0.0.1/src/topologies/experiments/pipeline.py +592 -0
- topologies-0.0.1/src/topologies/experiments/planning.py +289 -0
- topologies-0.0.1/src/topologies/experiments/pruners.py +86 -0
- topologies-0.0.1/src/topologies/experiments/render/__init__.py +1 -0
- topologies-0.0.1/src/topologies/experiments/render/console.py +489 -0
- topologies-0.0.1/src/topologies/experiments/render/html.py +512 -0
- topologies-0.0.1/src/topologies/experiments/render/live.py +130 -0
- topologies-0.0.1/src/topologies/experiments/render/markdown.py +73 -0
- topologies-0.0.1/src/topologies/experiments/results.py +694 -0
- topologies-0.0.1/src/topologies/experiments/routing.py +510 -0
- topologies-0.0.1/src/topologies/experiments/runner.py +233 -0
- topologies-0.0.1/src/topologies/experiments/space.py +513 -0
- topologies-0.0.1/src/topologies/experiments/spec.py +221 -0
- topologies-0.0.1/src/topologies/experiments/stats.py +322 -0
- topologies-0.0.1/src/topologies/experiments/store.py +198 -0
- topologies-0.0.1/src/topologies/experiments/table.py +134 -0
- topologies-0.0.1/src/topologies/experiments/tasks.py +905 -0
- topologies-0.0.1/src/topologies/experiments/trial.py +425 -0
- topologies-0.0.1/src/topologies/experiments/tui.py +14 -0
- topologies-0.0.1/src/topologies/experiments/yaml_spec.py +221 -0
- topologies-0.0.1/src/topologies/generation.py +252 -0
- topologies-0.0.1/src/topologies/implementations/losses/a2c.py +53 -0
- topologies-0.0.1/src/topologies/implementations/losses/poly1_loss.py +43 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/adabelief.py +54 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/adafactor.py +66 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/adahessian.py +76 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/adam8bit.py +75 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/adam_mini.py +78 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/ademamix.py +59 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/asam.py +71 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/core.py +29 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/ema.py +49 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/fadam.py +58 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/fisher_sam.py +76 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/flora.py +41 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/galore.py +42 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/grokfast.py +46 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/gsam.py +67 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/kfac.py +43 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/lamb.py +63 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/lars.py +46 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/lion.py +49 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/lookahead.py +53 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/model_soup.py +50 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/muon.py +41 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/novograd.py +62 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/polynomial_decay.py +52 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/qlora.py +18 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/radam.py +71 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/relora.py +17 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/sam.py +78 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/schedule_free.py +38 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/shampoo.py +60 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/sm3.py +44 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/soap.py +54 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/sophia.py +61 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/swa.py +50 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/swad.py +52 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/swalp.py +43 -0
- topologies-0.0.1/src/topologies/implementations/optimizers/twa.py +50 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/clr.py +21 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/cooldown.py +37 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/cosine_min.py +19 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/cosine_restarts.py +24 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/exponential.py +32 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/grafted.py +43 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/multistep.py +21 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/noam.py +39 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/onecycle.py +10 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/reduce_on_plateau.py +22 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/sgdr.py +24 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/trapezoidal.py +42 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/warmup_linear.py +32 -0
- topologies-0.0.1/src/topologies/implementations/schedulers/wsd.py +46 -0
- topologies-0.0.1/src/topologies/implementations/training/core.py +298 -0
- topologies-0.0.1/src/topologies/inference.py +88 -0
- topologies-0.0.1/src/topologies/inference_tasks/__init__.py +1 -0
- topologies-0.0.1/src/topologies/inference_tasks/audio.py +26 -0
- topologies-0.0.1/src/topologies/inference_tasks/detection.py +63 -0
- topologies-0.0.1/src/topologies/inference_tasks/diffusion.py +114 -0
- topologies-0.0.1/src/topologies/inference_tasks/music.py +34 -0
- topologies-0.0.1/src/topologies/inference_tasks/retrieval.py +19 -0
- topologies-0.0.1/src/topologies/introspection/__init__.py +84 -0
- topologies-0.0.1/src/topologies/introspection/_tree.py +111 -0
- topologies-0.0.1/src/topologies/introspection/adapters/__init__.py +159 -0
- topologies-0.0.1/src/topologies/introspection/adapters/builtins.py +20 -0
- topologies-0.0.1/src/topologies/introspection/adapters/diffusion.py +41 -0
- topologies-0.0.1/src/topologies/introspection/adapters/generic.py +17 -0
- topologies-0.0.1/src/topologies/introspection/adapters/huggingface.py +96 -0
- topologies-0.0.1/src/topologies/introspection/adapters/topologies_catalog.py +212 -0
- topologies-0.0.1/src/topologies/introspection/adapters/torch_modules.py +34 -0
- topologies-0.0.1/src/topologies/introspection/api.py +303 -0
- topologies-0.0.1/src/topologies/introspection/capture.py +580 -0
- topologies-0.0.1/src/topologies/introspection/cli.py +70 -0
- topologies-0.0.1/src/topologies/introspection/collection.py +457 -0
- topologies-0.0.1/src/topologies/introspection/errors.py +35 -0
- topologies-0.0.1/src/topologies/introspection/functional.py +339 -0
- topologies-0.0.1/src/topologies/introspection/hooks.py +262 -0
- topologies-0.0.1/src/topologies/introspection/html.py +122 -0
- topologies-0.0.1/src/topologies/introspection/inputs.py +180 -0
- topologies-0.0.1/src/topologies/introspection/integrations.py +110 -0
- topologies-0.0.1/src/topologies/introspection/interventions.py +388 -0
- topologies-0.0.1/src/topologies/introspection/objectives.py +261 -0
- topologies-0.0.1/src/topologies/introspection/payloads.py +116 -0
- topologies-0.0.1/src/topologies/introspection/reducers.py +755 -0
- topologies-0.0.1/src/topologies/introspection/report.py +454 -0
- topologies-0.0.1/src/topologies/introspection/semantics.py +218 -0
- topologies-0.0.1/src/topologies/introspection/session.py +386 -0
- topologies-0.0.1/src/topologies/introspection/sites.py +501 -0
- topologies-0.0.1/src/topologies/introspection/store.py +148 -0
- topologies-0.0.1/src/topologies/introspection/subject.py +236 -0
- topologies-0.0.1/src/topologies/introspection/techniques/__init__.py +220 -0
- topologies-0.0.1/src/topologies/introspection/techniques/_features.py +109 -0
- topologies-0.0.1/src/topologies/introspection/techniques/activations.py +215 -0
- topologies-0.0.1/src/topologies/introspection/techniques/anomalies.py +224 -0
- topologies-0.0.1/src/topologies/introspection/techniques/attention.py +280 -0
- topologies-0.0.1/src/topologies/introspection/techniques/attribution.py +450 -0
- topologies-0.0.1/src/topologies/introspection/techniques/dictionary.py +205 -0
- topologies-0.0.1/src/topologies/introspection/techniques/geometry.py +124 -0
- topologies-0.0.1/src/topologies/introspection/techniques/gradients.py +144 -0
- topologies-0.0.1/src/topologies/introspection/techniques/interventions.py +447 -0
- topologies-0.0.1/src/topologies/introspection/techniques/lens.py +173 -0
- topologies-0.0.1/src/topologies/introspection/techniques/maximization.py +191 -0
- topologies-0.0.1/src/topologies/introspection/techniques/neurons.py +206 -0
- topologies-0.0.1/src/topologies/introspection/techniques/probing.py +224 -0
- topologies-0.0.1/src/topologies/introspection/techniques/process.py +261 -0
- topologies-0.0.1/src/topologies/introspection/techniques/similarity.py +176 -0
- topologies-0.0.1/src/topologies/introspection/techniques/structure.py +246 -0
- topologies-0.0.1/src/topologies/introspection/techniques/weights.py +156 -0
- topologies-0.0.1/src/topologies/introspection/trace.py +324 -0
- topologies-0.0.1/src/topologies/introspection/views.py +168 -0
- topologies-0.0.1/src/topologies/layers/__init__.py +0 -0
- topologies-0.0.1/src/topologies/layers/alphafold.py +27 -0
- topologies-0.0.1/src/topologies/layers/audio_frontend.py +138 -0
- topologies-0.0.1/src/topologies/layers/audio_perturb.py +61 -0
- topologies-0.0.1/src/topologies/layers/augmix.py +62 -0
- topologies-0.0.1/src/topologies/layers/auprc.py +26 -0
- topologies-0.0.1/src/topologies/layers/autoformer.py +26 -0
- topologies-0.0.1/src/topologies/layers/awp.py +55 -0
- topologies-0.0.1/src/topologies/layers/bcq.py +40 -0
- topologies-0.0.1/src/topologies/layers/bevformer.py +111 -0
- topologies-0.0.1/src/topologies/layers/bic.py +17 -0
- topologies-0.0.1/src/topologies/layers/bidirectional_rnn_wrapper.py +34 -0
- topologies-0.0.1/src/topologies/layers/bifpn.py +28 -0
- topologies-0.0.1/src/topologies/layers/bigbird_attention.py +52 -0
- topologies-0.0.1/src/topologies/layers/bigvgan.py +23 -0
- topologies-0.0.1/src/topologies/layers/box_coder.py +53 -0
- topologies-0.0.1/src/topologies/layers/causal_temporal_block.py +22 -0
- topologies-0.0.1/src/topologies/layers/cbam_gate.py +43 -0
- topologies-0.0.1/src/topologies/layers/chemformer.py +79 -0
- topologies-0.0.1/src/topologies/layers/chunked_cross_attention.py +47 -0
- topologies-0.0.1/src/topologies/layers/cluster_gcn.py +59 -0
- topologies-0.0.1/src/topologies/layers/cnp.py +15 -0
- topologies-0.0.1/src/topologies/layers/complex.py +44 -0
- topologies-0.0.1/src/topologies/layers/conformer.py +59 -0
- topologies-0.0.1/src/topologies/layers/contrastive_search.py +45 -0
- topologies-0.0.1/src/topologies/layers/conv_blocks.py +26 -0
- topologies-0.0.1/src/topologies/layers/coverage.py +11 -0
- topologies-0.0.1/src/topologies/layers/cpvt.py +41 -0
- topologies-0.0.1/src/topologies/layers/crepe.py +43 -0
- topologies-0.0.1/src/topologies/layers/crestereo.py +88 -0
- topologies-0.0.1/src/topologies/layers/cvt.py +36 -0
- topologies-0.0.1/src/topologies/layers/ddim.py +21 -0
- topologies-0.0.1/src/topologies/layers/deeplob.py +111 -0
- topologies-0.0.1/src/topologies/layers/depth_mu_p.py +45 -0
- topologies-0.0.1/src/topologies/layers/diagonal_ssm.py +33 -0
- topologies-0.0.1/src/topologies/layers/diffusion_policy.py +121 -0
- topologies-0.0.1/src/topologies/layers/diffusion_residual.py +27 -0
- topologies-0.0.1/src/topologies/layers/distmult.py +23 -0
- topologies-0.0.1/src/topologies/layers/drop_path.py +19 -0
- topologies-0.0.1/src/topologies/layers/dsp_utils.py +79 -0
- topologies-0.0.1/src/topologies/layers/dueling_q_head.py +16 -0
- topologies-0.0.1/src/topologies/layers/edgenext.py +53 -0
- topologies-0.0.1/src/topologies/layers/edm_heun.py +26 -0
- topologies-0.0.1/src/topologies/layers/efficientad.py +11 -0
- topologies-0.0.1/src/topologies/layers/efficientdet.py +32 -0
- topologies-0.0.1/src/topologies/layers/feature_pyramid.py +22 -0
- topologies-0.0.1/src/topologies/layers/feed_forward.py +27 -0
- topologies-0.0.1/src/topologies/layers/film.py +42 -0
- topologies-0.0.1/src/topologies/layers/fir_filter.py +22 -0
- topologies-0.0.1/src/topologies/layers/flownet2.py +51 -0
- topologies-0.0.1/src/topologies/layers/fm_interaction.py +11 -0
- topologies-0.0.1/src/topologies/layers/fsq.py +24 -0
- topologies-0.0.1/src/topologies/layers/ft_transformer.py +21 -0
- topologies-0.0.1/src/topologies/layers/gail.py +33 -0
- topologies-0.0.1/src/topologies/layers/galerkin_attention.py +24 -0
- topologies-0.0.1/src/topologies/layers/galore.py +18 -0
- topologies-0.0.1/src/topologies/layers/geglu.py +17 -0
- topologies-0.0.1/src/topologies/layers/geglu_ffn.py +21 -0
- topologies-0.0.1/src/topologies/layers/geneformer.py +45 -0
- topologies-0.0.1/src/topologies/layers/ghostnet.py +44 -0
- topologies-0.0.1/src/topologies/layers/git.py +55 -0
- topologies-0.0.1/src/topologies/layers/gmt.py +44 -0
- topologies-0.0.1/src/topologies/layers/gp_bo.py +39 -0
- topologies-0.0.1/src/topologies/layers/graph_convolution.py +64 -0
- topologies-0.0.1/src/topologies/layers/graph_readout.py +32 -0
- topologies-0.0.1/src/topologies/layers/groupnorm_block.py +21 -0
- topologies-0.0.1/src/topologies/layers/hard_swish.py +8 -0
- topologies-0.0.1/src/topologies/layers/harmonic_oscillator.py +39 -0
- topologies-0.0.1/src/topologies/layers/hole.py +37 -0
- topologies-0.0.1/src/topologies/layers/hstu.py +42 -0
- topologies-0.0.1/src/topologies/layers/hyena_block.py +51 -0
- topologies-0.0.1/src/topologies/layers/hyena_operator.py +36 -0
- topologies-0.0.1/src/topologies/layers/i3d.py +17 -0
- topologies-0.0.1/src/topologies/layers/imap.py +37 -0
- topologies-0.0.1/src/topologies/layers/inception_resnet.py +17 -0
- topologies-0.0.1/src/topologies/layers/influence.py +20 -0
- topologies-0.0.1/src/topologies/layers/informer.py +22 -0
- topologies-0.0.1/src/topologies/layers/instancenorm_block.py +20 -0
- topologies-0.0.1/src/topologies/layers/instant_ngp.py +54 -0
- topologies-0.0.1/src/topologies/layers/integrated_gradients.py +26 -0
- topologies-0.0.1/src/topologies/layers/inverted_residual.py +21 -0
- topologies-0.0.1/src/topologies/layers/joint_energy.py +44 -0
- topologies-0.0.1/src/topologies/layers/jukebox.py +44 -0
- topologies-0.0.1/src/topologies/layers/k_planes.py +45 -0
- topologies-0.0.1/src/topologies/layers/kan.py +64 -0
- topologies-0.0.1/src/topologies/layers/kerple.py +34 -0
- topologies-0.0.1/src/topologies/layers/kpconv.py +61 -0
- topologies-0.0.1/src/topologies/layers/lars.py +28 -0
- topologies-0.0.1/src/topologies/layers/layer_scale.py +20 -0
- topologies-0.0.1/src/topologies/layers/layernorm.py +20 -0
- topologies-0.0.1/src/topologies/layers/layernorm_affine.py +18 -0
- topologies-0.0.1/src/topologies/layers/layoutlm.py +37 -0
- topologies-0.0.1/src/topologies/layers/leaf.py +60 -0
- topologies-0.0.1/src/topologies/layers/learned_position.py +22 -0
- topologies-0.0.1/src/topologies/layers/linear_attention.py +42 -0
- topologies-0.0.1/src/topologies/layers/linear_chain_crf.py +58 -0
- topologies-0.0.1/src/topologies/layers/linformer.py +37 -0
- topologies-0.0.1/src/topologies/layers/llava.py +19 -0
- topologies-0.0.1/src/topologies/layers/mamba_block.py +39 -0
- topologies-0.0.1/src/topologies/layers/masked_pooling.py +44 -0
- topologies-0.0.1/src/topologies/layers/memorizing_attention.py +86 -0
- topologies-0.0.1/src/topologies/layers/meshcnn.py +379 -0
- topologies-0.0.1/src/topologies/layers/minigpt4.py +18 -0
- topologies-0.0.1/src/topologies/layers/moirai.py +24 -0
- topologies-0.0.1/src/topologies/layers/multi_head_attention.py +59 -0
- topologies-0.0.1/src/topologies/layers/nearest_neighbor_retriever.py +23 -0
- topologies-0.0.1/src/topologies/layers/neuralucb_bonus.py +16 -0
- topologies-0.0.1/src/topologies/layers/nlu_heads.py +46 -0
- topologies-0.0.1/src/topologies/layers/node_ensemble.py +71 -0
- topologies-0.0.1/src/topologies/layers/noisy_nets.py +55 -0
- topologies-0.0.1/src/topologies/layers/paged_attention.py +103 -0
- topologies-0.0.1/src/topologies/layers/pali.py +52 -0
- topologies-0.0.1/src/topologies/layers/patch_embedding.py +25 -0
- topologies-0.0.1/src/topologies/layers/pcen.py +37 -0
- topologies-0.0.1/src/topologies/layers/perceiver_io.py +30 -0
- topologies-0.0.1/src/topologies/layers/performer.py +46 -0
- topologies-0.0.1/src/topologies/layers/pointcnn.py +62 -0
- topologies-0.0.1/src/topologies/layers/pointllm.py +25 -0
- topologies-0.0.1/src/topologies/layers/ppo_clip.py +19 -0
- topologies-0.0.1/src/topologies/layers/prelu.py +21 -0
- topologies-0.0.1/src/topologies/layers/projection_head.py +16 -0
- topologies-0.0.1/src/topologies/layers/qk_norm_attention.py +58 -0
- topologies-0.0.1/src/topologies/layers/quartznet.py +41 -0
- topologies-0.0.1/src/topologies/layers/quate.py +52 -0
- topologies-0.0.1/src/topologies/layers/randaugment.py +99 -0
- topologies-0.0.1/src/topologies/layers/rectified_flow.py +63 -0
- topologies-0.0.1/src/topologies/layers/recurrent_backbone.py +27 -0
- topologies-0.0.1/src/topologies/layers/replaced_token_detection_head.py +36 -0
- topologies-0.0.1/src/topologies/layers/residual_blocks.py +19 -0
- topologies-0.0.1/src/topologies/layers/resnet_v15.py +43 -0
- topologies-0.0.1/src/topologies/layers/rmsnorm.py +16 -0
- topologies-0.0.1/src/topologies/layers/rnn_transducer.py +42 -0
- topologies-0.0.1/src/topologies/layers/rotary_embedding.py +24 -0
- topologies-0.0.1/src/topologies/layers/rotate.py +46 -0
- topologies-0.0.1/src/topologies/layers/rwkv_channel_mix.py +30 -0
- topologies-0.0.1/src/topologies/layers/s3d.py +33 -0
- topologies-0.0.1/src/topologies/layers/s4_kernel.py +93 -0
- topologies-0.0.1/src/topologies/layers/saint.py +26 -0
- topologies-0.0.1/src/topologies/layers/score_sde.py +46 -0
- topologies-0.0.1/src/topologies/layers/seanet.py +43 -0
- topologies-0.0.1/src/topologies/layers/sentence_bert.py +21 -0
- topologies-0.0.1/src/topologies/layers/shufflenet_v2.py +27 -0
- topologies-0.0.1/src/topologies/layers/silu_activation.py +16 -0
- topologies-0.0.1/src/topologies/layers/simkd.py +28 -0
- topologies-0.0.1/src/topologies/layers/simple.py +33 -0
- topologies-0.0.1/src/topologies/layers/simplenet.py +47 -0
- topologies-0.0.1/src/topologies/layers/sinusoidal_position.py +22 -0
- topologies-0.0.1/src/topologies/layers/skip_upsample.py +20 -0
- topologies-0.0.1/src/topologies/layers/skipinit.py +18 -0
- topologies-0.0.1/src/topologies/layers/slot_attention.py +57 -0
- topologies-0.0.1/src/topologies/layers/snake.py +13 -0
- topologies-0.0.1/src/topologies/layers/snip.py +13 -0
- topologies-0.0.1/src/topologies/layers/source_separation.py +15 -0
- topologies-0.0.1/src/topologies/layers/speaker_embedding.py +22 -0
- topologies-0.0.1/src/topologies/layers/specaugment.py +65 -0
- topologies-0.0.1/src/topologies/layers/squeeze_excitation.py +18 -0
- topologies-0.0.1/src/topologies/layers/stargan_v2.py +65 -0
- topologies-0.0.1/src/topologies/layers/stft_frontend.py +27 -0
- topologies-0.0.1/src/topologies/layers/structure_module.py +111 -0
- topologies-0.0.1/src/topologies/layers/stylegan2.py +57 -0
- topologies-0.0.1/src/topologies/layers/surrogate_gradient.py +57 -0
- topologies-0.0.1/src/topologies/layers/tabtransformer.py +69 -0
- topologies-0.0.1/src/topologies/layers/temporal_shift.py +31 -0
- topologies-0.0.1/src/topologies/layers/tfn.py +60 -0
- topologies-0.0.1/src/topologies/layers/tide.py +136 -0
- topologies-0.0.1/src/topologies/layers/top_2_router.py +46 -0
- topologies-0.0.1/src/topologies/layers/topk_moe_layer.py +63 -0
- topologies-0.0.1/src/topologies/layers/transducer_asr.py +53 -0
- topologies-0.0.1/src/topologies/layers/transformer_block.py +66 -0
- topologies-0.0.1/src/topologies/layers/transh.py +26 -0
- topologies-0.0.1/src/topologies/layers/tsmixer.py +60 -0
- topologies-0.0.1/src/topologies/layers/ttt_titans_mblm.py +50 -0
- topologies-0.0.1/src/topologies/layers/tucker.py +40 -0
- topologies-0.0.1/src/topologies/layers/unet3plus.py +60 -0
- topologies-0.0.1/src/topologies/layers/unetpp.py +42 -0
- topologies-0.0.1/src/topologies/layers/v_trace.py +82 -0
- topologies-0.0.1/src/topologies/layers/vector_quantizer.py +25 -0
- topologies-0.0.1/src/topologies/layers/waveglow.py +45 -0
- topologies-0.0.1/src/topologies/layers/whisper.py +54 -0
- topologies-0.0.1/src/topologies/losses/catalog.json +529 -0
- topologies-0.0.1/src/topologies/metadata/__init__.py +30 -0
- topologies-0.0.1/src/topologies/metadata/schema.py +446 -0
- topologies-0.0.1/src/topologies/metrics.py +206 -0
- topologies-0.0.1/src/topologies/models/__init__.py +0 -0
- topologies-0.0.1/src/topologies/models/actor_critic.py +33 -0
- topologies-0.0.1/src/topologies/models/alexnet.py +57 -0
- topologies-0.0.1/src/topologies/models/attention_nmt.py +93 -0
- topologies-0.0.1/src/topologies/models/attn2d.py +68 -0
- topologies-0.0.1/src/topologies/models/audio_crnn.py +79 -0
- topologies-0.0.1/src/topologies/models/autoencoders.py +106 -0
- topologies-0.0.1/src/topologies/models/autoint.py +50 -0
- topologies-0.0.1/src/topologies/models/awr.py +46 -0
- topologies-0.0.1/src/topologies/models/bart.py +121 -0
- topologies-0.0.1/src/topologies/models/bert4rec.py +70 -0
- topologies-0.0.1/src/topologies/models/bert_encoder.py +845 -0
- topologies-0.0.1/src/topologies/models/bigbird.py +95 -0
- topologies-0.0.1/src/topologies/models/biggan.py +407 -0
- topologies-0.0.1/src/topologies/models/blip2.py +37 -0
- topologies-0.0.1/src/topologies/models/bst.py +41 -0
- topologies-0.0.1/src/topologies/models/byt5.py +204 -0
- topologies-0.0.1/src/topologies/models/c3d.py +27 -0
- topologies-0.0.1/src/topologies/models/canine.py +221 -0
- topologies-0.0.1/src/topologies/models/cascade_rcnn.py +123 -0
- topologies-0.0.1/src/topologies/models/caser.py +51 -0
- topologies-0.0.1/src/topologies/models/centernet.py +26 -0
- topologies-0.0.1/src/topologies/models/cgan.py +52 -0
- topologies-0.0.1/src/topologies/models/charformer.py +272 -0
- topologies-0.0.1/src/topologies/models/compact_convnet.py +38 -0
- topologies-0.0.1/src/topologies/models/consistency_model.py +103 -0
- topologies-0.0.1/src/topologies/models/contrastive_vision.py +90 -0
- topologies-0.0.1/src/topologies/models/conv_asr.py +59 -0
- topologies-0.0.1/src/topologies/models/convnext.py +88 -0
- topologies-0.0.1/src/topologies/models/ctc_acoustic_model.py +63 -0
- topologies-0.0.1/src/topologies/models/cyclegan.py +41 -0
- topologies-0.0.1/src/topologies/models/dalle.py +56 -0
- topologies-0.0.1/src/topologies/models/dbn.py +67 -0
- topologies-0.0.1/src/topologies/models/dcgan.py +65 -0
- topologies-0.0.1/src/topologies/models/dcn_v2.py +40 -0
- topologies-0.0.1/src/topologies/models/dcrnn.py +160 -0
- topologies-0.0.1/src/topologies/models/ddsp_synthesizer.py +51 -0
- topologies-0.0.1/src/topologies/models/deberta.py +344 -0
- topologies-0.0.1/src/topologies/models/decision_transformer.py +67 -0
- topologies-0.0.1/src/topologies/models/deepfm.py +36 -0
- topologies-0.0.1/src/topologies/models/deis.py +25 -0
- topologies-0.0.1/src/topologies/models/deit.py +69 -0
- topologies-0.0.1/src/topologies/models/densenet.py +85 -0
- topologies-0.0.1/src/topologies/models/detect_ops.py +101 -0
- topologies-0.0.1/src/topologies/models/detr.py +53 -0
- topologies-0.0.1/src/topologies/models/diagonal_ssm_model.py +38 -0
- topologies-0.0.1/src/topologies/models/diffusion_unet.py +94 -0
- topologies-0.0.1/src/topologies/models/dinov2.py +53 -0
- topologies-0.0.1/src/topologies/models/dip_vae.py +65 -0
- topologies-0.0.1/src/topologies/models/distributional_q.py +119 -0
- topologies-0.0.1/src/topologies/models/dit.py +109 -0
- topologies-0.0.1/src/topologies/models/dlinear.py +47 -0
- topologies-0.0.1/src/topologies/models/dpm_solver.py +31 -0
- topologies-0.0.1/src/topologies/models/dqn_family.py +33 -0
- topologies-0.0.1/src/topologies/models/dreamerv2.py +55 -0
- topologies-0.0.1/src/topologies/models/dreamerv3.py +52 -0
- topologies-0.0.1/src/topologies/models/droq.py +49 -0
- topologies-0.0.1/src/topologies/models/dss_s5_mamba.py +156 -0
- topologies-0.0.1/src/topologies/models/e_branchformer.py +51 -0
- topologies-0.0.1/src/topologies/models/ease.py +48 -0
- topologies-0.0.1/src/topologies/models/ecapa_tdnn.py +40 -0
- topologies-0.0.1/src/topologies/models/efficient_attention.py +94 -0
- topologies-0.0.1/src/topologies/models/efficient_conv.py +102 -0
- topologies-0.0.1/src/topologies/models/fastpitch.py +41 -0
- topologies-0.0.1/src/topologies/models/fastspeech2.py +45 -0
- topologies-0.0.1/src/topologies/models/ffjord.py +48 -0
- topologies-0.0.1/src/topologies/models/fibinet.py +53 -0
- topologies-0.0.1/src/topologies/models/fits.py +34 -0
- topologies-0.0.1/src/topologies/models/flow_matching.py +36 -0
- topologies-0.0.1/src/topologies/models/fno.py +38 -0
- topologies-0.0.1/src/topologies/models/fsq.py +22 -0
- topologies-0.0.1/src/topologies/models/fullsubnet.py +60 -0
- topologies-0.0.1/src/topologies/models/gatv2.py +50 -0
- topologies-0.0.1/src/topologies/models/gemma.py +189 -0
- topologies-0.0.1/src/topologies/models/glow_tts.py +109 -0
- topologies-0.0.1/src/topologies/models/gmlp.py +57 -0
- topologies-0.0.1/src/topologies/models/gmt.py +44 -0
- topologies-0.0.1/src/topologies/models/gpt2.py +120 -0
- topologies-0.0.1/src/topologies/models/gpt_decoder_family.py +297 -0
- topologies-0.0.1/src/topologies/models/gpt_decoder_lm.py +71 -0
- topologies-0.0.1/src/topologies/models/gpt_j_neox_pythia.py +135 -0
- topologies-0.0.1/src/topologies/models/grace.py +73 -0
- topologies-0.0.1/src/topologies/models/grad_tts.py +42 -0
- topologies-0.0.1/src/topologies/models/graph_network.py +61 -0
- topologies-0.0.1/src/topologies/models/graph_wavenet.py +34 -0
- topologies-0.0.1/src/topologies/models/graphsage.py +57 -0
- topologies-0.0.1/src/topologies/models/graphsaint.py +11 -0
- topologies-0.0.1/src/topologies/models/han.py +10 -0
- topologies-0.0.1/src/topologies/models/hnn.py +38 -0
- topologies-0.0.1/src/topologies/models/hornet.py +29 -0
- topologies-0.0.1/src/topologies/models/hyena.py +71 -0
- topologies-0.0.1/src/topologies/models/i_jepa.py +32 -0
- topologies-0.0.1/src/topologies/models/iaf.py +57 -0
- topologies-0.0.1/src/topologies/models/icm.py +45 -0
- topologies-0.0.1/src/topologies/models/image_text_dual_encoder.py +77 -0
- topologies-0.0.1/src/topologies/models/imagegpt.py +53 -0
- topologies-0.0.1/src/topologies/models/imap.py +38 -0
- topologies-0.0.1/src/topologies/models/inception.py +94 -0
- topologies-0.0.1/src/topologies/models/infovae.py +65 -0
- topologies-0.0.1/src/topologies/models/ipa.py +69 -0
- topologies-0.0.1/src/topologies/models/ipo.py +31 -0
- topologies-0.0.1/src/topologies/models/iqn.py +52 -0
- topologies-0.0.1/src/topologies/models/item_knn.py +26 -0
- topologies-0.0.1/src/topologies/models/itransformer.py +31 -0
- topologies-0.0.1/src/topologies/models/jamba_griffin_hawk.py +155 -0
- topologies-0.0.1/src/topologies/models/kto.py +38 -0
- topologies-0.0.1/src/topologies/models/ladder_vae.py +66 -0
- topologies-0.0.1/src/topologies/models/latent_diffusion.py +303 -0
- topologies-0.0.1/src/topologies/models/lenet.py +42 -0
- topologies-0.0.1/src/topologies/models/lfq.py +22 -0
- topologies-0.0.1/src/topologies/models/lightgcn.py +107 -0
- topologies-0.0.1/src/topologies/models/line.py +45 -0
- topologies-0.0.1/src/topologies/models/llama.py +171 -0
- topologies-0.0.1/src/topologies/models/llama_decoder.py +84 -0
- topologies-0.0.1/src/topologies/models/lnn.py +59 -0
- topologies-0.0.1/src/topologies/models/longformer.py +106 -0
- topologies-0.0.1/src/topologies/models/lstnet.py +45 -0
- topologies-0.0.1/src/topologies/models/maddpg.py +50 -0
- topologies-0.0.1/src/topologies/models/maf.py +57 -0
- topologies-0.0.1/src/topologies/models/magvit.py +89 -0
- topologies-0.0.1/src/topologies/models/mamba2_gated_deltanet_h3.py +122 -0
- topologies-0.0.1/src/topologies/models/mappo.py +60 -0
- topologies-0.0.1/src/topologies/models/marian.py +105 -0
- topologies-0.0.1/src/topologies/models/mask2former.py +46 -0
- topologies-0.0.1/src/topologies/models/mask_rcnn.py +26 -0
- topologies-0.0.1/src/topologies/models/masked_image.py +62 -0
- topologies-0.0.1/src/topologies/models/maskfeat.py +80 -0
- topologies-0.0.1/src/topologies/models/maskgit.py +87 -0
- topologies-0.0.1/src/topologies/models/matcha_tts.py +55 -0
- topologies-0.0.1/src/topologies/models/matchboxnet.py +36 -0
- topologies-0.0.1/src/topologies/models/matrix_factorization.py +31 -0
- topologies-0.0.1/src/topologies/models/mistral.py +153 -0
- topologies-0.0.1/src/topologies/models/mistral_decoder.py +103 -0
- topologies-0.0.1/src/topologies/models/mixtral.py +180 -0
- topologies-0.0.1/src/topologies/models/mlp.py +50 -0
- topologies-0.0.1/src/topologies/models/mobilenet_v2.py +60 -0
- topologies-0.0.1/src/topologies/models/modern_conv.py +99 -0
- topologies-0.0.1/src/topologies/models/modernbert.py +182 -0
- topologies-0.0.1/src/topologies/models/mpo.py +47 -0
- topologies-0.0.1/src/topologies/models/mult_vae.py +48 -0
- topologies-0.0.1/src/topologies/models/music_transformer.py +61 -0
- topologies-0.0.1/src/topologies/models/musicgen.py +102 -0
- topologies-0.0.1/src/topologies/models/muzero.py +59 -0
- topologies-0.0.1/src/topologies/models/mvit.py +42 -0
- topologies-0.0.1/src/topologies/models/nbeats.py +66 -0
- topologies-0.0.1/src/topologies/models/neural_ode.py +26 -0
- topologies-0.0.1/src/topologies/models/neural_spline_flows.py +78 -0
- topologies-0.0.1/src/topologies/models/nextitnet.py +69 -0
- topologies-0.0.1/src/topologies/models/ngcf.py +60 -0
- topologies-0.0.1/src/topologies/models/nhits.py +68 -0
- topologies-0.0.1/src/topologies/models/nice.py +33 -0
- topologies-0.0.1/src/topologies/models/noam.py +32 -0
- topologies-0.0.1/src/topologies/models/node.py +14 -0
- topologies-0.0.1/src/topologies/models/node_ensemble.py +62 -0
- topologies-0.0.1/src/topologies/models/noisy_nets.py +54 -0
- topologies-0.0.1/src/topologies/models/nvae.py +50 -0
- topologies-0.0.1/src/topologies/models/ot_cfm.py +103 -0
- topologies-0.0.1/src/topologies/models/pairre.py +27 -0
- topologies-0.0.1/src/topologies/models/panoptic.py +68 -0
- topologies-0.0.1/src/topologies/models/phi.py +186 -0
- topologies-0.0.1/src/topologies/models/pinn.py +29 -0
- topologies-0.0.1/src/topologies/models/pinsage.py +55 -0
- topologies-0.0.1/src/topologies/models/pix2pix.py +68 -0
- topologies-0.0.1/src/topologies/models/pixart.py +42 -0
- topologies-0.0.1/src/topologies/models/pixelcnn.py +26 -0
- topologies-0.0.1/src/topologies/models/pna.py +56 -0
- topologies-0.0.1/src/topologies/models/pnn.py +42 -0
- topologies-0.0.1/src/topologies/models/point_cloud.py +57 -0
- topologies-0.0.1/src/topologies/models/point_transformer.py +45 -0
- topologies-0.0.1/src/topologies/models/point_transformer_v2.py +67 -0
- topologies-0.0.1/src/topologies/models/pointpillars.py +78 -0
- topologies-0.0.1/src/topologies/models/poolformer.py +49 -0
- topologies-0.0.1/src/topologies/models/pos2d.py +51 -0
- topologies-0.0.1/src/topologies/models/pose.py +62 -0
- topologies-0.0.1/src/topologies/models/ppo_clip.py +18 -0
- topologies-0.0.1/src/topologies/models/progan.py +39 -0
- topologies-0.0.1/src/topologies/models/prompt_seg.py +71 -0
- topologies-0.0.1/src/topologies/models/q_network.py +23 -0
- topologies-0.0.1/src/topologies/models/qmix.py +58 -0
- topologies-0.0.1/src/topologies/models/qr_dqn.py +56 -0
- topologies-0.0.1/src/topologies/models/qwen.py +136 -0
- topologies-0.0.1/src/topologies/models/rag_fid.py +90 -0
- topologies-0.0.1/src/topologies/models/realnvp.py +84 -0
- topologies-0.0.1/src/topologies/models/recurrent_models.py +102 -0
- topologies-0.0.1/src/topologies/models/regnet.py +75 -0
- topologies-0.0.1/src/topologies/models/resmlp.py +72 -0
- topologies-0.0.1/src/topologies/models/resnet.py +80 -0
- topologies-0.0.1/src/topologies/models/resnet_v1d.py +79 -0
- topologies-0.0.1/src/topologies/models/resnet_variants.py +102 -0
- topologies-0.0.1/src/topologies/models/retinanet.py +120 -0
- topologies-0.0.1/src/topologies/models/retnet.py +111 -0
- topologies-0.0.1/src/topologies/models/retro.py +90 -0
- topologies-0.0.1/src/topologies/models/reversible_residual.py +18 -0
- topologies-0.0.1/src/topologies/models/rgcn.py +139 -0
- topologies-0.0.1/src/topologies/models/rloo.py +32 -0
- topologies-0.0.1/src/topologies/models/rnd.py +19 -0
- topologies-0.0.1/src/topologies/models/rq_transformer.py +56 -0
- topologies-0.0.1/src/topologies/models/rq_vae.py +46 -0
- topologies-0.0.1/src/topologies/models/rwkv.py +132 -0
- topologies-0.0.1/src/topologies/models/s4.py +99 -0
- topologies-0.0.1/src/topologies/models/sac.py +98 -0
- topologies-0.0.1/src/topologies/models/sagan.py +37 -0
- topologies-0.0.1/src/topologies/models/sagpool.py +35 -0
- topologies-0.0.1/src/topologies/models/sasrec.py +59 -0
- topologies-0.0.1/src/topologies/models/segment.py +77 -0
- topologies-0.0.1/src/topologies/models/sentence_bert.py +21 -0
- topologies-0.0.1/src/topologies/models/sentence_encoder.py +38 -0
- topologies-0.0.1/src/topologies/models/sepformer.py +97 -0
- topologies-0.0.1/src/topologies/models/seq2seq.py +169 -0
- topologies-0.0.1/src/topologies/models/set_detect.py +72 -0
- topologies-0.0.1/src/topologies/models/setr.py +36 -0
- topologies-0.0.1/src/topologies/models/sgc.py +39 -0
- topologies-0.0.1/src/topologies/models/sgc_precompute.py +30 -0
- topologies-0.0.1/src/topologies/models/siglip.py +35 -0
- topologies-0.0.1/src/topologies/models/sign.py +28 -0
- topologies-0.0.1/src/topologies/models/sim.py +56 -0
- topologies-0.0.1/src/topologies/models/simclr.py +32 -0
- topologies-0.0.1/src/topologies/models/sincnet.py +54 -0
- topologies-0.0.1/src/topologies/models/single_stage_detect.py +79 -0
- topologies-0.0.1/src/topologies/models/sit.py +31 -0
- topologies-0.0.1/src/topologies/models/sngan.py +30 -0
- topologies-0.0.1/src/topologies/models/solov2.py +38 -0
- topologies-0.0.1/src/topologies/models/sortpool.py +49 -0
- topologies-0.0.1/src/topologies/models/soundstream.py +33 -0
- topologies-0.0.1/src/topologies/models/spade.py +36 -0
- topologies-0.0.1/src/topologies/models/spanbert.py +295 -0
- topologies-0.0.1/src/topologies/models/sr_gnn.py +88 -0
- topologies-0.0.1/src/topologies/models/static_embeddings.py +68 -0
- topologies-0.0.1/src/topologies/models/stylegan3.py +26 -0
- topologies-0.0.1/src/topologies/models/swin.py +99 -0
- topologies-0.0.1/src/topologies/models/swin_unet.py +146 -0
- topologies-0.0.1/src/topologies/models/t5.py +162 -0
- topologies-0.0.1/src/topologies/models/tabtransformer.py +68 -0
- topologies-0.0.1/src/topologies/models/tabular_mlp.py +104 -0
- topologies-0.0.1/src/topologies/models/tacotron2.py +64 -0
- topologies-0.0.1/src/topologies/models/tc_resnet.py +62 -0
- topologies-0.0.1/src/topologies/models/tcvae.py +51 -0
- topologies-0.0.1/src/topologies/models/td3.py +76 -0
- topologies-0.0.1/src/topologies/models/td3_bc.py +44 -0
- topologies-0.0.1/src/topologies/models/temporal_convnet.py +53 -0
- topologies-0.0.1/src/topologies/models/text_cnn.py +57 -0
- topologies-0.0.1/src/topologies/models/tf_gridnet.py +27 -0
- topologies-0.0.1/src/topologies/models/tft.py +290 -0
- topologies-0.0.1/src/topologies/models/token_mixer.py +73 -0
- topologies-0.0.1/src/topologies/models/transformer_models.py +174 -0
- topologies-0.0.1/src/topologies/models/transformer_xl.py +142 -0
- topologies-0.0.1/src/topologies/models/transitional_text.py +97 -0
- topologies-0.0.1/src/topologies/models/tsm.py +39 -0
- topologies-0.0.1/src/topologies/models/two_stage_detect.py +84 -0
- topologies-0.0.1/src/topologies/models/u_vit.py +52 -0
- topologies-0.0.1/src/topologies/models/ul2.py +204 -0
- topologies-0.0.1/src/topologies/models/unet.py +54 -0
- topologies-0.0.1/src/topologies/models/unipc.py +62 -0
- topologies-0.0.1/src/topologies/models/v_pred.py +24 -0
- topologies-0.0.1/src/topologies/models/vae_mlp.py +67 -0
- topologies-0.0.1/src/topologies/models/variational_autoencoders.py +129 -0
- topologies-0.0.1/src/topologies/models/vgg.py +60 -0
- topologies-0.0.1/src/topologies/models/video_conv.py +57 -0
- topologies-0.0.1/src/topologies/models/video_ssl.py +51 -0
- topologies-0.0.1/src/topologies/models/video_transformer.py +66 -0
- topologies-0.0.1/src/topologies/models/vision_neck.py +64 -0
- topologies-0.0.1/src/topologies/models/vision_norm.py +76 -0
- topologies-0.0.1/src/topologies/models/vision_ssl.py +93 -0
- topologies-0.0.1/src/topologies/models/vision_transformer.py +75 -0
- topologies-0.0.1/src/topologies/models/vit_22b.py +23 -0
- topologies-0.0.1/src/topologies/models/vit_encoder.py +88 -0
- topologies-0.0.1/src/topologies/models/volume_seg.py +55 -0
- topologies-0.0.1/src/topologies/models/voxel_detect.py +53 -0
- topologies-0.0.1/src/topologies/models/vqvae.py +70 -0
- topologies-0.0.1/src/topologies/models/wav2vec2.py +29 -0
- topologies-0.0.1/src/topologies/models/wavenet.py +80 -0
- topologies-0.0.1/src/topologies/models/wgan.py +32 -0
- topologies-0.0.1/src/topologies/models/wgan_gp.py +58 -0
- topologies-0.0.1/src/topologies/models/wide_and_deep.py +31 -0
- topologies-0.0.1/src/topologies/models/world_model.py +73 -0
- topologies-0.0.1/src/topologies/models/x3d.py +168 -0
- topologies-0.0.1/src/topologies/models/x_vector.py +27 -0
- topologies-0.0.1/src/topologies/models/xdeepfm.py +49 -0
- topologies-0.0.1/src/topologies/models/xlnet.py +261 -0
- topologies-0.0.1/src/topologies/models/xlstm.py +98 -0
- topologies-0.0.1/src/topologies/models/xmem.py +500 -0
- topologies-0.0.1/src/topologies/models/xtts.py +40 -0
- topologies-0.0.1/src/topologies/models/yolov3.py +45 -0
- topologies-0.0.1/src/topologies/models/yourtts.py +21 -0
- topologies-0.0.1/src/topologies/nn/__init__.py +73 -0
- topologies-0.0.1/src/topologies/nn/aux.py +51 -0
- topologies-0.0.1/src/topologies/nn/causal.py +401 -0
- topologies-0.0.1/src/topologies/nn/layers.py +317 -0
- topologies-0.0.1/src/topologies/nn/mixers.py +246 -0
- topologies-0.0.1/src/topologies/nn/models.py +552 -0
- topologies-0.0.1/src/topologies/nn/slots.py +230 -0
- topologies-0.0.1/src/topologies/optimization/__init__.py +95 -0
- topologies-0.0.1/src/topologies/optimization/_torch_ao.py +50 -0
- topologies-0.0.1/src/topologies/optimization/artifacts.py +396 -0
- topologies-0.0.1/src/topologies/optimization/contracts.py +90 -0
- topologies-0.0.1/src/topologies/optimization/distillation.py +632 -0
- topologies-0.0.1/src/topologies/optimization/integrations.py +256 -0
- topologies-0.0.1/src/topologies/optimization/pruning.py +416 -0
- topologies-0.0.1/src/topologies/optimization/quantization.py +1261 -0
- topologies-0.0.1/src/topologies/optimization/recommendations.py +165 -0
- topologies-0.0.1/src/topologies/optimization/reporting.py +91 -0
- topologies-0.0.1/src/topologies/optimization/torchao_pt2e.py +781 -0
- topologies-0.0.1/src/topologies/optimization/workflow.py +368 -0
- topologies-0.0.1/src/topologies/optimizers/catalog.json +136 -0
- topologies-0.0.1/src/topologies/pretrained.py +278 -0
- topologies-0.0.1/src/topologies/registry/__init__.py +6 -0
- topologies-0.0.1/src/topologies/registry/api.py +59 -0
- topologies-0.0.1/src/topologies/registry/index.py +165 -0
- topologies-0.0.1/src/topologies/registry/loader.py +75 -0
- topologies-0.0.1/src/topologies/registry/validate.py +470 -0
- topologies-0.0.1/src/topologies/synth/__init__.py +178 -0
- topologies-0.0.1/src/topologies/synth/attributes.py +570 -0
- topologies-0.0.1/src/topologies/synth/blueprints/__init__.py +65 -0
- topologies-0.0.1/src/topologies/synth/blueprints/_base.py +126 -0
- topologies-0.0.1/src/topologies/synth/blueprints/algorithmic.py +312 -0
- topologies-0.0.1/src/topologies/synth/blueprints/reasoning.py +42 -0
- topologies-0.0.1/src/topologies/synth/blueprints/structured.py +240 -0
- topologies-0.0.1/src/topologies/synth/blueprints/vision.py +88 -0
- topologies-0.0.1/src/topologies/synth/cli.py +501 -0
- topologies-0.0.1/src/topologies/synth/codecs.py +215 -0
- topologies-0.0.1/src/topologies/synth/core.py +1331 -0
- topologies-0.0.1/src/topologies/synth/counterfactuals.py +162 -0
- topologies-0.0.1/src/topologies/synth/ctor.py +110 -0
- topologies-0.0.1/src/topologies/synth/dataset.py +433 -0
- topologies-0.0.1/src/topologies/synth/dists.py +461 -0
- topologies-0.0.1/src/topologies/synth/dsp.py +352 -0
- topologies-0.0.1/src/topologies/synth/errors.py +68 -0
- topologies-0.0.1/src/topologies/synth/evaluate.py +659 -0
- topologies-0.0.1/src/topologies/synth/export.py +150 -0
- topologies-0.0.1/src/topologies/synth/fingerprint.py +124 -0
- topologies-0.0.1/src/topologies/synth/generators/__init__.py +0 -0
- topologies-0.0.1/src/topologies/synth/generators/augment.py +12 -0
- topologies-0.0.1/src/topologies/synth/generators/base.py +200 -0
- topologies-0.0.1/src/topologies/synth/generators/grammar.py +5 -0
- topologies-0.0.1/src/topologies/synth/generators/program.py +9 -0
- topologies-0.0.1/src/topologies/synth/introspect.py +107 -0
- topologies-0.0.1/src/topologies/synth/jsonutil.py +241 -0
- topologies-0.0.1/src/topologies/synth/langid.py +150 -0
- topologies-0.0.1/src/topologies/synth/lib/__init__.py +105 -0
- topologies-0.0.1/src/topologies/synth/lib/audio/__init__.py +768 -0
- topologies-0.0.1/src/topologies/synth/lib/graph/__init__.py +340 -0
- topologies-0.0.1/src/topologies/synth/lib/image/__init__.py +762 -0
- topologies-0.0.1/src/topologies/synth/lib/math/__init__.py +175 -0
- topologies-0.0.1/src/topologies/synth/lib/math/expr.py +191 -0
- topologies-0.0.1/src/topologies/synth/lib/process/__init__.py +231 -0
- topologies-0.0.1/src/topologies/synth/lib/random/__init__.py +240 -0
- topologies-0.0.1/src/topologies/synth/lib/seq/__init__.py +418 -0
- topologies-0.0.1/src/topologies/synth/lib/series/__init__.py +268 -0
- topologies-0.0.1/src/topologies/synth/lib/tabular/__init__.py +733 -0
- topologies-0.0.1/src/topologies/synth/lib/text/__init__.py +53 -0
- topologies-0.0.1/src/topologies/synth/lib/text/faker.py +227 -0
- topologies-0.0.1/src/topologies/synth/lib/text/formats.py +165 -0
- topologies-0.0.1/src/topologies/synth/lib/text/generate.py +346 -0
- topologies-0.0.1/src/topologies/synth/lib/text/grammar.py +189 -0
- topologies-0.0.1/src/topologies/synth/lib/text/legacy.py +118 -0
- topologies-0.0.1/src/topologies/synth/lib/text/noise.py +357 -0
- topologies-0.0.1/src/topologies/synth/lib/text/numbers.py +115 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/de/faker.yaml +143 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/de/keyboard.yaml +2 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/de/translit.yaml +6 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/en/faker.yaml +174 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/en/keyboard.yaml +3 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/en/lexicon.yaml +4 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/en/numbers.yaml +34 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/en/paradigms.yaml +1 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/en/pools.yaml +7 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/en/translit.yaml +1 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/faker.yaml +121 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/keyboard.yaml +3 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/lexicon.yaml +17 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/numbers.yaml +54 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/paradigms.yaml +20 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/pools.yaml +7 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs/uk/translit.yaml +37 -0
- topologies-0.0.1/src/topologies/synth/lib/text/packs.py +122 -0
- topologies-0.0.1/src/topologies/synth/lib/text/pools.py +58 -0
- topologies-0.0.1/src/topologies/synth/lib/text/slots.py +94 -0
- topologies-0.0.1/src/topologies/synth/lib/text/template.py +357 -0
- topologies-0.0.1/src/topologies/synth/limits.py +83 -0
- topologies-0.0.1/src/topologies/synth/loop.py +390 -0
- topologies-0.0.1/src/topologies/synth/ops/__init__.py +66 -0
- topologies-0.0.1/src/topologies/synth/ops/_callable.py +29 -0
- topologies-0.0.1/src/topologies/synth/ops/augment.py +171 -0
- topologies-0.0.1/src/topologies/synth/ops/balance.py +154 -0
- topologies-0.0.1/src/topologies/synth/ops/base.py +235 -0
- topologies-0.0.1/src/topologies/synth/ops/contamination.py +191 -0
- topologies-0.0.1/src/topologies/synth/ops/dedup.py +402 -0
- topologies-0.0.1/src/topologies/synth/ops/filters.py +186 -0
- topologies-0.0.1/src/topologies/synth/ops/label.py +230 -0
- topologies-0.0.1/src/topologies/synth/ops/parse.py +118 -0
- topologies-0.0.1/src/topologies/synth/ops/split.py +217 -0
- topologies-0.0.1/src/topologies/synth/ops/validate.py +95 -0
- topologies-0.0.1/src/topologies/synth/ops/verify.py +530 -0
- topologies-0.0.1/src/topologies/synth/overrides.py +225 -0
- topologies-0.0.1/src/topologies/synth/pipeline.py +1054 -0
- topologies-0.0.1/src/topologies/synth/planning.py +243 -0
- topologies-0.0.1/src/topologies/synth/pools.py +19 -0
- topologies-0.0.1/src/topologies/synth/previews.py +187 -0
- topologies-0.0.1/src/topologies/synth/quality/__init__.py +34 -0
- topologies-0.0.1/src/topologies/synth/quality/embed.py +87 -0
- topologies-0.0.1/src/topologies/synth/quality/fidelity.py +421 -0
- topologies-0.0.1/src/topologies/synth/quality/insights.py +202 -0
- topologies-0.0.1/src/topologies/synth/quality/intrinsic.py +406 -0
- topologies-0.0.1/src/topologies/synth/quality/privacy.py +85 -0
- topologies-0.0.1/src/topologies/synth/quality/report.py +356 -0
- topologies-0.0.1/src/topologies/synth/quality/slices.py +118 -0
- topologies-0.0.1/src/topologies/synth/quality/utility.py +363 -0
- topologies-0.0.1/src/topologies/synth/records.py +171 -0
- topologies-0.0.1/src/topologies/synth/report.py +195 -0
- topologies-0.0.1/src/topologies/synth/reviewing.py +332 -0
- topologies-0.0.1/src/topologies/synth/rng.py +225 -0
- topologies-0.0.1/src/topologies/synth/runtime.py +33 -0
- topologies-0.0.1/src/topologies/synth/sampling.py +613 -0
- topologies-0.0.1/src/topologies/synth/schema.py +268 -0
- topologies-0.0.1/src/topologies/synth/spec.py +366 -0
- topologies-0.0.1/src/topologies/synth/store.py +374 -0
- topologies-0.0.1/src/topologies/synth/streaming.py +241 -0
- topologies-0.0.1/src/topologies/synth/tasks.py +717 -0
- topologies-0.0.1/src/topologies/synth/textutil.py +287 -0
- topologies-0.0.1/src/topologies/synth/torch_views.py +1157 -0
- topologies-0.0.1/src/topologies/synth/valuetypes.py +1741 -0
- topologies-0.0.1/src/topologies/synth/verdict.py +74 -0
- topologies-0.0.1/src/topologies/synth/versions.py +218 -0
- topologies-0.0.1/src/topologies/synth/yaml_spec.py +600 -0
- topologies-0.0.1/src/topologies/taxonomy/__init__.py +8 -0
- topologies-0.0.1/src/topologies/taxonomy/validate.py +74 -0
- topologies-0.0.1/src/topologies/tokenizers/__init__.py +78 -0
- topologies-0.0.1/src/topologies/tokenizers/base.py +350 -0
- topologies-0.0.1/src/topologies/tokenizers/bpe.py +190 -0
- topologies-0.0.1/src/topologies/tokenizers/hf.py +157 -0
- topologies-0.0.1/src/topologies/tokenizers/simple.py +139 -0
- topologies-0.0.1/src/topologies/tokenizers/stats.py +86 -0
- topologies-0.0.1/src/topologies/training/__init__.py +88 -0
- topologies-0.0.1/src/topologies/training/_checkpointing.py +232 -0
- topologies-0.0.1/src/topologies/training/_loops.py +215 -0
- topologies-0.0.1/src/topologies/training/_observability.py +148 -0
- topologies-0.0.1/src/topologies/training/_optim_factory.py +341 -0
- topologies-0.0.1/src/topologies/training/_run_info.py +93 -0
- topologies-0.0.1/src/topologies/training/_step_metrics.py +128 -0
- topologies-0.0.1/src/topologies/training/_steps.py +358 -0
- topologies-0.0.1/src/topologies/training/_trainer_config.py +50 -0
- topologies-0.0.1/src/topologies/training/adapters.py +207 -0
- topologies-0.0.1/src/topologies/training/aux.py +62 -0
- topologies-0.0.1/src/topologies/training/bart_data.py +124 -0
- topologies-0.0.1/src/topologies/training/byt5_data.py +99 -0
- topologies-0.0.1/src/topologies/training/callbacks.py +73 -0
- topologies-0.0.1/src/topologies/training/canine_data.py +69 -0
- topologies-0.0.1/src/topologies/training/charformer_data.py +65 -0
- topologies-0.0.1/src/topologies/training/checkpoint.py +207 -0
- topologies-0.0.1/src/topologies/training/components.py +106 -0
- topologies-0.0.1/src/topologies/training/configs.py +309 -0
- topologies-0.0.1/src/topologies/training/data.py +203 -0
- topologies-0.0.1/src/topologies/training/device.py +103 -0
- topologies-0.0.1/src/topologies/training/distributed.py +165 -0
- topologies-0.0.1/src/topologies/training/engine.py +1085 -0
- topologies-0.0.1/src/topologies/training/errors.py +61 -0
- topologies-0.0.1/src/topologies/training/introspect.py +19 -0
- topologies-0.0.1/src/topologies/training/logging/__init__.py +21 -0
- topologies-0.0.1/src/topologies/training/logging/base.py +82 -0
- topologies-0.0.1/src/topologies/training/logging/factory.py +69 -0
- topologies-0.0.1/src/topologies/training/logging/jsonl.py +210 -0
- topologies-0.0.1/src/topologies/training/logging/tabular.py +125 -0
- topologies-0.0.1/src/topologies/training/logging/text.py +137 -0
- topologies-0.0.1/src/topologies/training/logging_.py +23 -0
- topologies-0.0.1/src/topologies/training/losses.py +218 -0
- topologies-0.0.1/src/topologies/training/marian_data.py +51 -0
- topologies-0.0.1/src/topologies/training/model_recipes.json +1331 -0
- topologies-0.0.1/src/topologies/training/monitoring/__init__.py +15 -0
- topologies-0.0.1/src/topologies/training/monitoring/base.py +55 -0
- topologies-0.0.1/src/topologies/training/monitoring/dashboard/__init__.py +17 -0
- topologies-0.0.1/src/topologies/training/monitoring/dashboard/app.py +87 -0
- topologies-0.0.1/src/topologies/training/monitoring/dashboard/charts.py +50 -0
- topologies-0.0.1/src/topologies/training/monitoring/dashboard/state.py +202 -0
- topologies-0.0.1/src/topologies/training/monitoring/dashboard/views.py +36 -0
- topologies-0.0.1/src/topologies/training/monitoring/dashboard/widgets.py +64 -0
- topologies-0.0.1/src/topologies/training/monitoring/factory.py +88 -0
- topologies-0.0.1/src/topologies/training/monitoring/standard.py +264 -0
- topologies-0.0.1/src/topologies/training/monitoring/system.py +112 -0
- topologies-0.0.1/src/topologies/training/recipe_base.py +447 -0
- topologies-0.0.1/src/topologies/training/recipes/__init__.py +30 -0
- topologies-0.0.1/src/topologies/training/recipes/audio.py +88 -0
- topologies-0.0.1/src/topologies/training/recipes/autoenc.py +124 -0
- topologies-0.0.1/src/topologies/training/recipes/canine.py +47 -0
- topologies-0.0.1/src/topologies/training/recipes/classify.py +440 -0
- topologies-0.0.1/src/topologies/training/recipes/contrastive.py +83 -0
- topologies-0.0.1/src/topologies/training/recipes/density.py +50 -0
- topologies-0.0.1/src/topologies/training/recipes/detection.py +118 -0
- topologies-0.0.1/src/topologies/training/recipes/diffusion.py +88 -0
- topologies-0.0.1/src/topologies/training/recipes/distillation.py +128 -0
- topologies-0.0.1/src/topologies/training/recipes/gan.py +142 -0
- topologies-0.0.1/src/topologies/training/recipes/graph.py +29 -0
- topologies-0.0.1/src/topologies/training/recipes/lm.py +935 -0
- topologies-0.0.1/src/topologies/training/recipes/mixtral.py +66 -0
- topologies-0.0.1/src/topologies/training/recipes/permutation_lm.py +134 -0
- topologies-0.0.1/src/topologies/training/recipes/ppo.py +233 -0
- topologies-0.0.1/src/topologies/training/recipes/recommender.py +104 -0
- topologies-0.0.1/src/topologies/training/recipes/router_probe.py +63 -0
- topologies-0.0.1/src/topologies/training/recipes/segmentation.py +76 -0
- topologies-0.0.1/src/topologies/training/recipes/spanbert.py +125 -0
- topologies-0.0.1/src/topologies/training/recipes/speech.py +35 -0
- topologies-0.0.1/src/topologies/training/recipes/static.py +28 -0
- topologies-0.0.1/src/topologies/training/recipes/structured_objective.py +64 -0
- topologies-0.0.1/src/topologies/training/recipes/transformer_xl.py +76 -0
- topologies-0.0.1/src/topologies/training/recipes/tts.py +86 -0
- topologies-0.0.1/src/topologies/training/recipes/value_learning.py +39 -0
- topologies-0.0.1/src/topologies/training/recipes/vision.py +555 -0
- topologies-0.0.1/src/topologies/training/registries.py +573 -0
- topologies-0.0.1/src/topologies/training/reporting/__init__.py +35 -0
- topologies-0.0.1/src/topologies/training/reporting/events.py +210 -0
- topologies-0.0.1/src/topologies/training/reporting/reporter.py +204 -0
- topologies-0.0.1/src/topologies/training/resolve.py +248 -0
- topologies-0.0.1/src/topologies/training/schema.py +4 -0
- topologies-0.0.1/src/topologies/training/stability/activation_offload.py +6 -0
- topologies-0.0.1/src/topologies/training/stability/bf16_recipe.py +6 -0
- topologies-0.0.1/src/topologies/training/stability/compile_parity.py +6 -0
- topologies-0.0.1/src/topologies/training/stability/fsdp_wrap.py +6 -0
- topologies-0.0.1/src/topologies/training/stability/grad_checkpoint.py +16 -0
- topologies-0.0.1/src/topologies/training/stability/grad_clip.py +57 -0
- topologies-0.0.1/src/topologies/training/stability/llrd.py +8 -0
- topologies-0.0.1/src/topologies/training/stability/loss_scale.py +6 -0
- topologies-0.0.1/src/topologies/training/stability/nan_skip.py +8 -0
- topologies-0.0.1/src/topologies/training/stability/zero_partition.py +6 -0
- topologies-0.0.1/src/topologies/training/t5_data.py +90 -0
- topologies-0.0.1/src/topologies/training/ul2_data.py +62 -0
- topologies-0.0.1/src/topologies/training-artifacts.lock.json +302 -0
- topologies-0.0.1/src/topologies/training_schema.py +99 -0
- topologies-0.0.1/src/topologies/utils/__init__.py +8 -0
- topologies-0.0.1/src/topologies/utils/locking.py +80 -0
- topologies-0.0.1/src/topologies/utils/logging.py +40 -0
- topologies-0.0.1/src/topologies/validation/__init__.py +4 -0
- topologies-0.0.1/src/topologies/validation/catalog_specs.json +953 -0
- topologies-0.0.1/src/topologies/validation/catalog_specs.py +65 -0
- topologies-0.0.1/src/topologies/validation/catalog_workflows.py +1854 -0
- topologies-0.0.1/src/topologies/validation/component_workflows.py +588 -0
- topologies-0.0.1/src/topologies/validation/framework.py +151 -0
- topologies-0.0.1/src/topologies/validation/gate.py +87 -0
- topologies-0.0.1/src/topologies/validation/implementation.py +35 -0
- topologies-0.0.1/src/topologies/validation/source_style.py +23 -0
- topologies-0.0.1/src/topologies/visualization/__init__.py +103 -0
- topologies-0.0.1/src/topologies/visualization/adapters.py +225 -0
- topologies-0.0.1/src/topologies/visualization/core.py +872 -0
- topologies-0.0.1/src/topologies/visualization/experiment_views.py +765 -0
- topologies-0.0.1/src/topologies/visualization/format.py +322 -0
- topologies-0.0.1/src/topologies/visualization/live.py +223 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/__init__.py +498 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/live.py +247 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/style.py +262 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/views/__init__.py +84 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/views/analysis.py +182 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/views/engineering.py +206 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/views/experiments.py +344 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/views/generic.py +87 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/views/hardware.py +64 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/views/profile.py +77 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib/views/training.py +286 -0
- topologies-0.0.1/src/topologies/visualization/matplotlib_views.py +325 -0
- topologies-0.0.1/src/topologies/visualization/state.py +989 -0
- topologies-0.0.1/src/topologies/visualization/terminal/__init__.py +133 -0
- topologies-0.0.1/src/topologies/visualization/terminal/caps.py +195 -0
- topologies-0.0.1/src/topologies/visualization/terminal/driver.py +356 -0
- topologies-0.0.1/src/topologies/visualization/terminal/glyphs.py +102 -0
- topologies-0.0.1/src/topologies/visualization/terminal/keys.py +146 -0
- topologies-0.0.1/src/topologies/visualization/terminal/layout.py +162 -0
- topologies-0.0.1/src/topologies/visualization/terminal/panels.py +128 -0
- topologies-0.0.1/src/topologies/visualization/terminal/progress.py +76 -0
- topologies-0.0.1/src/topologies/visualization/terminal/screen.py +503 -0
- topologies-0.0.1/src/topologies/visualization/terminal/text.py +401 -0
- topologies-0.0.1/src/topologies/visualization/terminal/widgets.py +967 -0
- topologies-0.0.1/src/topologies/visualization/testing.py +827 -0
- topologies-0.0.1/src/topologies/visualization/theme.py +252 -0
- topologies-0.0.1/src/topologies/visualization/training_views.py +1280 -0
- 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.
|
topologies-0.0.1/LICENSE
ADDED
|
@@ -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.*"]
|