6 Commits

Author	SHA1	Message	Date
Jake VanderPlas	60256df668	[typing] define additional methods & properties on jax.Array ... These are the methods that are only valid for actual materialized arrays (i.e. not Tracers) In order to simplify the experience for users, we want to maintain only a single jax.Array type, so we define all methods here and raise explicit errors on Tracer instances.	2023-02-10 09:42:32 -08:00
Jake VanderPlas	e1738af5b2	[typing] add missing attributes to jax.Array	2023-01-11 10:00:59 -08:00
Jake VanderPlas	777754d595	[typing] fix Shard/Sharding type TODO	2022-12-07 09:44:03 -08:00
Jake VanderPlas	8fbf8da810	Declare Array.sharding & raise an error on tracers	2022-11-08 14:20:46 -08:00
Jake VanderPlas	c1328a67c8	[typing] overloads for jnp.linalg.svd & jnp.linalg.qr	2022-10-05 14:29:00 -07:00
Jake VanderPlas	0cb233eec9	Add initial jax.Array base class for instance checks & annotation	2022-09-26 07:48:43 -07:00