import { Tensor } from '@tensorflow/tfjs-core';
import { default as BaseLayer, ForwardAttributes } from './BaseLayer';
import { GPTConfig } from '../main';
export interface RMSNormConfig {
    useGamma?: boolean;
}
export default class RMSNorm extends BaseLayer {
    private GAMMA;
    private rmsConfig;
    constructor(config: GPTConfig, rmsConfig: RMSNormConfig, name?: string, parent?: BaseLayer);
    forward(_: ForwardAttributes, x: Tensor): Tensor;
}
