Remaining API changes 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

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 [array API] move api metadata into jax.numpy namespace #22734
  • jnp.fft.fftfreq
  • jnp.fft.rfftfreq

Linalg

• Added trace alias to jnp.linalg #21442
• 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

[jakevdp]

Closed by #22818