You are developing a process for training and running your custom machine learning model in a production environment. Ensuring you can trace the lineage of your model and its predictions is crucial for model management and compliance. What steps should you follow to achieve this?