Show HN: PyTorch on Java

  • Posted 2 hours ago by pdsminer
  • 2 points
https://github.com/haifengl/smile/tree/master/deep
The smile-deep module provides idiomatic Java API for deep learning on the JVM while still reaching CPU, CUDA, and MPS backends by wrapping the PyTorch / LibTorch C++ runtime. It also provides tiktoken BPE tokenizer, LLaMA-3 inference, EfficientNet-V2, and an image classification pipeline out of the box.

0 comments