We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
a5b622b
Incorporating some minor code changes to be compatible with latest jax releases. (E.g., jax 0.4.24 no longer has row_stack.)
row_stack