You would need to access the model via RPC back to the host (the PC running artiq_master). There, you have teh full flexibility of normal Python available, including interfaces to Pytorch, etc. Unless the resulting model is very small, it's probably not a good idea to attempt to run it on the core device (and in any case, you would need to implement the necessary primitives there in ARTIQ Python, etc.).