aws-cdk-neuronx-patterns
Version:
> [!WARNING] > This library is experimental module.
19 lines (18 loc) • 906 B
TypeScript
import * as batch from "aws-cdk-lib/aws-batch";
import { Construct } from "constructs";
import { INeuronxContainerImage, NeuronxCompiledModel, NeuronxCompilerBase, NeuronxCompilerBaseProps, ComputeEnvironmentResult } from "./neuronx-compiler-base";
/**
* Props of NeuronxNativeCompiler.
*/
export interface NeuronxNativeCompilerProps extends NeuronxCompilerBaseProps {
}
/**
* Neuronx compiler construct.
* Compile the model to work with Inferentia2 and Trainium1 and upload it to an S3 bucket.
*/
export declare class NeuronxNativeCompiler extends NeuronxCompilerBase {
constructor(scope: Construct, id: string, props: NeuronxNativeCompilerProps);
protected createComputeEnvironment(props: NeuronxCompilerBaseProps): ComputeEnvironmentResult;
protected createJobDefinition(props: NeuronxCompilerBaseProps): batch.IJobDefinition;
}
export { INeuronxContainerImage, NeuronxCompiledModel };