Enable GPU/parallelize for pyswarms

ljvmiranda921 / pyswarms

A research toolkit for particle swarm optimization in Python

https://pyswarms.readthedocs.io/en/latest/

MIT License

1.29k stars 332 forks source link

Enable GPU/parallelize for pyswarms #495

Open Berlyli866 opened 2 years ago

Berlyli866 commented 2 years ago

Is it possible to use gpu for pyswarms (in python)? My loss function calls the simulator so it takes a lot of time and I wonder if it is possible to parallelize the calculation for the particles of the swarm in each iteration?

or do we have an existing function to use?

thanks for the help.