Hacker News new | ask | show | jobs
Show HN: PyTorch on Java (github.com)
2 points by pdsminer 14 hours ago
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.