Skip to content

An accelerated Particle Swarm Optimization tool that uses JAX key component

License

Notifications You must be signed in to change notification settings

joelermantraut/PSO-JAX

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 

Repository files navigation

PSO-JAX

An accelerated Particle Swarm Optimization tool that uses JAX key component. Here are the main benefits:

  • Runs on Python.
  • Supports multi-core CPUs, GPUs, and TPUs.
  • Compatible with Colab environment.

Getting Started

See this example

References

About

An accelerated Particle Swarm Optimization tool that uses JAX key component

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published