├── .github ├── FUNDING.yml ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── PULL_REQUEST_TEMPLATE.yml ├── dependabot.yml └── workflows │ ├── code_quality_control.yml │ ├── cos_integration.yml │ ├── docs.yml │ ├── docs_test.yml │ ├── label.yml │ ├── lints.yml │ ├── pr_request_checks.yml │ ├── pull-request-links.yml │ ├── pylint.yml │ ├── python-publish.yml │ ├── quality.yml │ ├── ruff.yml │ ├── run_test.yml │ ├── stale.yml │ ├── test.yml │ ├── testing.yml │ ├── unit-test.yml │ └── welcome.yml ├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── docs ├── .DS_Store ├── .readthedocs.yaml ├── 20_swarm_model_ideas.md ├── README.md ├── assets │ ├── css │ │ └── extra.css │ └── img │ │ ├── SwarmsLogoIcon.png │ │ ├── swarmsbanner.png │ │ ├── tools │ │ ├── output.png │ │ ├── poetry_setup.png │ │ └── toml.png │ │ └── zetascale.png ├── contributing.md ├── index.md ├── mkdocs.yml ├── overrides │ └── main.html ├── requirements.txt ├── stylesheets │ └── extra.css ├── swarms │ ├── aco.md │ ├── firefly.md │ ├── index.md │ ├── pso.md │ ├── qb.md │ └── so.md └── vision.md ├── example.py ├── examples ├── ant_colony.py ├── fire_fly_example.py ├── fish_school_example.py ├── mixture_of_mambas.py ├── new_mergers.py ├── nnt.py ├── queen_bee_transformer_hierarchy_example.py ├── silu_visualization.py ├── simple_moe.py ├── sop.py ├── swarmalator_example.py └── switch_moe.py ├── multi_modal_mergers.py ├── pyproject.toml ├── queen_bee_transformer_hierarchy.py ├── requirements.txt ├── scripts ├── code_quality.sh ├── get_package_requirements.py ├── requirementstxt_to_pyproject.py ├── test_name.sh └── tests.sh ├── swarms_torch ├── __init__.py ├── mergers │ ├── __init__.py │ ├── all_new_evo_mergers.py │ └── mm_mergers.py ├── pso │ ├── __init__.py │ ├── multi_swarm_pso.py │ ├── multi_swarm_pso2.py │ ├── multi_swarm_pso_transformer.py │ └── transformer_pso.py ├── structs │ ├── __init__.py │ ├── ant_colony_swarm.py │ ├── cellular_transformer.py │ ├── firefly.py │ ├── fish_school.py │ ├── graph_cellular_automa.py │ ├── hivemind_swarm_transformer.py │ ├── ma_agent.py │ ├── mas_model.py │ ├── mixtral_expert.py │ ├── mixture_of_mamba.py │ ├── neuronal_transformer.py │ ├── parallel_wrapper.py │ ├── queen_bee.py │ ├── simple_moe.py │ └── switch_moe.py ├── swarmalators │ ├── __init__.py │ ├── swarmalator_base.py │ ├── swarmalator_transformer.py │ └── swarmalator_visualize.py └── utils │ ├── __init__.py │ ├── particle_swarm.py │ ├── sakana.py │ └── spiral_optimization.py ├── test.py ├── test_switch_moe_fix.py └── tests ├── ant_colony.py ├── cellular_swarm.py ├── fish_school.py ├── neuronal_transformer.py ├── particle_swarm.py ├── queen_bee.py ├── spiral_optimization.py ├── swarmalator_base.py ├── test.py ├── test_mixture_of_mamba.py ├── transformer_hive.py └── transformer_pso.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/PULL_REQUEST_TEMPLATE.yml -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/code_quality_control.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/code_quality_control.yml -------------------------------------------------------------------------------- /.github/workflows/cos_integration.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/cos_integration.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/docs_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/docs_test.yml -------------------------------------------------------------------------------- /.github/workflows/label.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/label.yml -------------------------------------------------------------------------------- /.github/workflows/lints.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/lints.yml -------------------------------------------------------------------------------- /.github/workflows/pr_request_checks.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/pr_request_checks.yml -------------------------------------------------------------------------------- /.github/workflows/pull-request-links.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/pull-request-links.yml -------------------------------------------------------------------------------- /.github/workflows/pylint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/pylint.yml -------------------------------------------------------------------------------- /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.github/workflows/quality.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/quality.yml -------------------------------------------------------------------------------- /.github/workflows/ruff.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/ruff.yml -------------------------------------------------------------------------------- /.github/workflows/run_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/run_test.yml -------------------------------------------------------------------------------- /.github/workflows/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/stale.yml -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.github/workflows/testing.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/testing.yml -------------------------------------------------------------------------------- /.github/workflows/unit-test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/unit-test.yml -------------------------------------------------------------------------------- /.github/workflows/welcome.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.github/workflows/welcome.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /docs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/.DS_Store -------------------------------------------------------------------------------- /docs/.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/.readthedocs.yaml -------------------------------------------------------------------------------- /docs/20_swarm_model_ideas.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/20_swarm_model_ideas.md -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/assets/css/extra.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/assets/css/extra.css -------------------------------------------------------------------------------- /docs/assets/img/SwarmsLogoIcon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/assets/img/SwarmsLogoIcon.png -------------------------------------------------------------------------------- /docs/assets/img/swarmsbanner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/assets/img/swarmsbanner.png -------------------------------------------------------------------------------- /docs/assets/img/tools/output.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/assets/img/tools/output.png -------------------------------------------------------------------------------- /docs/assets/img/tools/poetry_setup.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/assets/img/tools/poetry_setup.png -------------------------------------------------------------------------------- /docs/assets/img/tools/toml.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/assets/img/tools/toml.png -------------------------------------------------------------------------------- /docs/assets/img/zetascale.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/assets/img/zetascale.png -------------------------------------------------------------------------------- /docs/contributing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/contributing.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/mkdocs.yml -------------------------------------------------------------------------------- /docs/overrides/main.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/overrides/main.html -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/stylesheets/extra.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/stylesheets/extra.css -------------------------------------------------------------------------------- /docs/swarms/aco.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/swarms/aco.md -------------------------------------------------------------------------------- /docs/swarms/firefly.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/swarms/firefly.md -------------------------------------------------------------------------------- /docs/swarms/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/swarms/index.md -------------------------------------------------------------------------------- /docs/swarms/pso.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/swarms/pso.md -------------------------------------------------------------------------------- /docs/swarms/qb.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/swarms/qb.md -------------------------------------------------------------------------------- /docs/swarms/so.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/swarms/so.md -------------------------------------------------------------------------------- /docs/vision.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/docs/vision.md -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/example.py -------------------------------------------------------------------------------- /examples/ant_colony.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/ant_colony.py -------------------------------------------------------------------------------- /examples/fire_fly_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/fire_fly_example.py -------------------------------------------------------------------------------- /examples/fish_school_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/fish_school_example.py -------------------------------------------------------------------------------- /examples/mixture_of_mambas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/mixture_of_mambas.py -------------------------------------------------------------------------------- /examples/new_mergers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/new_mergers.py -------------------------------------------------------------------------------- /examples/nnt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/nnt.py -------------------------------------------------------------------------------- /examples/queen_bee_transformer_hierarchy_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/queen_bee_transformer_hierarchy_example.py -------------------------------------------------------------------------------- /examples/silu_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/silu_visualization.py -------------------------------------------------------------------------------- /examples/simple_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/simple_moe.py -------------------------------------------------------------------------------- /examples/sop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/sop.py -------------------------------------------------------------------------------- /examples/swarmalator_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/swarmalator_example.py -------------------------------------------------------------------------------- /examples/switch_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/examples/switch_moe.py -------------------------------------------------------------------------------- /multi_modal_mergers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/multi_modal_mergers.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /queen_bee_transformer_hierarchy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/queen_bee_transformer_hierarchy.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/code_quality.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/scripts/code_quality.sh -------------------------------------------------------------------------------- /scripts/get_package_requirements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/scripts/get_package_requirements.py -------------------------------------------------------------------------------- /scripts/requirementstxt_to_pyproject.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/scripts/requirementstxt_to_pyproject.py -------------------------------------------------------------------------------- /scripts/test_name.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/scripts/test_name.sh -------------------------------------------------------------------------------- /scripts/tests.sh: -------------------------------------------------------------------------------- 1 | find ./tests -name '*.py' -exec pytest {} \; -------------------------------------------------------------------------------- /swarms_torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/__init__.py -------------------------------------------------------------------------------- /swarms_torch/mergers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/mergers/__init__.py -------------------------------------------------------------------------------- /swarms_torch/mergers/all_new_evo_mergers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/mergers/all_new_evo_mergers.py -------------------------------------------------------------------------------- /swarms_torch/mergers/mm_mergers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/mergers/mm_mergers.py -------------------------------------------------------------------------------- /swarms_torch/pso/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /swarms_torch/pso/multi_swarm_pso.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/pso/multi_swarm_pso.py -------------------------------------------------------------------------------- /swarms_torch/pso/multi_swarm_pso2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/pso/multi_swarm_pso2.py -------------------------------------------------------------------------------- /swarms_torch/pso/multi_swarm_pso_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/pso/multi_swarm_pso_transformer.py -------------------------------------------------------------------------------- /swarms_torch/pso/transformer_pso.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/pso/transformer_pso.py -------------------------------------------------------------------------------- /swarms_torch/structs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/__init__.py -------------------------------------------------------------------------------- /swarms_torch/structs/ant_colony_swarm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/ant_colony_swarm.py -------------------------------------------------------------------------------- /swarms_torch/structs/cellular_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/cellular_transformer.py -------------------------------------------------------------------------------- /swarms_torch/structs/firefly.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/firefly.py -------------------------------------------------------------------------------- /swarms_torch/structs/fish_school.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/fish_school.py -------------------------------------------------------------------------------- /swarms_torch/structs/graph_cellular_automa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/graph_cellular_automa.py -------------------------------------------------------------------------------- /swarms_torch/structs/hivemind_swarm_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/hivemind_swarm_transformer.py -------------------------------------------------------------------------------- /swarms_torch/structs/ma_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/ma_agent.py -------------------------------------------------------------------------------- /swarms_torch/structs/mas_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/mas_model.py -------------------------------------------------------------------------------- /swarms_torch/structs/mixtral_expert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/mixtral_expert.py -------------------------------------------------------------------------------- /swarms_torch/structs/mixture_of_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/mixture_of_mamba.py -------------------------------------------------------------------------------- /swarms_torch/structs/neuronal_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/neuronal_transformer.py -------------------------------------------------------------------------------- /swarms_torch/structs/parallel_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/parallel_wrapper.py -------------------------------------------------------------------------------- /swarms_torch/structs/queen_bee.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/queen_bee.py -------------------------------------------------------------------------------- /swarms_torch/structs/simple_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/simple_moe.py -------------------------------------------------------------------------------- /swarms_torch/structs/switch_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/structs/switch_moe.py -------------------------------------------------------------------------------- /swarms_torch/swarmalators/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /swarms_torch/swarmalators/swarmalator_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/swarmalators/swarmalator_base.py -------------------------------------------------------------------------------- /swarms_torch/swarmalators/swarmalator_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/swarmalators/swarmalator_transformer.py -------------------------------------------------------------------------------- /swarms_torch/swarmalators/swarmalator_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/swarmalators/swarmalator_visualize.py -------------------------------------------------------------------------------- /swarms_torch/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /swarms_torch/utils/particle_swarm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/utils/particle_swarm.py -------------------------------------------------------------------------------- /swarms_torch/utils/sakana.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/utils/sakana.py -------------------------------------------------------------------------------- /swarms_torch/utils/spiral_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/swarms_torch/utils/spiral_optimization.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/test.py -------------------------------------------------------------------------------- /test_switch_moe_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/test_switch_moe_fix.py -------------------------------------------------------------------------------- /tests/ant_colony.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/ant_colony.py -------------------------------------------------------------------------------- /tests/cellular_swarm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/cellular_swarm.py -------------------------------------------------------------------------------- /tests/fish_school.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/fish_school.py -------------------------------------------------------------------------------- /tests/neuronal_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/neuronal_transformer.py -------------------------------------------------------------------------------- /tests/particle_swarm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/particle_swarm.py -------------------------------------------------------------------------------- /tests/queen_bee.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/queen_bee.py -------------------------------------------------------------------------------- /tests/spiral_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/spiral_optimization.py -------------------------------------------------------------------------------- /tests/swarmalator_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/swarmalator_base.py -------------------------------------------------------------------------------- /tests/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/test.py -------------------------------------------------------------------------------- /tests/test_mixture_of_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/test_mixture_of_mamba.py -------------------------------------------------------------------------------- /tests/transformer_hive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/transformer_hive.py -------------------------------------------------------------------------------- /tests/transformer_pso.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kyegomez/swarms-pytorch/HEAD/tests/transformer_pso.py --------------------------------------------------------------------------------