"отстаёт" - это не более чем ваше личное ощущение.
На Пайтон есть Х реализации для ML, как и на других языках.
Если вам "понравилась/нужна" Х библиотека на Пайтон, то вы можете её подключить в свой код на Джава/Котлин и вызывать нужные методы в вашей программе.