Access Xspec models and corresponding JAX/XLA ops.
Project description
xspex
Access Xspec models and corresponding JAX/XLA ops, based on xspec_models_cxc and extending-jax.
Installation
Note: HEASoft
& Xspec
are required to be installed on your system. You can download it from here.
pip install xspex
Example
import jax
import jax.numpy as jnp
import xspex
# For accuracy, it is recommended to enable double precision
jax.config.update('jax_enable_x64', True)
# Get APEC model primitive, whose JVP rule is defined by finite difference
apec, info = xspex.get_primitive('apec')
# Evaluate the model
value = apec(
params=jnp.array([1.0, 1.0, 0.0]),
egrid=jnp.geomspace(0.1, 0.2, 6),
spec_num=1,
)
print(value) # [1.27358561 0.37946811 0.2477116 0.1071355 0.10049102]
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
xspex-0.0.3.tar.gz
(62.7 kB
view hashes)