Explore index
Pallas Kernels
A framework in JAX for writing high-performance custom operators, often used with the XLA compiler to optimize computation graph
No public content is connected to this entity yet.
Explore index
A framework in JAX for writing high-performance custom operators, often used with the XLA compiler to optimize computation graph
No public content is connected to this entity yet.