Question about jax-metal development and missing operations

[a-gn]

Hi! I've tested jax-metal on a mac and encountered a few of these missing operations, and others, while implementing common papers.

I'd like to know if there is an ETA for ~full support of Apple Silicon platforms in JAX. It would be a great help if the neural/GPU cores on my mac could replace cloud compute for some of these workflows, but right now I have to convert to numpy and back for some operations.

I'm also interested in contributing, but not sure if it's within my competence yet. Is there something I can read (part of the codebase, guide, etc.) to know where to start adding support for new operations on Metal?

Thanks!

[jakevdp]

cc/ @shuhand0 @kulinseth