* black, isort, flake8 -> ruff * remove unused * add back needed import * fix
* chunked cross entropy loss * refactor so we can add test * use relative import * update schema description