Add jax functionality
This MR directly follows from !35 (merged) to add new features in the upcoming gwpop
v1.
The relevant changes can be seen here.
Edited by Colm Talbot
This MR directly follows from !35 (merged) to add new features in the upcoming gwpop
v1.
The relevant changes can be seen here.