Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/jacobusmmsmit/social-force-jax

An implementation of a simple Social Force model in JAX
https://github.com/jacobusmmsmit/social-force-jax

social-force-model

Last synced: about 1 month ago
JSON representation

An implementation of a simple Social Force model in JAX

Awesome Lists containing this project

README

        

# social-force-jax
An implementation of a simple Social Force model in JAX + Diffrax as well as a differentiable density-based optimisation of the parameters with Optax or numpyro.

To run the code in this repo, install the poetry package manager, navigate to the folder, and run `poetry install`. This will install all dependencies.

Note: The local file imports are kinda messed up for this repo, you may need to play around with the path to get it working properly.

### Known issues:
* The wall force sometimes lets pedestrians escape (see animation), oops.