Rockpool: a python package for design and training of light and ultra low-power edge ML applications