Operator Registration#
General Operator Customization Interface#
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
just-in-time compilation for class objects
automatic differentiation for class objects
dedicated operators for brain dynamics modeling
activation functions
device/dtype switching
and others
Details in the following.
CPU Operator Customization with Taichi#
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
just-in-time compilation for class objects
automatic differentiation for class objects
dedicated operators for brain dynamics modeling
activation functions
device/dtype switching
and others
Details in the following.
CPU Operator Customization with Numba#
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
just-in-time compilation for class objects
automatic differentiation for class objects
dedicated operators for brain dynamics modeling
activation functions
device/dtype switching
and others
Details in the following.
Operator Autograd Customization#
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
just-in-time compilation for class objects
automatic differentiation for class objects
dedicated operators for brain dynamics modeling
activation functions
device/dtype switching
and others
Details in the following.