[D] What are some well-written ML codebases to refer to get inspiration on good ML software design?

What publicly available ml projects would you refer to as examples of good software design for ML? I’m referring to aspects like how the abstract model/data set/metric classes defined, how easy is it to add a new functionality based on that design, and about overall experience of using them.

For example, I believe scikit-learn is an example of good design. The fit/preditct paradigm is extremely easy to understand even for a newcomer.

Most modern projects seem to be using a …

abstract data data set design easy examples good inspiration machinelearning ml projects projects set software