brainpy.math.compat module#