@shumai/shumai/shumai/optim/index.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 type OptimizerFn = (
  grads: Record<
    string,
    {
      grad: Tensor
      tensor: Tensor
    }
  >,
  learning_rate?: number
) => Promise<void> | void
export * from './adam'
export * from './optim'
export * from './sgd'