Skip to content

Update jax.experimental.array_api to v2023.12 API #20200

Open
@Micky774

Description

@Micky774

This issue tracks the changes necessary to adopt the v2023.12 Array API. This was originally mentioned in #18353. Note that there may be some specifications that we already satisfy, however the vast majority of these will need alterations.

API Updates

New API

Breaking Changes

For specific details on what has changed, look on their specification pages for "Changed in version 2023.12 ..."

Common Utilities Refactor

  • Device placement + copy semantics
    • Initial draft can be found here
    • Get Device from Device | Sharding

Metadata

Metadata

Assignees

Labels

P1 (soon)Assignee is working on this now, among other tasks. (Assignee required)enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions