Remaining deprecations for array API compliance

[Micky774]

This issue tracks the deprecations that must still be started to ensure that the jax.numpy namespace is array API compliant. This covers some issues remaining which are not explicitly tracked in #18353, with some overlap between both.

Core

• Deprecate newshape argument of jnp.reshape #21130
• Add device kwarg
  • jnp.arange
  • jnp.array
  • jnp.eye
  • jnp.linspace
• Wait on final decision on finfo should not require float type for the result fields data-apis/array-api#405
• Preserve integral dtype
  • jnp.ceil
  • jnp.floor
  • jnp.trunc
• Added correction arg to jnp.var and jnp.std for array-api compliance #21262
  • jnp.std
  • jnp.var
• Add info namespace containing the same functions as Add __array_namespace_info__ and corresponding utilities #20294

Compliance Finalization

• WIP jax.numpy array API compliance finalization #21323
  • Add __array_namespace__ to ArrayImpl
  • Add to_device to ArrayImpl
  • Add device to ArrayImpl
  • Add __array_namespace_info__ to jax.numpy
  • Add __array_api_version__ version information

FFT

• Add device kwarg
  • jnp.fft.fftfreq
  • jnp.fft.rfftfreq

Linalg

• Add trace function to jax.numpy.linalg
• Rename rcond/tol to rtol in linalg.matrix_rank and linalg.pinv #21226
  • jnp.linalg.matrix_rank: Change tol arg to rtol
  • jnp.linalg.pinv: Change rcond arg to rtol

[vfdev-5]

Task

Add info namespace containing the same functions as #20294

Nothing to be done in jax as array-api-tests should be fixed: #21506