swarms-torch 0.0.3

swarms-torch - Pytorch

Swarms in Torch

Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch primitives😊

Installation

You can install the package using pip

pip3 install swarms-torch

Usage

• We have just PSO now, but we're adding in ant colony and others!

from swarms_torch import ParticleSwarmOptimization

#test
pso = ParticleSwarmOptimization(goal="Attention is all you need", n_particles=100)
pso.optimize(iterations=1000)

• Ant Colony Optimization

from swarms_torch.ant_colony_swarm import AntColonyOptimization

# Usage:
goal_string = "Hello ACO"
aco = AntColonyOptimization(goal_string, num_iterations=1000)
best_solution = aco.optimize()
print("Best Matched String:", best_solution)

License

MIT