flax
Avoid passing python collections to jax reductions in SST2 example.
#462
Merged
Merged