Absolutely.
Any iPad is capable of handling of "machine learning." To be more clear, any programmable device is capable of machine learning. The iPad Pro 10.5 is ridiculously fast, faster than many laptops.
Is there something specific you are looking to accomplish?