BHI-Research / PSO-JAX Public

Notifications You must be signed in to change notification settings
Fork 3
Star 4

An accelerated Particle Swarm Optimization tool that uses JAX key component

Apache-2.0 license

4 stars 3 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 12 Commits
examples/knapspack		examples/knapspack
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md

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

BHI Research
JAX
PySwarms is used in the examples to compare results.

About

An accelerated Particle Swarm Optimization tool that uses JAX key component

gpu optimization-tools tpu jax pso-algorithm

Apache-2.0 license

Custom properties

Report repository

Releases

No releases published

Packages

No packages published

Contributors 2