Removed jax-rx dependency