Managing artificial intelligence model partitions for edge computing environment

A method, comprising:

• obtaining an intermediate representation of an artificial intelligence model;
• obtaining a computation graph generated based on the intermediate representation;
• partitioning the computation graph into a set of partitions;
• scheduling the set of partitions for respective execution on a set of computing devices in an edge computing environment;
• and
• causing deployment of the set of partitions respectively to the set of computing devices for execution in the edge computing environment;
• wherein the steps are performed by one or more processing devices.

A computer program product stored on a non-transitory computer-readable medium and comprising machine executable instructions, the machine executable instructions, when executed, causing a processing device to:

• obtain an intermediate representation of an artificial intelligence model;
• obtain a computation graph generated based on the intermediate representation;
• partition the computation graph into a set of partitions;
• schedule the set of partitions for respective execution on a set of computing devices in an edge computing environment;
• and
• cause deployment of the set of partitions respectively to the set of computing devices for execution in the edge computing environment.