astroquery jax poliastro pytest requests torchdiffeq