Если вам действительно нужна оптимальная производительность, вам, вероятно, понадобится выполнять вычисления в собственном коде (что, как я понимаю, и делает NumPy).Мы использовали JAI в предыдущем проекте, и он работал хорошо, но мы не использовали его для чего-то более сложного.Кажется, он поддерживает ядра и свертки , поэтому он может работать в ваших целях.Однако я не верю, что он такой же гибкий, как NumPy.