fix(ci): pin numpy and jax during pip install to prevent silent upgrades
chore(release): bump version to v0.1.1 and update changelog