nodepyx 1.0.0

package/src/types/stubs/numpy.d.ts

@@ -0,0 +1,441 @@
+// =============================================================
+// nodepyx — Pre-built TypeScript stub for numpy
+// Version: 1.x / 2.x
+// Source: https://numpy.org
+// =============================================================
+
+import type { PyProxy } from 'nodepyx';
+
+declare module 'nodepyx/numpy' {
+
+  // ── TypedArray alias ──────────────────────────────────────────────────────
+
+  export type NDArray =
+    | Float32Array
+    | Float64Array
+    | Int8Array
+    | Int16Array
+    | Int32Array
+    | BigInt64Array
+    | Uint8Array
+    | Uint16Array
+    | Uint32Array
+    | BigUint64Array;
+
+  // ── ndarray class ─────────────────────────────────────────────────────────
+
+  export class ndarray {
+    constructor(shape: number[], dtype?: string, buffer?: ArrayBuffer, strides?: number[]);
+
+    readonly shape: Promise<number[]>;
+    readonly dtype: Promise<string>;
+    readonly ndim: Promise<number>;
+    readonly size: Promise<number>;
+    readonly nbytes: Promise<number>;
+    readonly itemsize: Promise<number>;
+    readonly strides: Promise<number[]>;
+    readonly T: Promise<NDArray>;
+    readonly flat: PyProxy;
+    readonly real: Promise<NDArray>;
+    readonly imag: Promise<NDArray>;
+
+    // Conversion
+    tolist(): Promise<unknown[]>;
+    tobytes(order?: 'C' | 'F'): Promise<Uint8Array>;
+    astype(dtype: string, order?: string, copy?: boolean): Promise<NDArray>;
+    copy(order?: 'C' | 'F'): Promise<NDArray>;
+    flatten(order?: 'C' | 'F'): Promise<NDArray>;
+    ravel(order?: 'C' | 'F'): Promise<NDArray>;
+    reshape(...shape: number[]): Promise<NDArray>;
+    resize(new_shape: number | number[]): Promise<void>;
+    squeeze(axis?: number | number[]): Promise<NDArray>;
+    transpose(...axes: number[]): Promise<NDArray>;
+
+    // Reduction
+    sum(axis?: number | number[], dtype?: string, keepdims?: boolean): Promise<number | NDArray>;
+    mean(axis?: number | number[], dtype?: string, keepdims?: boolean): Promise<number | NDArray>;
+    std(axis?: number | number[], dtype?: string, ddof?: number, keepdims?: boolean): Promise<number | NDArray>;
+    var(axis?: number | number[], dtype?: string, ddof?: number): Promise<number | NDArray>;
+    min(axis?: number | number[], keepdims?: boolean): Promise<number | NDArray>;
+    max(axis?: number | number[], keepdims?: boolean): Promise<number | NDArray>;
+    argmin(axis?: number): Promise<number | NDArray>;
+    argmax(axis?: number): Promise<number | NDArray>;
+    cumsum(axis?: number, dtype?: string): Promise<NDArray>;
+    cumprod(axis?: number, dtype?: string): Promise<NDArray>;
+    prod(axis?: number | number[], dtype?: string, keepdims?: boolean): Promise<number | NDArray>;
+    all(axis?: number | number[], keepdims?: boolean): Promise<boolean | NDArray>;
+    any(axis?: number | number[], keepdims?: boolean): Promise<boolean | NDArray>;
+    trace(offset?: number, axis1?: number, axis2?: number): Promise<number | NDArray>;
+    ptp(axis?: number): Promise<number | NDArray>;
+
+    // Linear algebra
+    dot(b: NDArray | PyProxy): Promise<NDArray | number>;
+    cross(b: NDArray | PyProxy, axisa?: number, axisb?: number): Promise<NDArray>;
+
+    // Search / sort
+    argsort(axis?: number, kind?: string): Promise<NDArray>;
+    sort(axis?: number, kind?: string): Promise<void>;
+    searchsorted(v: number | NDArray, side?: 'left' | 'right'): Promise<number | NDArray>;
+    nonzero(): Promise<NDArray[]>;
+    where(): Promise<NDArray>;
+
+    // Item access
+    item(...args: number[]): Promise<number | boolean | string>;
+    itemset(...args: unknown[]): Promise<void>;
+    take(indices: number | number[] | NDArray, axis?: number): Promise<NDArray>;
+    put(indices: number | number[], values: number | number[]): Promise<void>;
+    choose(choices: NDArray[] | PyProxy): Promise<NDArray>;
+    compress(condition: NDArray, axis?: number): Promise<NDArray>;
+    diagonal(offset?: number, axis1?: number, axis2?: number): Promise<NDArray>;
+
+    // Utility
+    clip(a_min: number | null, a_max: number | null): Promise<NDArray>;
+    round(decimals?: number): Promise<NDArray>;
+    conj(): Promise<NDArray>;
+    conjugate(): Promise<NDArray>;
+    view(dtype?: string): Promise<NDArray>;
+    byteswap(): Promise<NDArray>;
+    fill(value: number): Promise<void>;
+    repeat(repeats: number | number[], axis?: number): Promise<NDArray>;
+    tile(reps: number | number[]): Promise<NDArray>;
+
+    __len__(): Promise<number>;
+    [key: string]: unknown;
+  }
+
+  // ── Array creation ─────────────────────────────────────────────────────────
+
+  export function array(
+    object: unknown[] | NDArray | PyProxy,
+    dtype?: string,
+    copy?: boolean,
+    order?: 'K' | 'A' | 'C' | 'F',
+    ndmin?: number,
+  ): Promise<NDArray>;
+
+  export function asarray(a: unknown, dtype?: string, order?: 'C' | 'F'): Promise<NDArray>;
+  export function ascontiguousarray(a: unknown, dtype?: string): Promise<NDArray>;
+  export function asfortranarray(a: unknown, dtype?: string): Promise<NDArray>;
+
+  export function zeros(shape: number | number[], dtype?: string, order?: 'C' | 'F'): Promise<NDArray>;
+  export function ones(shape: number | number[], dtype?: string, order?: 'C' | 'F'): Promise<NDArray>;
+  export function empty(shape: number | number[], dtype?: string, order?: 'C' | 'F'): Promise<NDArray>;
+  export function full(shape: number | number[], fill_value: number, dtype?: string): Promise<NDArray>;
+  export function eye(N: number, M?: number, k?: number, dtype?: string): Promise<NDArray>;
+  export function identity(n: number, dtype?: string): Promise<NDArray>;
+  export function zeros_like(a: NDArray | PyProxy, dtype?: string): Promise<NDArray>;
+  export function ones_like(a: NDArray | PyProxy, dtype?: string): Promise<NDArray>;
+  export function empty_like(a: NDArray | PyProxy, dtype?: string): Promise<NDArray>;
+  export function full_like(a: NDArray | PyProxy, fill_value: number, dtype?: string): Promise<NDArray>;
+
+  export function arange(
+    start: number,
+    stop?: number,
+    step?: number,
+    dtype?: string,
+  ): Promise<NDArray>;
+
+  export function linspace(
+    start: number,
+    stop: number,
+    num?: number,
+    endpoint?: boolean,
+    retstep?: false,
+    dtype?: string,
+  ): Promise<NDArray>;
+
+  export function logspace(
+    start: number,
+    stop: number,
+    num?: number,
+    endpoint?: boolean,
+    base?: number,
+    dtype?: string,
+  ): Promise<NDArray>;
+
+  export function geomspace(start: number, stop: number, num?: number, endpoint?: boolean, dtype?: string): Promise<NDArray>;
+
+  export function meshgrid(...xi: NDArray[]): Promise<NDArray[]>;
+  export const mgrid: PyProxy;
+  export const ogrid: PyProxy;
+
+  export function fromiter(iterable: PyProxy, dtype: string, count?: number): Promise<NDArray>;
+  export function frombuffer(buffer: ArrayBuffer | Uint8Array, dtype?: string, count?: number, offset?: number): Promise<NDArray>;
+  export function fromfunction(func: PyProxy | ((...args: unknown[]) => unknown), shape: number[], dtype?: string): Promise<NDArray>;
+
+  // ── Manipulation ──────────────────────────────────────────────────────────
+
+  export function reshape(a: NDArray | PyProxy, newshape: number | number[], order?: 'C' | 'F' | 'A'): Promise<NDArray>;
+  export function ravel(a: NDArray | PyProxy, order?: 'C' | 'F'): Promise<NDArray>;
+  export function flatten(a: NDArray | PyProxy, order?: 'C' | 'F'): Promise<NDArray>;
+  export function squeeze(a: NDArray | PyProxy, axis?: number | number[]): Promise<NDArray>;
+  export function expand_dims(a: NDArray | PyProxy, axis: number | number[]): Promise<NDArray>;
+  export function transpose(a: NDArray | PyProxy, axes?: number[]): Promise<NDArray>;
+  export function swapaxes(a: NDArray | PyProxy, axis1: number, axis2: number): Promise<NDArray>;
+  export function moveaxis(a: NDArray | PyProxy, source: number | number[], destination: number | number[]): Promise<NDArray>;
+  export function rollaxis(a: NDArray | PyProxy, axis: number, start?: number): Promise<NDArray>;
+  export function flip(m: NDArray | PyProxy, axis?: number | number[]): Promise<NDArray>;
+  export function fliplr(m: NDArray | PyProxy): Promise<NDArray>;
+  export function flipud(m: NDArray | PyProxy): Promise<NDArray>;
+  export function rot90(m: NDArray | PyProxy, k?: number, axes?: [number, number]): Promise<NDArray>;
+  export function roll(a: NDArray | PyProxy, shift: number | number[], axis?: number | number[]): Promise<NDArray>;
+
+  export function concatenate(a: NDArray[] | PyProxy, axis?: number): Promise<NDArray>;
+  export function stack(arrays: NDArray[] | PyProxy, axis?: number): Promise<NDArray>;
+  export function vstack(tup: NDArray[] | PyProxy): Promise<NDArray>;
+  export function hstack(tup: NDArray[] | PyProxy): Promise<NDArray>;
+  export function dstack(tup: NDArray[] | PyProxy): Promise<NDArray>;
+  export function block(arrays: unknown[]): Promise<NDArray>;
+  export function split(ary: NDArray | PyProxy, indices_or_sections: number | number[], axis?: number): Promise<NDArray[]>;
+  export function hsplit(ary: NDArray | PyProxy, indices_or_sections: number | number[]): Promise<NDArray[]>;
+  export function vsplit(ary: NDArray | PyProxy, indices_or_sections: number | number[]): Promise<NDArray[]>;
+  export function dsplit(ary: NDArray | PyProxy, indices_or_sections: number | number[]): Promise<NDArray[]>;
+  export function tile(A: NDArray | PyProxy, reps: number | number[]): Promise<NDArray>;
+  export function repeat(a: NDArray | PyProxy, repeats: number | number[], axis?: number): Promise<NDArray>;
+  export function delete_(arr: NDArray | PyProxy, obj: number | number[], axis?: number): Promise<NDArray>;
+  export function insert(arr: NDArray | PyProxy, obj: number | number[], values: unknown, axis?: number): Promise<NDArray>;
+  export function append(arr: NDArray | PyProxy, values: NDArray | PyProxy, axis?: number): Promise<NDArray>;
+  export function resize(a: NDArray | PyProxy, new_shape: number | number[]): Promise<NDArray>;
+  export function unique(ar: NDArray | PyProxy, return_index?: boolean, return_inverse?: boolean, return_counts?: boolean): Promise<NDArray | [NDArray, NDArray, NDArray, NDArray]>;
+
+  // ── Mathematics ───────────────────────────────────────────────────────────
+
+  // Trigonometric
+  export function sin(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function cos(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function tan(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function arcsin(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function arccos(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function arctan(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function arctan2(y: NDArray | number | PyProxy, x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function sinh(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function cosh(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function tanh(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function deg2rad(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function rad2deg(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+
+  // Exponential / logarithmic
+  export function exp(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function exp2(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function log(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function log2(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function log10(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function log1p(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function expm1(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function power(x: NDArray | number | PyProxy, y: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function sqrt(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function cbrt(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function square(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+
+  // Rounding
+  export function round_(a: NDArray | number | PyProxy, decimals?: number): Promise<NDArray | number>;
+  export function around(a: NDArray | number | PyProxy, decimals?: number): Promise<NDArray | number>;
+  export function floor(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function ceil(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function trunc(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function fix(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+
+  // Arithmetic
+  export function add(x1: NDArray | number | PyProxy, x2: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function subtract(x1: NDArray | number | PyProxy, x2: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function multiply(x1: NDArray | number | PyProxy, x2: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function divide(x1: NDArray | number | PyProxy, x2: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function true_divide(x1: NDArray | number | PyProxy, x2: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function floor_divide(x1: NDArray | number | PyProxy, x2: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function mod(x1: NDArray | number | PyProxy, x2: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function remainder(x1: NDArray | number | PyProxy, x2: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function absolute(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function abs(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function sign(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+  export function clip(a: NDArray | PyProxy, a_min: number | null, a_max: number | null): Promise<NDArray>;
+  export function negative(x: NDArray | number | PyProxy): Promise<NDArray | number>;
+
+  // Aggregation
+  export function sum(a: NDArray | PyProxy, axis?: number | number[], dtype?: string, keepdims?: boolean): Promise<number | NDArray>;
+  export function prod(a: NDArray | PyProxy, axis?: number | number[], dtype?: string): Promise<number | NDArray>;
+  export function mean(a: NDArray | PyProxy, axis?: number | number[], dtype?: string, keepdims?: boolean): Promise<number | NDArray>;
+  export function std(a: NDArray | PyProxy, axis?: number | number[], dtype?: string, ddof?: number): Promise<number | NDArray>;
+  export function var_(a: NDArray | PyProxy, axis?: number | number[], dtype?: string, ddof?: number): Promise<number | NDArray>;
+  export function min(a: NDArray | PyProxy, axis?: number | number[]): Promise<number | NDArray>;
+  export function max(a: NDArray | PyProxy, axis?: number | number[]): Promise<number | NDArray>;
+  export function amin(a: NDArray | PyProxy, axis?: number | number[]): Promise<number | NDArray>;
+  export function amax(a: NDArray | PyProxy, axis?: number | number[]): Promise<number | NDArray>;
+  export function argmin(a: NDArray | PyProxy, axis?: number): Promise<number | NDArray>;
+  export function argmax(a: NDArray | PyProxy, axis?: number): Promise<number | NDArray>;
+  export function cumsum(a: NDArray | PyProxy, axis?: number, dtype?: string): Promise<NDArray>;
+  export function cumprod(a: NDArray | PyProxy, axis?: number, dtype?: string): Promise<NDArray>;
+  export function diff(a: NDArray | PyProxy, n?: number, axis?: number): Promise<NDArray>;
+  export function gradient(f: NDArray | PyProxy, varargs?: NDArray | number): Promise<NDArray | NDArray[]>;
+  export function median(a: NDArray | PyProxy, axis?: number, keepdims?: boolean): Promise<number | NDArray>;
+  export function average(a: NDArray | PyProxy, axis?: number, weights?: NDArray): Promise<number | NDArray>;
+  export function percentile(a: NDArray | PyProxy, q: number | number[], axis?: number): Promise<number | NDArray>;
+  export function quantile(a: NDArray | PyProxy, q: number | number[], axis?: number): Promise<number | NDArray>;
+  export function nanmean(a: NDArray | PyProxy, axis?: number): Promise<number | NDArray>;
+  export function nanstd(a: NDArray | PyProxy, axis?: number, ddof?: number): Promise<number | NDArray>;
+  export function nansum(a: NDArray | PyProxy, axis?: number): Promise<number | NDArray>;
+  export function nanmin(a: NDArray | PyProxy, axis?: number): Promise<number | NDArray>;
+  export function nanmax(a: NDArray | PyProxy, axis?: number): Promise<number | NDArray>;
+
+  // Comparison
+  export function all(a: NDArray | PyProxy, axis?: number, keepdims?: boolean): Promise<boolean | NDArray>;
+  export function any(a: NDArray | PyProxy, axis?: number, keepdims?: boolean): Promise<boolean | NDArray>;
+  export function equal(x1: NDArray | PyProxy, x2: NDArray | PyProxy): Promise<NDArray>;
+  export function not_equal(x1: NDArray | PyProxy, x2: NDArray | PyProxy): Promise<NDArray>;
+  export function greater(x1: NDArray | PyProxy, x2: NDArray | PyProxy): Promise<NDArray>;
+  export function greater_equal(x1: NDArray | PyProxy, x2: NDArray | PyProxy): Promise<NDArray>;
+  export function less(x1: NDArray | PyProxy, x2: NDArray | PyProxy): Promise<NDArray>;
+  export function less_equal(x1: NDArray | PyProxy, x2: NDArray | PyProxy): Promise<NDArray>;
+  export function isnan(x: NDArray | PyProxy): Promise<NDArray | boolean>;
+  export function isinf(x: NDArray | PyProxy): Promise<NDArray | boolean>;
+  export function isfinite(x: NDArray | PyProxy): Promise<NDArray | boolean>;
+  export function isneginf(x: NDArray | PyProxy): Promise<NDArray | boolean>;
+  export function isposinf(x: NDArray | PyProxy): Promise<NDArray | boolean>;
+
+  // ── Linear Algebra ─────────────────────────────────────────────────────────
+
+  export namespace linalg {
+    export function dot(a: NDArray | PyProxy, b: NDArray | PyProxy): Promise<NDArray | number>;
+    export function vdot(a: NDArray | PyProxy, b: NDArray | PyProxy): Promise<number>;
+    export function inner(a: NDArray | PyProxy, b: NDArray | PyProxy): Promise<NDArray | number>;
+    export function outer(a: NDArray | PyProxy, b: NDArray | PyProxy): Promise<NDArray>;
+    export function matmul(x1: NDArray | PyProxy, x2: NDArray | PyProxy): Promise<NDArray>;
+    export function matrix_power(M: NDArray | PyProxy, n: number): Promise<NDArray>;
+    export function tensordot(a: NDArray | PyProxy, b: NDArray | PyProxy, axes?: number | number[][]): Promise<NDArray>;
+
+    export function det(a: NDArray | PyProxy): Promise<number>;
+    export function inv(a: NDArray | PyProxy): Promise<NDArray>;
+    export function pinv(a: NDArray | PyProxy, rcond?: number): Promise<NDArray>;
+    export function solve(a: NDArray | PyProxy, b: NDArray | PyProxy): Promise<NDArray>;
+    export function lstsq(a: NDArray | PyProxy, b: NDArray | PyProxy, rcond?: number | null): Promise<[NDArray, NDArray, number, NDArray]>;
+
+    export function svd(a: NDArray | PyProxy, full_matrices?: boolean, compute_uv?: boolean): Promise<[NDArray, NDArray, NDArray]>;
+    export function eig(a: NDArray | PyProxy): Promise<[NDArray, NDArray]>;
+    export function eigh(a: NDArray | PyProxy, UPLO?: 'L' | 'U'): Promise<[NDArray, NDArray]>;
+    export function eigvals(a: NDArray | PyProxy): Promise<NDArray>;
+    export function eigvalsh(a: NDArray | PyProxy, UPLO?: 'L' | 'U'): Promise<NDArray>;
+
+    export function cholesky(a: NDArray | PyProxy): Promise<NDArray>;
+    export function qr(a: NDArray | PyProxy, mode?: 'reduced' | 'complete' | 'r' | 'raw'): Promise<[NDArray, NDArray]>;
+    export function norm(x: NDArray | PyProxy, ord?: number | string, axis?: number, keepdims?: boolean): Promise<number | NDArray>;
+    export function cond(x: NDArray | PyProxy, p?: number | string): Promise<number>;
+    export function matrix_rank(M: NDArray | PyProxy, tol?: number): Promise<number>;
+    export function slogdet(a: NDArray | PyProxy): Promise<[number, number]>;
+    export function trace(a: NDArray | PyProxy, offset?: number, axis1?: number, axis2?: number): Promise<number | NDArray>;
+    export function cross(a: NDArray | PyProxy, b: NDArray | PyProxy): Promise<NDArray>;
+  }
+
+  // ── Random module ─────────────────────────────────────────────────────────
+
+  export namespace random {
+    export function seed(seed?: number): Promise<void>;
+    export function rand(...d: number[]): Promise<NDArray>;
+    export function randn(...d: number[]): Promise<NDArray>;
+    export function randint(low: number, high?: number, size?: number | number[], dtype?: string): Promise<NDArray | number>;
+    export function random(size?: number | number[]): Promise<NDArray | number>;
+    export function uniform(low?: number, high?: number, size?: number | number[]): Promise<NDArray | number>;
+    export function normal(loc?: number, scale?: number, size?: number | number[]): Promise<NDArray | number>;
+    export function standard_normal(size?: number | number[]): Promise<NDArray | number>;
+    export function exponential(scale?: number, size?: number | number[]): Promise<NDArray | number>;
+    export function poisson(lam?: number, size?: number | number[]): Promise<NDArray | number>;
+    export function binomial(n: number, p: number, size?: number | number[]): Promise<NDArray | number>;
+    export function choice(a: number | NDArray | PyProxy, size?: number | number[], replace?: boolean, p?: NDArray): Promise<NDArray | unknown>;
+    export function shuffle(x: NDArray | PyProxy): Promise<void>;
+    export function permutation(x: number | NDArray | PyProxy): Promise<NDArray>;
+    export function multivariate_normal(mean: NDArray | number[], cov: NDArray | number[][], size?: number | number[]): Promise<NDArray>;
+    export function beta(a: number, b: number, size?: number | number[]): Promise<NDArray | number>;
+    export function gamma(shape: number, scale?: number, size?: number | number[]): Promise<NDArray | number>;
+    export function dirichlet(alpha: NDArray | number[], size?: number | number[]): Promise<NDArray>;
+    export function get_state(): Promise<PyProxy>;
+    export function set_state(state: PyProxy): Promise<void>;
+
+    export class Generator {
+      constructor(bit_generator?: PyProxy);
+      random(size?: number | number[]): Promise<NDArray | number>;
+      uniform(low?: number, high?: number, size?: number | number[]): Promise<NDArray | number>;
+      normal(loc?: number, scale?: number, size?: number | number[]): Promise<NDArray | number>;
+      integers(low: number, high?: number, size?: number | number[], dtype?: string): Promise<NDArray | number>;
+      choice(a: number | NDArray, size?: number | number[], replace?: boolean): Promise<NDArray | unknown>;
+      shuffle(x: NDArray): Promise<void>;
+      [key: string]: unknown;
+    }
+
+    export function default_rng(seed?: number | null): Promise<Generator>;
+  }
+
+  // ── FFT ────────────────────────────────────────────────────────────────────
+
+  export namespace fft {
+    export function fft(a: NDArray | PyProxy, n?: number, axis?: number): Promise<NDArray>;
+    export function ifft(a: NDArray | PyProxy, n?: number, axis?: number): Promise<NDArray>;
+    export function rfft(a: NDArray | PyProxy, n?: number, axis?: number): Promise<NDArray>;
+    export function irfft(a: NDArray | PyProxy, n?: number, axis?: number): Promise<NDArray>;
+    export function fft2(a: NDArray | PyProxy, s?: [number, number], axes?: [number, number]): Promise<NDArray>;
+    export function ifft2(a: NDArray | PyProxy, s?: [number, number], axes?: [number, number]): Promise<NDArray>;
+    export function fftn(a: NDArray | PyProxy, s?: number[], axes?: number[]): Promise<NDArray>;
+    export function ifftn(a: NDArray | PyProxy, s?: number[], axes?: number[]): Promise<NDArray>;
+    export function fftshift(x: NDArray | PyProxy, axes?: number | number[]): Promise<NDArray>;
+    export function ifftshift(x: NDArray | PyProxy, axes?: number | number[]): Promise<NDArray>;
+    export function fftfreq(n: number, d?: number): Promise<NDArray>;
+    export function rfftfreq(n: number, d?: number): Promise<NDArray>;
+  }
+
+  // ── Constants ─────────────────────────────────────────────────────────────
+
+  export const pi: number;
+  export const e: number;
+  export const inf: number;
+  export const nan: number;
+  export const newaxis: null;
+  export const __version__: string;
+
+  // ── dtype objects ─────────────────────────────────────────────────────────
+
+  export const float32: PyProxy;
+  export const float64: PyProxy;
+  export const int8: PyProxy;
+  export const int16: PyProxy;
+  export const int32: PyProxy;
+  export const int64: PyProxy;
+  export const uint8: PyProxy;
+  export const uint16: PyProxy;
+  export const uint32: PyProxy;
+  export const uint64: PyProxy;
+  export const bool_: PyProxy;
+  export const complex64: PyProxy;
+  export const complex128: PyProxy;
+  export const str_: PyProxy;
+  export const object_: PyProxy;
+
+  // ── Indexing helpers ───────────────────────────────────────────────────────
+
+  export function where(
+    condition: NDArray | PyProxy,
+    x?: NDArray | number | PyProxy,
+    y?: NDArray | number | PyProxy,
+  ): Promise<NDArray | NDArray[]>;
+
+  export function nonzero(a: NDArray | PyProxy): Promise<NDArray[]>;
+  export function argwhere(a: NDArray | PyProxy): Promise<NDArray>;
+  export function flatnonzero(a: NDArray | PyProxy): Promise<NDArray>;
+  export function searchsorted(a: NDArray | PyProxy, v: NDArray | number | PyProxy, side?: 'left' | 'right'): Promise<NDArray | number>;
+  export function extract(condition: NDArray | PyProxy, arr: NDArray | PyProxy): Promise<NDArray>;
+
+  // ── Set operations ────────────────────────────────────────────────────────
+
+  export function union1d(ar1: NDArray | PyProxy, ar2: NDArray | PyProxy): Promise<NDArray>;
+  export function intersect1d(ar1: NDArray | PyProxy, ar2: NDArray | PyProxy, assume_unique?: boolean, return_indices?: boolean): Promise<NDArray | [NDArray, NDArray, NDArray]>;
+  export function setdiff1d(ar1: NDArray | PyProxy, ar2: NDArray | PyProxy): Promise<NDArray>;
+  export function in1d(ar: NDArray | PyProxy, test_elements: NDArray | PyProxy): Promise<NDArray>;
+  export function isin(element: NDArray | PyProxy, test_elements: NDArray | PyProxy): Promise<NDArray>;
+
+  // ── IO ────────────────────────────────────────────────────────────────────
+
+  export function save(file: string, arr: NDArray | PyProxy): Promise<void>;
+  export function savez(file: string, ...arrays: NDArray[]): Promise<void>;
+  export function savez_compressed(file: string, ...arrays: NDArray[]): Promise<void>;
+  export function load(file: string, mmap_mode?: string): Promise<NDArray | PyProxy>;
+  export function savetxt(fname: string, X: NDArray | PyProxy, fmt?: string, delimiter?: string, header?: string): Promise<void>;
+  export function loadtxt(fname: string, dtype?: string, delimiter?: string, skiprows?: number, usecols?: number | number[]): Promise<NDArray>;
+  export function genfromtxt(fname: string, dtype?: string, delimiter?: string, skip_header?: number, names?: string[]): Promise<NDArray>;
+}
+