I recommend checking out https://github.com/Point72/csp