Use the micronumpy module. This module provides a very basic numpy-like interface. Major use-case is to show how jit scales for other code.