removed device variable from the basic object

[yfukai]

Solves #101. We leave the device selection to the users for more flexibility. https://jax.readthedocs.io/en/latest/faq.html#controlling-data-and-computation-placement-on-devices