Since we are assuming you are fairly familiar with NumPy, this chapter is by no means a thorough coverage of the topic and will just serve as a quick refresher. We will practice using JAX while discovering the equivalence between normal NumPy and the JAX version of NumPy’s syntax.
Note: As a reference, we’ll use
jnprespectively for default and JAX NumPy versions in our codes.
The inner product of two vectors can be calculated by any of the three syntaxes (
@), as shown below: