flax
331bdc2c - Fix ConcretizationTypeError in jnp.split.

Commit

5 years ago

Fix ConcretizationTypeError in jnp.split.

References

#648 - Add optimized LSTM cell.

Author

bastings

bastings

Parents

Loading