react-native-executorch/src/native/NativeETModule.ts

An easy way to run AI models in react native with ExecuTorch

import type { TurboModule } from 'react-native';
import { TurboModuleRegistry } from 'react-native';

export interface Spec extends TurboModule {
  loadModule(modelSource: string): Promise<number>;
  forward(
    inputs: number[][],
    shapes: number[][],
    inputTypes: number[]
  ): Promise<number[][]>;
  loadMethod(methodName: string): Promise<number>;
}

export default TurboModuleRegistry.get<Spec>('ETModule');