brainpy.math module

The math module for whole BrainPy ecosystem. This module provides basic mathematical operations, including:

  • numpy-like array operations

  • linear algebra functions

  • random sampling functions

  • discrete fourier transform functions

  • compilations of jit, vmap, pmap for class objects

  • automatic differentiation of grad, jacocian, hessian, etc. for class objects

  • loss functions

  • activation functions

  • optimization classes

Details in the following.