@shumai/shumai/shumai/optim/optim.ts

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

export class Optimizer extends Function {
  private __self__: this
  constructor() {
    super('...args', 'return this.__self__.step(...args)')
    const self = this.bind(this)
    this.__self__ = self
    return self
  }

  // eslint-disable-next-line @typescript-eslint/no-unused-vars
  step(grads: Record<string, { grad: sm.Tensor; tensor: sm.Tensor; id: number }>) {
    throw new Error('You must implement a `step()` method in your optimizer')
  }
}