Comparison Table

Here is a list of NumPy APIs and its corresponding BrainPy implementations.

- in BrainPy column denotes that implementation is not provided yet. We welcome contributions for these functions.

Multi-dimensional Array

NumPy

brainpy.math.numpy

brainpy.math.jax

numpy.ndarray.all()

brainpy.math.numpy.ndarray.all()

brainpy.math.jax.ndarray.all()

numpy.ndarray.any()

brainpy.math.numpy.ndarray.any()

brainpy.math.jax.ndarray.any()

numpy.ndarray.argmax()

brainpy.math.numpy.ndarray.argmax()

brainpy.math.jax.ndarray.argmax()

numpy.ndarray.argmin()

brainpy.math.numpy.ndarray.argmin()

brainpy.math.jax.ndarray.argmin()

numpy.ndarray.argpartition()

brainpy.math.numpy.ndarray.argpartition()

brainpy.math.jax.ndarray.argpartition()

numpy.ndarray.argsort()

brainpy.math.numpy.ndarray.argsort()

brainpy.math.jax.ndarray.argsort()

numpy.ndarray.astype()

brainpy.math.numpy.ndarray.astype()

brainpy.math.jax.ndarray.astype()

numpy.ndarray.byteswap()

brainpy.math.numpy.ndarray.byteswap()

brainpy.math.jax.ndarray.byteswap()

numpy.ndarray.choose()

brainpy.math.numpy.ndarray.choose()

brainpy.math.jax.ndarray.choose()

numpy.ndarray.clip()

brainpy.math.numpy.ndarray.clip()

brainpy.math.jax.ndarray.clip()

numpy.ndarray.compress()

brainpy.math.numpy.ndarray.compress()

brainpy.math.jax.ndarray.compress()

numpy.ndarray.conj()

brainpy.math.numpy.ndarray.conj()

brainpy.math.jax.ndarray.conj()

numpy.ndarray.conjugate()

brainpy.math.numpy.ndarray.conjugate()

brainpy.math.jax.ndarray.conjugate()

numpy.ndarray.copy()

brainpy.math.numpy.ndarray.copy()

brainpy.math.jax.ndarray.copy()

numpy.ndarray.cumprod()

brainpy.math.numpy.ndarray.cumprod()

brainpy.math.jax.ndarray.cumprod()

numpy.ndarray.cumsum()

brainpy.math.numpy.ndarray.cumsum()

brainpy.math.jax.ndarray.cumsum()

numpy.ndarray.diagonal()

brainpy.math.numpy.ndarray.diagonal()

brainpy.math.jax.ndarray.diagonal()

numpy.ndarray.dot()

brainpy.math.numpy.ndarray.dot()

brainpy.math.jax.ndarray.dot()

numpy.ndarray.dump()

brainpy.math.numpy.ndarray.dump()

-

numpy.ndarray.dumps()

brainpy.math.numpy.ndarray.dumps()

-

numpy.ndarray.fill()

brainpy.math.numpy.ndarray.fill()

brainpy.math.jax.ndarray.fill()

numpy.ndarray.flatten()

brainpy.math.numpy.ndarray.flatten()

brainpy.math.jax.ndarray.flatten()

numpy.ndarray.getfield()

brainpy.math.numpy.ndarray.getfield()

-

numpy.ndarray.item()

brainpy.math.numpy.ndarray.item()

brainpy.math.jax.ndarray.item()

numpy.ndarray.itemset()

brainpy.math.numpy.ndarray.itemset()

-

numpy.ndarray.max()

brainpy.math.numpy.ndarray.max()

brainpy.math.jax.ndarray.max()

numpy.ndarray.mean()

brainpy.math.numpy.ndarray.mean()

brainpy.math.jax.ndarray.mean()

numpy.ndarray.min()

brainpy.math.numpy.ndarray.min()

brainpy.math.jax.ndarray.min()

numpy.ndarray.newbyteorder()

brainpy.math.numpy.ndarray.newbyteorder()

-

numpy.ndarray.nonzero()

brainpy.math.numpy.ndarray.nonzero()

brainpy.math.jax.ndarray.nonzero()

numpy.ndarray.partition()

brainpy.math.numpy.ndarray.partition()

-

numpy.ndarray.prod()

brainpy.math.numpy.ndarray.prod()

brainpy.math.jax.ndarray.prod()

numpy.ndarray.ptp()

brainpy.math.numpy.ndarray.ptp()

brainpy.math.jax.ndarray.ptp()

numpy.ndarray.put()

brainpy.math.numpy.ndarray.put()

-

numpy.ndarray.ravel()

brainpy.math.numpy.ndarray.ravel()

brainpy.math.jax.ndarray.ravel()

numpy.ndarray.repeat()

brainpy.math.numpy.ndarray.repeat()

brainpy.math.jax.ndarray.repeat()

numpy.ndarray.reshape()

brainpy.math.numpy.ndarray.reshape()

brainpy.math.jax.ndarray.reshape()

numpy.ndarray.resize()

brainpy.math.numpy.ndarray.resize()

-

numpy.ndarray.round()

brainpy.math.numpy.ndarray.round()

brainpy.math.jax.ndarray.round()

numpy.ndarray.searchsorted()

brainpy.math.numpy.ndarray.searchsorted()

brainpy.math.jax.ndarray.searchsorted()

numpy.ndarray.setfield()

brainpy.math.numpy.ndarray.setfield()

-

numpy.ndarray.setflags()

brainpy.math.numpy.ndarray.setflags()

-

numpy.ndarray.sort()

brainpy.math.numpy.ndarray.sort()

brainpy.math.jax.ndarray.sort()

numpy.ndarray.squeeze()

brainpy.math.numpy.ndarray.squeeze()

brainpy.math.jax.ndarray.squeeze()

numpy.ndarray.std()

brainpy.math.numpy.ndarray.std()

brainpy.math.jax.ndarray.std()

numpy.ndarray.sum()

brainpy.math.numpy.ndarray.sum()

brainpy.math.jax.ndarray.sum()

numpy.ndarray.swapaxes()

brainpy.math.numpy.ndarray.swapaxes()

brainpy.math.jax.ndarray.swapaxes()

numpy.ndarray.take()

brainpy.math.numpy.ndarray.take()

brainpy.math.jax.ndarray.take()

numpy.ndarray.tobytes()

brainpy.math.numpy.ndarray.tobytes()

brainpy.math.jax.ndarray.tobytes()

numpy.ndarray.tofile()

brainpy.math.numpy.ndarray.tofile()

-

numpy.ndarray.tolist()

brainpy.math.numpy.ndarray.tolist()

brainpy.math.jax.ndarray.tolist()

numpy.ndarray.tostring()

brainpy.math.numpy.ndarray.tostring()

-

numpy.ndarray.trace()

brainpy.math.numpy.ndarray.trace()

brainpy.math.jax.ndarray.trace()

numpy.ndarray.transpose()

brainpy.math.numpy.ndarray.transpose()

brainpy.math.jax.ndarray.transpose()

numpy.ndarray.var()

brainpy.math.numpy.ndarray.var()

brainpy.math.jax.ndarray.var()

numpy.ndarray.view()

brainpy.math.numpy.ndarray.view()

brainpy.math.jax.ndarray.view()

-

-

brainpy.math.jax.ndarray.block_host_until_ready()

-

-

brainpy.math.jax.ndarray.block_until_ready()

-

-

brainpy.math.jax.ndarray.numpy()

-

-

brainpy.math.jax.ndarray.split()

-

-

brainpy.math.jax.ndarray.tile()

Summary

  • Number of NumPy functions: 56

  • Number of functions covered by brainpy.math.numpy: 56

  • Number of functions covered by brainpy.math.jax: 44

Array Operations

NumPy

brainpy.math.numpy

brainpy.math.jax

numpy.abs

brainpy.math.numpy.abs

brainpy.math.jax.abs

numpy.absolute

brainpy.math.numpy.absolute

brainpy.math.jax.absolute

numpy.add

brainpy.math.numpy.add

brainpy.math.jax.add

numpy.add_docstring

-

-

numpy.add_newdoc

-

-

numpy.add_newdoc_ufunc

-

-

numpy.alen

-

-

numpy.all

brainpy.math.numpy.all

brainpy.math.jax.all

numpy.allclose

brainpy.math.numpy.allclose

brainpy.math.jax.allclose

numpy.alltrue

-

-

numpy.amax

-

-

numpy.amin

-

-

numpy.angle

brainpy.math.numpy.angle

brainpy.math.jax.angle

numpy.any

brainpy.math.numpy.any

brainpy.math.jax.any

numpy.append

brainpy.math.numpy.append

brainpy.math.jax.append

numpy.apply_along_axis

-

-

numpy.apply_over_axes

-

-

numpy.arange

brainpy.math.numpy.arange

brainpy.math.jax.arange

numpy.arccos

brainpy.math.numpy.arccos

brainpy.math.jax.arccos

numpy.arccosh

brainpy.math.numpy.arccosh

brainpy.math.jax.arccosh

numpy.arcsin

brainpy.math.numpy.arcsin

brainpy.math.jax.arcsin

numpy.arcsinh

brainpy.math.numpy.arcsinh

brainpy.math.jax.arcsinh

numpy.arctan

brainpy.math.numpy.arctan

brainpy.math.jax.arctan

numpy.arctan2

brainpy.math.numpy.arctan2

brainpy.math.jax.arctan2

numpy.arctanh

brainpy.math.numpy.arctanh

brainpy.math.jax.arctanh

numpy.argmax

brainpy.math.numpy.argmax

brainpy.math.jax.argmax

numpy.argmin

brainpy.math.numpy.argmin

brainpy.math.jax.argmin

numpy.argpartition

-

-

numpy.argsort

brainpy.math.numpy.argsort

brainpy.math.jax.argsort

numpy.argwhere

brainpy.math.numpy.argwhere

brainpy.math.jax.argwhere

numpy.around

brainpy.math.numpy.around

brainpy.math.jax.around

numpy.array

brainpy.math.numpy.array

brainpy.math.jax.array

numpy.array2string

-

-

numpy.array_equal

brainpy.math.numpy.array_equal

brainpy.math.jax.array_equal

numpy.array_equiv

-

-

numpy.array_repr

-

-

numpy.array_split

-

-

numpy.array_str

-

-

numpy.asanyarray

-

-

numpy.asarray

brainpy.math.numpy.asarray

brainpy.math.jax.asarray

numpy.asarray_chkfinite

-

-

numpy.ascontiguousarray

-

-

numpy.asfarray

-

-

numpy.asfortranarray

-

-

numpy.asmatrix

-

-

numpy.asscalar

-

-

numpy.atleast_1d

brainpy.math.numpy.atleast_1d

brainpy.math.jax.atleast_1d

numpy.atleast_2d

brainpy.math.numpy.atleast_2d

brainpy.math.jax.atleast_2d

numpy.atleast_3d

brainpy.math.numpy.atleast_3d

brainpy.math.jax.atleast_3d

numpy.average

brainpy.math.numpy.average

brainpy.math.jax.average

numpy.bartlett

brainpy.math.numpy.bartlett

brainpy.math.jax.bartlett

numpy.base_repr

-

-

numpy.binary_repr

-

-

numpy.bincount

brainpy.math.numpy.bincount

brainpy.math.jax.bincount

numpy.bitwise_and

brainpy.math.numpy.bitwise_and

brainpy.math.jax.bitwise_and

numpy.bitwise_not

brainpy.math.numpy.bitwise_not

brainpy.math.jax.bitwise_not

numpy.bitwise_or

brainpy.math.numpy.bitwise_or

brainpy.math.jax.bitwise_or

numpy.bitwise_xor

brainpy.math.numpy.bitwise_xor

brainpy.math.jax.bitwise_xor

numpy.blackman

brainpy.math.numpy.blackman

brainpy.math.jax.blackman

numpy.block

-

-

numpy.bmat

-

-

numpy.broadcast_arrays

-

-

numpy.broadcast_shapes

-

-

numpy.broadcast_to

-

-

numpy.busday_count

-

-

numpy.busday_offset

-

-

numpy.byte_bounds

-

-

numpy.can_cast

-

-

numpy.cbrt

brainpy.math.numpy.cbrt

brainpy.math.jax.cbrt

numpy.ceil

brainpy.math.numpy.ceil

brainpy.math.jax.ceil

numpy.choose

-

-

numpy.clip

brainpy.math.numpy.clip

brainpy.math.jax.clip

numpy.column_stack

brainpy.math.numpy.column_stack

brainpy.math.jax.column_stack

numpy.common_type

-

-

numpy.compare_chararrays

-

-

numpy.compress

-

-

numpy.concatenate

brainpy.math.numpy.concatenate

brainpy.math.jax.concatenate

numpy.conj

brainpy.math.numpy.conj

brainpy.math.jax.conj

numpy.conjugate

brainpy.math.numpy.conjugate

brainpy.math.jax.conjugate

numpy.convolve

brainpy.math.numpy.convolve

brainpy.math.jax.convolve

numpy.copy

-

-

numpy.copysign

brainpy.math.numpy.copysign

brainpy.math.jax.copysign

numpy.copyto

-

-

numpy.corrcoef

brainpy.math.numpy.corrcoef

brainpy.math.jax.corrcoef

numpy.correlate

brainpy.math.numpy.correlate

brainpy.math.jax.correlate

numpy.cos

brainpy.math.numpy.cos

brainpy.math.jax.cos

numpy.cosh

brainpy.math.numpy.cosh

brainpy.math.jax.cosh

numpy.count_nonzero

brainpy.math.numpy.count_nonzero

brainpy.math.jax.count_nonzero

numpy.cov

brainpy.math.numpy.cov

brainpy.math.jax.cov

numpy.cross

brainpy.math.numpy.cross

brainpy.math.jax.cross

numpy.cumprod

brainpy.math.numpy.cumprod

brainpy.math.jax.cumprod

numpy.cumproduct

-

-

numpy.cumsum

brainpy.math.numpy.cumsum

brainpy.math.jax.cumsum

numpy.datetime_as_string

-

-

numpy.datetime_data

-

-

numpy.deg2rad

brainpy.math.numpy.deg2rad

brainpy.math.jax.deg2rad

numpy.degrees

brainpy.math.numpy.degrees

brainpy.math.jax.degrees

numpy.delete

-

-

numpy.deprecate

-

-

numpy.deprecate_with_doc

-

-

numpy.diag

brainpy.math.numpy.diag

brainpy.math.jax.diag

numpy.diag_indices

-

-

numpy.diag_indices_from

-

-

numpy.diagflat

-

-

numpy.diagonal

-

-

numpy.diff

brainpy.math.numpy.diff

brainpy.math.jax.diff

numpy.digitize

brainpy.math.numpy.digitize

brainpy.math.jax.digitize

numpy.disp

-

-

numpy.divide

brainpy.math.numpy.divide

brainpy.math.jax.divide

numpy.divmod

brainpy.math.numpy.divmod

brainpy.math.jax.divmod

numpy.dot

brainpy.math.numpy.dot

brainpy.math.jax.dot

numpy.dsplit

brainpy.math.numpy.dsplit

brainpy.math.jax.dsplit

numpy.dstack

brainpy.math.numpy.dstack

brainpy.math.jax.dstack

numpy.ediff1d

brainpy.math.numpy.ediff1d

brainpy.math.jax.ediff1d

numpy.einsum

-

-

numpy.einsum_path

-

-

numpy.empty

brainpy.math.numpy.empty

brainpy.math.jax.empty

numpy.empty_like

brainpy.math.numpy.empty_like

brainpy.math.jax.empty_like

numpy.equal

brainpy.math.numpy.equal

brainpy.math.jax.equal

numpy.exp

brainpy.math.numpy.exp

brainpy.math.jax.exp

numpy.exp2

brainpy.math.numpy.exp2

brainpy.math.jax.exp2

numpy.expand_dims

brainpy.math.numpy.expand_dims

brainpy.math.jax.expand_dims

numpy.expm1

brainpy.math.numpy.expm1

brainpy.math.jax.expm1

numpy.extract

brainpy.math.numpy.extract

brainpy.math.jax.extract

numpy.eye

brainpy.math.numpy.eye

brainpy.math.jax.eye

numpy.fabs

brainpy.math.numpy.fabs

brainpy.math.jax.fabs

numpy.fastCopyAndTranspose

-

-

numpy.fill_diagonal

brainpy.math.numpy.fill_diagonal

brainpy.math.jax.fill_diagonal

numpy.find_common_type

-

-

numpy.fix

brainpy.math.numpy.fix

brainpy.math.jax.fix

numpy.flatnonzero

brainpy.math.numpy.flatnonzero

brainpy.math.jax.flatnonzero

numpy.flip

brainpy.math.numpy.flip

brainpy.math.jax.flip

numpy.fliplr

brainpy.math.numpy.fliplr

brainpy.math.jax.fliplr

numpy.flipud

brainpy.math.numpy.flipud

brainpy.math.jax.flipud

numpy.float_power

brainpy.math.numpy.float_power

brainpy.math.jax.float_power

numpy.floor

brainpy.math.numpy.floor

brainpy.math.jax.floor

numpy.floor_divide

brainpy.math.numpy.floor_divide

brainpy.math.jax.floor_divide

numpy.fmax

brainpy.math.numpy.fmax

brainpy.math.jax.fmax

numpy.fmin

brainpy.math.numpy.fmin

brainpy.math.jax.fmin

numpy.fmod

brainpy.math.numpy.fmod

brainpy.math.jax.fmod

numpy.format_float_positional

-

-

numpy.format_float_scientific

-

-

numpy.frexp

brainpy.math.numpy.frexp

brainpy.math.jax.frexp

numpy.frombuffer

-

-

numpy.fromfile

-

-

numpy.fromfunction

-

-

numpy.fromiter

-

-

numpy.frompyfunc

-

-

numpy.fromregex

-

-

numpy.fromstring

-

-

numpy.full

brainpy.math.numpy.full

brainpy.math.jax.full

numpy.full_like

brainpy.math.numpy.full_like

brainpy.math.jax.full_like

numpy.gcd

brainpy.math.numpy.gcd

brainpy.math.jax.gcd

numpy.genfromtxt

-

-

numpy.geomspace

-

-

numpy.get_array_wrap

-

-

numpy.get_include

-

-

numpy.get_printoptions

-

-

numpy.getbufsize

-

-

numpy.geterr

-

-

numpy.geterrcall

-

-

numpy.geterrobj

-

-

numpy.gradient

-

-

numpy.greater

brainpy.math.numpy.greater

brainpy.math.jax.greater

numpy.greater_equal

brainpy.math.numpy.greater_equal

brainpy.math.jax.greater_equal

numpy.hamming

brainpy.math.numpy.hamming

brainpy.math.jax.hamming

numpy.hanning

brainpy.math.numpy.hanning

brainpy.math.jax.hanning

numpy.heaviside

brainpy.math.numpy.heaviside

brainpy.math.jax.heaviside

numpy.histogram

brainpy.math.numpy.histogram

brainpy.math.jax.histogram

numpy.histogram2d

-

-

numpy.histogram_bin_edges

-

-

numpy.histogramdd

-

-

numpy.hsplit

brainpy.math.numpy.hsplit

brainpy.math.jax.hsplit

numpy.hstack

brainpy.math.numpy.hstack

brainpy.math.jax.hstack

numpy.hypot

brainpy.math.numpy.hypot

brainpy.math.jax.hypot

numpy.i0

-

-

numpy.identity

brainpy.math.numpy.identity

brainpy.math.jax.identity

numpy.imag

brainpy.math.numpy.imag

brainpy.math.jax.imag

numpy.in1d

-

-

numpy.indices

-

-

numpy.info

-

-

numpy.inner

brainpy.math.numpy.inner

brainpy.math.jax.inner

numpy.insert

-

-

numpy.interp

brainpy.math.numpy.interp

brainpy.math.jax.interp

numpy.intersect1d

-

-

numpy.invert

brainpy.math.numpy.invert

brainpy.math.jax.invert

numpy.is_busday

-

-

numpy.isclose

brainpy.math.numpy.isclose

brainpy.math.jax.isclose

numpy.iscomplex

-

-

numpy.iscomplexobj

-

-

numpy.isfinite

brainpy.math.numpy.isfinite

brainpy.math.jax.isfinite

numpy.isfortran

-

-

numpy.isin

-

-

numpy.isinf

brainpy.math.numpy.isinf

brainpy.math.jax.isinf

numpy.isnan

brainpy.math.numpy.isnan

brainpy.math.jax.isnan

numpy.isnat

-

-

numpy.isneginf

-

-

numpy.isposinf

-

-

numpy.isreal

brainpy.math.numpy.isreal

brainpy.math.jax.isreal

numpy.isrealobj

-

-

numpy.isscalar

brainpy.math.numpy.isscalar

brainpy.math.jax.isscalar

numpy.issctype

-

-

numpy.issubclass_

-

-

numpy.issubdtype

-

-

numpy.issubsctype

-

-

numpy.iterable

-

-

numpy.ix_

-

-

numpy.kaiser

brainpy.math.numpy.kaiser

brainpy.math.jax.kaiser

numpy.kron

brainpy.math.numpy.kron

brainpy.math.jax.kron

numpy.lcm

brainpy.math.numpy.lcm

brainpy.math.jax.lcm

numpy.ldexp

brainpy.math.numpy.ldexp

brainpy.math.jax.ldexp

numpy.left_shift

brainpy.math.numpy.left_shift

brainpy.math.jax.left_shift

numpy.less

brainpy.math.numpy.less

brainpy.math.jax.less

numpy.less_equal

brainpy.math.numpy.less_equal

brainpy.math.jax.less_equal

numpy.lexsort

-

-

numpy.linspace

brainpy.math.numpy.linspace

brainpy.math.jax.linspace

numpy.load

-

-

numpy.loads

-

-

numpy.loadtxt

-

-

numpy.log

brainpy.math.numpy.log

brainpy.math.jax.log

numpy.log10

brainpy.math.numpy.log10

brainpy.math.jax.log10

numpy.log1p

brainpy.math.numpy.log1p

brainpy.math.jax.log1p

numpy.log2

brainpy.math.numpy.log2

brainpy.math.jax.log2

numpy.logaddexp

brainpy.math.numpy.logaddexp

brainpy.math.jax.logaddexp

numpy.logaddexp2

brainpy.math.numpy.logaddexp2

brainpy.math.jax.logaddexp2

numpy.logical_and

brainpy.math.numpy.logical_and

brainpy.math.jax.logical_and

numpy.logical_not

brainpy.math.numpy.logical_not

brainpy.math.jax.logical_not

numpy.logical_or

brainpy.math.numpy.logical_or

brainpy.math.jax.logical_or

numpy.logical_xor

brainpy.math.numpy.logical_xor

brainpy.math.jax.logical_xor

numpy.logspace

brainpy.math.numpy.logspace

brainpy.math.jax.logspace

numpy.lookfor

-

-

numpy.mafromtxt

-

-

numpy.mask_indices

-

-

numpy.mat

-

-

numpy.matmul

brainpy.math.numpy.matmul

brainpy.math.jax.matmul

numpy.max

brainpy.math.numpy.max

brainpy.math.jax.max

numpy.maximum

brainpy.math.numpy.maximum

brainpy.math.jax.maximum

numpy.maximum_sctype

-

-

numpy.may_share_memory

-

-

numpy.mean

brainpy.math.numpy.mean

brainpy.math.jax.mean

numpy.median

brainpy.math.numpy.median

brainpy.math.jax.median

numpy.meshgrid

brainpy.math.numpy.meshgrid

brainpy.math.jax.meshgrid

numpy.min

brainpy.math.numpy.min

brainpy.math.jax.min

numpy.min_scalar_type

-

-

numpy.minimum

brainpy.math.numpy.minimum

brainpy.math.jax.minimum

numpy.mintypecode

-

-

numpy.mod

brainpy.math.numpy.mod

brainpy.math.jax.mod

numpy.modf

brainpy.math.numpy.modf

brainpy.math.jax.modf

numpy.moveaxis

brainpy.math.numpy.moveaxis

brainpy.math.jax.moveaxis

numpy.msort

-

-

numpy.multiply

brainpy.math.numpy.multiply

brainpy.math.jax.multiply

numpy.nan_to_num

-

-

numpy.nanargmax

-

-

numpy.nanargmin

-

-

numpy.nancumprod

brainpy.math.numpy.nancumprod

brainpy.math.jax.nancumprod

numpy.nancumsum

brainpy.math.numpy.nancumsum

brainpy.math.jax.nancumsum

numpy.nanmax

brainpy.math.numpy.nanmax

brainpy.math.jax.nanmax

numpy.nanmean

brainpy.math.numpy.nanmean

brainpy.math.jax.nanmean

numpy.nanmedian

brainpy.math.numpy.nanmedian

brainpy.math.jax.nanmedian

numpy.nanmin

brainpy.math.numpy.nanmin

brainpy.math.jax.nanmin

numpy.nanpercentile

brainpy.math.numpy.nanpercentile

brainpy.math.jax.nanpercentile

numpy.nanprod

brainpy.math.numpy.nanprod

brainpy.math.jax.nanprod

numpy.nanquantile

brainpy.math.numpy.nanquantile

brainpy.math.jax.nanquantile

numpy.nanstd

brainpy.math.numpy.nanstd

brainpy.math.jax.nanstd

numpy.nansum

brainpy.math.numpy.nansum

brainpy.math.jax.nansum

numpy.nanvar

brainpy.math.numpy.nanvar

brainpy.math.jax.nanvar

numpy.ndfromtxt

-

-

numpy.ndim

brainpy.math.numpy.ndim

brainpy.math.jax.ndim

numpy.negative

brainpy.math.numpy.negative

brainpy.math.jax.negative

numpy.nested_iters

-

-

numpy.nextafter

brainpy.math.numpy.nextafter

brainpy.math.jax.nextafter

numpy.nonzero

brainpy.math.numpy.nonzero

brainpy.math.jax.nonzero

numpy.not_equal

brainpy.math.numpy.not_equal

brainpy.math.jax.not_equal

numpy.obj2sctype

-

-

numpy.ones

brainpy.math.numpy.ones

brainpy.math.jax.ones

numpy.ones_like

brainpy.math.numpy.ones_like

brainpy.math.jax.ones_like

numpy.outer

brainpy.math.numpy.outer

brainpy.math.jax.outer

numpy.packbits

-

-

numpy.pad

-

-

numpy.partition

-

-

numpy.percentile

brainpy.math.numpy.percentile

brainpy.math.jax.percentile

numpy.piecewise

-

-

numpy.place

-

-

numpy.poly

-

-

numpy.polyadd

-

-

numpy.polyder

-

-

numpy.polydiv

-

-

numpy.polyfit

-

-

numpy.polyint

-

-

numpy.polymul

-

-

numpy.polysub

-

-

numpy.polyval

-

-

numpy.positive

brainpy.math.numpy.positive

brainpy.math.jax.positive

numpy.power

brainpy.math.numpy.power

brainpy.math.jax.power

numpy.printoptions

-

-

numpy.prod

brainpy.math.numpy.prod

brainpy.math.jax.prod

numpy.product

-

-

numpy.promote_types

-

-

numpy.ptp

brainpy.math.numpy.ptp

brainpy.math.jax.ptp

numpy.put

-

-

numpy.put_along_axis

-

-

numpy.putmask

-

-

numpy.quantile

brainpy.math.numpy.quantile

brainpy.math.jax.quantile

numpy.rad2deg

brainpy.math.numpy.rad2deg

brainpy.math.jax.rad2deg

numpy.radians

brainpy.math.numpy.radians

brainpy.math.jax.radians

numpy.ravel

brainpy.math.numpy.ravel

brainpy.math.jax.ravel

numpy.ravel_multi_index

-

-

numpy.real

brainpy.math.numpy.real

brainpy.math.jax.real

numpy.real_if_close

-

-

numpy.recfromcsv

-

-

numpy.recfromtxt

-

-

numpy.reciprocal

brainpy.math.numpy.reciprocal

brainpy.math.jax.reciprocal

numpy.remainder

brainpy.math.numpy.remainder

brainpy.math.jax.remainder

numpy.repeat

brainpy.math.numpy.repeat

brainpy.math.jax.repeat

numpy.require

-

-

numpy.reshape

brainpy.math.numpy.reshape

brainpy.math.jax.reshape

numpy.resize

-

-

numpy.result_type

-

-

numpy.right_shift

brainpy.math.numpy.right_shift

brainpy.math.jax.right_shift

numpy.rint

brainpy.math.numpy.rint

brainpy.math.jax.rint

numpy.roll

brainpy.math.numpy.roll

brainpy.math.jax.roll

numpy.rollaxis

-

-

numpy.roots

-

-

numpy.rot90

-

-

numpy.round

brainpy.math.numpy.round

brainpy.math.jax.round

numpy.round_

brainpy.math.numpy.round_

brainpy.math.jax.round_

numpy.row_stack

-

-

numpy.safe_eval

-

-

numpy.save

-

-

numpy.savetxt

-

-

numpy.savez

-

-

numpy.savez_compressed

-

-

numpy.sctype2char

-

-

numpy.searchsorted

brainpy.math.numpy.searchsorted

brainpy.math.jax.searchsorted

numpy.select

brainpy.math.numpy.select

brainpy.math.jax.select

numpy.set_numeric_ops

-

-

numpy.set_printoptions

-

-

numpy.set_string_function

-

-

numpy.setbufsize

-

-

numpy.setdiff1d

-

-

numpy.seterr

-

-

numpy.seterrcall

-

-

numpy.seterrobj

-

-

numpy.setxor1d

-

-

numpy.shape

brainpy.math.numpy.shape

brainpy.math.jax.shape

numpy.shares_memory

-

-

numpy.show_config

-

-

numpy.sign

brainpy.math.numpy.sign

brainpy.math.jax.sign

numpy.signbit

brainpy.math.numpy.signbit

brainpy.math.jax.signbit

numpy.sin

brainpy.math.numpy.sin

brainpy.math.jax.sin

numpy.sinc

brainpy.math.numpy.sinc

brainpy.math.jax.sinc

numpy.sinh

brainpy.math.numpy.sinh

brainpy.math.jax.sinh

numpy.size

brainpy.math.numpy.size

brainpy.math.jax.size

numpy.sometrue

-

-

numpy.sort

brainpy.math.numpy.sort

brainpy.math.jax.sort

numpy.sort_complex

-

-

numpy.source

-

-

numpy.spacing

-

-

numpy.split

brainpy.math.numpy.split

brainpy.math.jax.split

numpy.sqrt

brainpy.math.numpy.sqrt

brainpy.math.jax.sqrt

numpy.square

brainpy.math.numpy.square

brainpy.math.jax.square

numpy.squeeze

brainpy.math.numpy.squeeze

brainpy.math.jax.squeeze

numpy.stack

brainpy.math.numpy.stack

brainpy.math.jax.stack

numpy.std

brainpy.math.numpy.std

brainpy.math.jax.std

numpy.subtract

brainpy.math.numpy.subtract

brainpy.math.jax.subtract

numpy.sum

brainpy.math.numpy.sum

brainpy.math.jax.sum

numpy.swapaxes

brainpy.math.numpy.swapaxes

brainpy.math.jax.swapaxes

numpy.take

brainpy.math.numpy.take

brainpy.math.jax.take

numpy.take_along_axis

brainpy.math.numpy.take_along_axis

brainpy.math.jax.take_along_axis

numpy.tan

brainpy.math.numpy.tan

brainpy.math.jax.tan

numpy.tanh

brainpy.math.numpy.tanh

brainpy.math.jax.tanh

numpy.tensordot

-

-

numpy.tile

brainpy.math.numpy.tile

brainpy.math.jax.tile

numpy.trace

brainpy.math.numpy.trace

brainpy.math.jax.trace

numpy.transpose

brainpy.math.numpy.transpose

brainpy.math.jax.transpose

numpy.trapz

brainpy.math.numpy.trapz

brainpy.math.jax.trapz

numpy.tri

brainpy.math.numpy.tri

brainpy.math.jax.tri

numpy.tril

brainpy.math.numpy.tril

brainpy.math.jax.tril

numpy.tril_indices

brainpy.math.numpy.tril_indices

brainpy.math.jax.tril_indices

numpy.tril_indices_from

brainpy.math.numpy.tril_indices_from

brainpy.math.jax.tril_indices_from

numpy.trim_zeros

-

-

numpy.triu

brainpy.math.numpy.triu

brainpy.math.jax.triu

numpy.triu_indices

brainpy.math.numpy.triu_indices

brainpy.math.jax.triu_indices

numpy.triu_indices_from

brainpy.math.numpy.triu_indices_from

brainpy.math.jax.triu_indices_from

numpy.true_divide

brainpy.math.numpy.true_divide

brainpy.math.jax.true_divide

numpy.trunc

brainpy.math.numpy.trunc

brainpy.math.jax.trunc

numpy.typename

-

-

numpy.union1d

-

-

numpy.unique

brainpy.math.numpy.unique

brainpy.math.jax.unique

numpy.unpackbits

-

-

numpy.unravel_index

-

-

numpy.unwrap

-

-

numpy.vander

brainpy.math.numpy.vander

brainpy.math.jax.vander

numpy.var

brainpy.math.numpy.var

brainpy.math.jax.var

numpy.vdot

brainpy.math.numpy.vdot

brainpy.math.jax.vdot

numpy.vsplit

brainpy.math.numpy.vsplit

brainpy.math.jax.vsplit

numpy.vstack

brainpy.math.numpy.vstack

brainpy.math.jax.vstack

numpy.where

brainpy.math.numpy.where

brainpy.math.jax.where

numpy.who

-

-

numpy.zeros

brainpy.math.numpy.zeros

brainpy.math.jax.zeros

numpy.zeros_like

brainpy.math.numpy.zeros_like

brainpy.math.jax.zeros_like

-

brainpy.math.numpy.clip_by_norm

brainpy.math.jax.clip_by_norm

-

brainpy.math.numpy.function

brainpy.math.jax.function

-

brainpy.math.numpy.set_complex_

brainpy.math.jax.set_complex_

-

brainpy.math.numpy.set_float_

brainpy.math.jax.set_float_

-

brainpy.math.numpy.set_int_

brainpy.math.jax.set_int_

Summary

  • Number of NumPy functions: 401

  • Number of functions covered by brainpy.math.numpy: 225

  • Number of functions covered by brainpy.math.jax: 225

Linear Algebra

NumPy

brainpy.math.numpy

brainpy.math.jax

numpy.linalg.cholesky

brainpy.math.numpy.linalg.cholesky

brainpy.math.jax.linalg.cholesky

numpy.linalg.cond

brainpy.math.numpy.linalg.cond

brainpy.math.jax.linalg.cond

numpy.linalg.det

brainpy.math.numpy.linalg.det

brainpy.math.jax.linalg.det

numpy.linalg.eig

brainpy.math.numpy.linalg.eig

brainpy.math.jax.linalg.eig

numpy.linalg.eigh

brainpy.math.numpy.linalg.eigh

brainpy.math.jax.linalg.eigh

numpy.linalg.eigvals

brainpy.math.numpy.linalg.eigvals

brainpy.math.jax.linalg.eigvals

numpy.linalg.eigvalsh

brainpy.math.numpy.linalg.eigvalsh

brainpy.math.jax.linalg.eigvalsh

numpy.linalg.inv

brainpy.math.numpy.linalg.inv

brainpy.math.jax.linalg.inv

numpy.linalg.lstsq

brainpy.math.numpy.linalg.lstsq

brainpy.math.jax.linalg.lstsq

numpy.linalg.matrix_power

brainpy.math.numpy.linalg.matrix_power

brainpy.math.jax.linalg.matrix_power

numpy.linalg.matrix_rank

brainpy.math.numpy.linalg.matrix_rank

brainpy.math.jax.linalg.matrix_rank

numpy.linalg.multi_dot

-

-

numpy.linalg.norm

brainpy.math.numpy.linalg.norm

brainpy.math.jax.linalg.norm

numpy.linalg.pinv

brainpy.math.numpy.linalg.pinv

brainpy.math.jax.linalg.pinv

numpy.linalg.qr

brainpy.math.numpy.linalg.qr

brainpy.math.jax.linalg.qr

numpy.linalg.slogdet

-

-

numpy.linalg.solve

-

-

numpy.linalg.svd

brainpy.math.numpy.linalg.svd

brainpy.math.jax.linalg.svd

numpy.linalg.tensorinv

-

-

numpy.linalg.tensorsolve

-

-

Summary

  • Number of NumPy functions: 20

  • Number of functions covered by brainpy.math.numpy: 15

  • Number of functions covered by brainpy.math.jax: 15

Discrete Fourier Transform

NumPy

brainpy.math.numpy

brainpy.math.jax

numpy.fft.fft

brainpy.math.numpy.fft.fft

brainpy.math.jax.fft.fft

numpy.fft.fft2

brainpy.math.numpy.fft.fft2

brainpy.math.jax.fft.fft2

numpy.fft.fftfreq

brainpy.math.numpy.fft.fftfreq

brainpy.math.jax.fft.fftfreq

numpy.fft.fftn

brainpy.math.numpy.fft.fftn

brainpy.math.jax.fft.fftn

numpy.fft.fftshift

brainpy.math.numpy.fft.fftshift

brainpy.math.jax.fft.fftshift

numpy.fft.hfft

brainpy.math.numpy.fft.hfft

brainpy.math.jax.fft.hfft

numpy.fft.ifft

brainpy.math.numpy.fft.ifft

brainpy.math.jax.fft.ifft

numpy.fft.ifft2

brainpy.math.numpy.fft.ifft2

brainpy.math.jax.fft.ifft2

numpy.fft.ifftn

brainpy.math.numpy.fft.ifftn

brainpy.math.jax.fft.ifftn

numpy.fft.ifftshift

brainpy.math.numpy.fft.ifftshift

brainpy.math.jax.fft.ifftshift

numpy.fft.ihfft

brainpy.math.numpy.fft.ihfft

brainpy.math.jax.fft.ihfft

numpy.fft.irfft

brainpy.math.numpy.fft.irfft

brainpy.math.jax.fft.irfft

numpy.fft.irfft2

brainpy.math.numpy.fft.irfft2

brainpy.math.jax.fft.irfft2

numpy.fft.irfftn

brainpy.math.numpy.fft.irfftn

brainpy.math.jax.fft.irfftn

numpy.fft.rfft

brainpy.math.numpy.fft.rfft

brainpy.math.jax.fft.rfft

numpy.fft.rfft2

brainpy.math.numpy.fft.rfft2

brainpy.math.jax.fft.rfft2

numpy.fft.rfftfreq

brainpy.math.numpy.fft.rfftfreq

brainpy.math.jax.fft.rfftfreq

numpy.fft.rfftn

brainpy.math.numpy.fft.rfftn

brainpy.math.jax.fft.rfftn

Summary

  • Number of NumPy functions: 18

  • Number of functions covered by brainpy.math.numpy: 18

  • Number of functions covered by brainpy.math.jax: 18

Random Sampling

NumPy

brainpy.math.numpy

brainpy.math.jax

numpy.random.beta

brainpy.math.numpy.random.beta

brainpy.math.jax.random.beta

numpy.random.binomial

-

-

numpy.random.bytes

-

-

numpy.random.chisquare

-

-

numpy.random.choice

brainpy.math.numpy.random.choice

brainpy.math.jax.random.choice

numpy.random.default_rng

-

-

numpy.random.dirichlet

-

-

numpy.random.exponential

brainpy.math.numpy.random.exponential

brainpy.math.jax.random.exponential

numpy.random.f

-

-

numpy.random.gamma

brainpy.math.numpy.random.gamma

brainpy.math.jax.random.gamma

numpy.random.geometric

-

-

numpy.random.get_state

-

-

numpy.random.gumbel

brainpy.math.numpy.random.gumbel

brainpy.math.jax.random.gumbel

numpy.random.hypergeometric

-

-

numpy.random.laplace

brainpy.math.numpy.random.laplace

brainpy.math.jax.random.laplace

numpy.random.logistic

brainpy.math.numpy.random.logistic

brainpy.math.jax.random.logistic

numpy.random.lognormal

-

-

numpy.random.logseries

-

-

numpy.random.multinomial

-

-

numpy.random.multivariate_normal

-

-

numpy.random.negative_binomial

-

-

numpy.random.noncentral_chisquare

-

-

numpy.random.noncentral_f

-

-

numpy.random.normal

brainpy.math.numpy.random.normal

brainpy.math.jax.random.normal

numpy.random.pareto

brainpy.math.numpy.random.pareto

brainpy.math.jax.random.pareto

numpy.random.permutation

brainpy.math.numpy.random.permutation

brainpy.math.jax.random.permutation

numpy.random.poisson

brainpy.math.numpy.random.poisson

brainpy.math.jax.random.poisson

numpy.random.power

-

-

numpy.random.rand

brainpy.math.numpy.random.rand

brainpy.math.jax.random.rand

numpy.random.randint

brainpy.math.numpy.random.randint

brainpy.math.jax.random.randint

numpy.random.randn

brainpy.math.numpy.random.randn

brainpy.math.jax.random.randn

numpy.random.random

brainpy.math.numpy.random.random

brainpy.math.jax.random.random

numpy.random.random_integers

-

-

numpy.random.random_sample

brainpy.math.numpy.random.random_sample

brainpy.math.jax.random.random_sample

numpy.random.ranf

brainpy.math.numpy.random.ranf

brainpy.math.jax.random.ranf

numpy.random.rayleigh

-

-

numpy.random.sample

brainpy.math.numpy.random.sample

brainpy.math.jax.random.sample

numpy.random.seed

brainpy.math.numpy.random.seed

brainpy.math.jax.random.seed

numpy.random.set_state

-

-

numpy.random.shuffle

brainpy.math.numpy.random.shuffle

brainpy.math.jax.random.shuffle

numpy.random.standard_cauchy

brainpy.math.numpy.random.standard_cauchy

brainpy.math.jax.random.standard_cauchy

numpy.random.standard_exponential

brainpy.math.numpy.random.standard_exponential

brainpy.math.jax.random.standard_exponential

numpy.random.standard_gamma

brainpy.math.numpy.random.standard_gamma

brainpy.math.jax.random.standard_gamma

numpy.random.standard_normal

brainpy.math.numpy.random.standard_normal

brainpy.math.jax.random.standard_normal

numpy.random.standard_t

brainpy.math.numpy.random.standard_t

brainpy.math.jax.random.standard_t

numpy.random.triangular

-

-

numpy.random.uniform

brainpy.math.numpy.random.uniform

brainpy.math.jax.random.uniform

numpy.random.vonmises

-

-

numpy.random.wald

-

-

numpy.random.weibull

-

-

numpy.random.zipf

-

-

-

brainpy.math.numpy.random.bernoulli

brainpy.math.jax.random.bernoulli

-

-

brainpy.math.jax.random.copy_doc

-

brainpy.math.numpy.random.numba_seed

-

-

brainpy.math.numpy.random.truncated_normal

brainpy.math.jax.random.truncated_normal

Summary

  • Number of NumPy functions: 51

  • Number of functions covered by brainpy.math.numpy: 26

  • Number of functions covered by brainpy.math.jax: 26