@shumai/shumai/dist/shumai/tensor/register_gradients.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';
declare type ArgType = Tensor | number | number[] | BigInt64Array | boolean;
export interface GradContext {
    forward_inputs: [Tensor, ...ArgType[]];
    forward_output: Tensor;
    backward_input: Tensor;
    backward_output_index: number;
}
export {};