Beyond sequential models: the functional API