[FEATURE REQUEST] Use official array API types in ribs.typing

[btjanaka]

Description

• Currently, we have Array, Device, and DType aliases in ribs.typing. These should be replaced with the official array API types once those are released -- see here: https://github.com/data-apis/array-api-typing/
• Furthermore, we should re-check annotations for ArrayStore
• And we can also make xp have a more specific type than ModuleType, like ArrayNamespace