fix error: jax_array config option is removed in jax v0.4.14

[chaihahaha]

Reopened pull request, jax_array config option has been removed since jax commit 59509dc.

[JoeyTeng]

Thank you for updating this! Just a quick reply, I would need a little bit of time to unblock the CI to merge it. I'll update you when I am done:)

[chaihahaha]

@JoeyTeng Hello, I found a backward compatibility issue of the original modification and fixed it. Could you check this PR again?

[JoeyTeng]

@chaihahaha sorry to let you wait for so long. I will check soon to fix the CI.
However, I see you are using dir(). It is not recommended to use that outside of interactive interpreter. Could you change that with hasattr()?
I don't have other issues with your PR. I just need to fix the CI in order to merge your PR.

[chaihahaha]

Thanks for pointing it out, and thanks for your effort to maintain this repo.

I have fixed my RP with respect to your concern.

[chaihahaha]

This pull request will close #18

[horizon-blue]

Hi hi, do we have any updates on this? The use of deprecated jax_array option is blocking the Colab example from running as well, and it would be great if this PR can be merged :)

[markusheimerl]

I can confirm, manually integrating the changes of this pull requests, makes the jaxrenderer functional again.