docs
examples
optax
requirements
