jax

d8f9a27a - Added transpose rhs option to ragged dot kernel, useful for implementing the backward function.