GitHub repository: BHI-Research/PSO-JAX
Description: An accelerated Particle Swarm Optimization tool that uses JAX key component