@shumai/shumai/dist/shumai/optim/index.d.ts

A fast, network-connected, differentiable tensor library for TypeScript (and JavaScript). Built with bun + flashlight for software engineers and researchers alike.

import type { Tensor } from '../tensor';
export declare type OptimizerFn = (grads: Record<string, {
    grad: Tensor;
    tensor: Tensor;
}>, learning_rate?: number) => Promise<void> | void;
export * from './sgd';