Fix local dev environment

• Move most dependencies from environment.yml to pyproject.toml
• Fix python version conflict
• Make lisabeta non-optional in pyproject.toml
• Add dev env install instructions (closes #107 (closed))
• Make tests runnable without JAX which is optional