Contribute to Ergo core

To get started:

  1. git clone

  2. poetry install

  3. poetry shell


Ergo uses poetry to manage its dependencies and environments.

Follow these directions to install poetry if you don’t already have it.

Troubleshooting: If you get Could not find a version that satisfies the requirement jaxlib ... after using poetry to install, this is probably because your virtual environment has old version of pip due to how poetry choses pip versions.


  1. poetry run pip install -U pip

  2. poetry install again

Before submitting a PR

  1. Run poetry install to make sure you have the latest dependencies

  2. Format code using make format (black, isort)

  3. Run linting using make lint (flake8, mypy, black check)

  4. Run tests using make test

    • To run the tests in, you’ll need our secret .env file. If you don’t have it, you can ask us for it, or rely on Travis CI to run those tests for you.

  5. Generate docs using make docs, load docs/build/html/index.html and review the generated docs

  6. Or run all of the above using make all


Import numpy as follows:

import jax.numpy as np
import numpy as onp