Is there any relatively small (<2MB with dependencies) multiplatform or at least pure-JVM library that provides me with basic learning+prediction functionality? Like linear regression, gradient descent with like 10 parameters, small data sets. Would be nice if it allowed me to define basic neural networks but that's not necessity.
I found libraries like Smile, DiffKt, KotlinDL, ONNX but they're all huge (100MB+) and require native dependencies. I also found KInference but it can't do training at all so I can't fine tune the model on user machine.
Usecase: to be embedded in minecraft mod to optimize map tiles prefetching, rendering priority and caching.
I have experience in tensorflow and pytorch but they're too big for this project and have native code dependencies.
I don't need anything fancy, just some basic functionalities for simple models, small datasets, a few parameters.