Update nn.py to use jnp.add rather than + operand.

Update nn.py to use jnp.add rather than + operand. #112