register `atexit` hook on module `_init_`

[sjkelly]

These hooks are usually registered at module __init__, partially to make the code easier to reason about. Though this now has greater significance with improved static compilation techniques, since the __init__ is the module entry and anything outside will be discarded from what I can tell.

[KristofferC]

Yeah that makes sense. The previous behavior feels broken...