This repository is a CPU-first, 30-lesson, workbook-style JAX course for learners who are comfortable with Python and NumPy-shaped array thinking and are new to JAX. It pairs importable lesson modules ...
input instead of the columns. That is, the :math:`i`'th row of the output below is the mapping of the :math:`i`'th row of the input under :math:`A`, plus the bias term. Look at the example below.