Remove jnp.DeviceArray since it is deprecated.
The latest version of JAX deprecates jax.numpy.DeviceArray. When using this version with Transformers, we get this error when instantiating FlaxBertModel: `module 'jax.numpy' has no attribute 'DeviceArray'`
cc @sanchit-gandhi <|||||>@sanchit-gandhi This pull request should be prioritised since the affected Flax models are currently not usable at all.<|||||>@mariecwhite Thanks for opening this PR! Running `make style` and pushing the changes will resolve the code quality CI checks<|||||>Hey @mariecwhite - it seems there is an issue with your CircleCI permissions, meaning the tests won't run!
Could you try refreshing your permissions as shown [here](https://support.circleci.com/hc/en-us/articles/360048210711-How-to-Refresh-User-Permissions-)? Let me know if you encounter any issues!<|||||>The docs for this PR live [here](https://moon-ci-docs.huggingface.co/docs/transformers/pr_24875). All of your documentation changes will be reflected on that endpoint.<|||||>> Hey @mariecwhite - it seems there is an issue with your CircleCI permissions, meaning the tests won't run! Could you try refreshing your permissions as shown [here](https://support.circleci.com/hc/en-us/articles/360048210711-How-to-Refresh-User-Permissions-)? Let me know if you encounter any issues!

I just updated my CircleCI permissions and rebased.<|||||>CircleCI is still failing for me. Since this is a priority, I'm happy to let https://github.com/huggingface/transformers/pull/25275 get merged instead of this.