API
===

.. automodule:: neural_schrodinger
    :members: