Open Issues Need Help
View All on GitHub Astronomy • Image Simulation
functions based on `jnp.arange` produce incompatibility with jax transformations. about 2 months ago
AI Summary: The task is to modify the `rubix` Python package to resolve incompatibility issues between its `jnp.arange` based functions and JAX transformations like `jax.jit`. This involves identifying functions in `telescope/utils.py` that use `jnp.arange` dynamically, and refactoring them to be compatible with JAX's tracing and compilation process. The goal is to ensure that the pipeline functions correctly within JAX's just-in-time compilation framework without errors.
Complexity:
4/5
bug documentation help wanted