Generating customized machine learning models to perform tasks using artificial intelligence

The disclosure relates to generating customized machine learning models to perform tasks using artificial intelligence. In one embodiment, a method includes providing a user interface for display to a client system of a user. The user interface may include a first set of options for selecting artificial intelligence (AI) tasks for integration into a user application, a second set of options for selecting one or more devices on which the user wishes to deploy the selected AI task and a third set of options for selecting one or more performance constraints specific to the selected device. A user specification may be received based on a user selection of the first set of options, the second set of options, and the third set of options. A customized AI model may be generated based on the user specification and sent to the client system of the user for integration into the user application. Once integrated, the customized AI model may enable the user application to perform the selected AI task on the selected devi