Open donno2048 opened 2 years ago
Thanks for the feature request!
I think adapting everything to work with masked arrays might take quite a bit of work. This may remain out of scope for us for a while. There's a lot of other high-priority work on our plates.
Some of this functionality could likely be implemented as a separate library on top of JAX itself.
Is this planned for any time in the future?
Please: Add masked arrays (
numpy.ma.array
) it doesn't have to even be a new feature, it can be done by simply addingmask
andfill_value
arguments to the existing arrays (jax.numpy.array
)