3 ms·

> see e.g. lax, which is include in JAX. I think jleber means stax here, for pre-defined NN layers.

• jlebar 7y ago
  Yes, mb!