Jake Vanderplas

> Fair enough, but isn't NEP 29 now superseded by SPEC 0? Not in JAX... our policy still explicitly points to NEP 29. We could have a discussion about changing...

I'm proposing switching to SPEC 0 in #18072

Well, it looks like the result of the SPEC-0 discussion is that we're going to support old Python releases for even longer than we did previously. If the current proposal...

Hi @NeilGirdhar – what's the status of this? Do you want to keep pushing on this approach?

Yes, we currently run the CI on mypy 1.6.1: https://github.com/google/jax/blob/953f4670d88d2a1c168a4ad0b44ed940f6c58829/.pre-commit-config.yaml#L29-L30

Seems like maybe a bad interaction with `functools.partial`?

> @jakevdp I think we should wait one more release of MyPy Sounds good

Pytype is failing with this message: ``` File ".../jax/_src/api.py", line 100, in : argument "covariant" to TypeVar not supported yet [not-supported-yet] ``` Seems this is a known issue (https://github.com/google/pytype/issues/1471). Is...

It seems like the compiler isn't making a great choice here with respect to the run-time/compile-time tradeoffs involved in constant folding. If you change your code slightly though, the problematic...

I don't know... it's really an XLA bug, and I'm not sure of a way to change what XLA does here. Maybe you could rewrite your code so that `nonzeros`...