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.

• General Functions
  • brainpy.math.use_backend
  • brainpy.math.get_backend_name
  • brainpy.math.set_dt
  • brainpy.math.get_dt
  • brainpy.math.set_int_
  • brainpy.math.set_float_
  • brainpy.math.set_complex_
• JAX backend Supports
  • Compilations
  • Variables
  • Functions
• NumPy backend Supports
  • Compilations
  • Variables
  • Functions
• JAX Special Supports
  • Parallel Compilation
  • Operators
  • Control Flows
  • Automatic Differentiation
  • Activation Functions
  • Loss Functions
  • Optimizers
• Comparison Table
  • Multi-dimensional Array
  • Array Operations
  • Linear Algebra
  • Discrete Fourier Transform
  • Random Sampling