jax tensor and numpy array convertion #1961

@yfji

Description

How to convert numpy array to the jax tensor, or from jax tensor to numpy array?