Hacker Newsnew | past | comments | ask | show | jobs | submitlogin
Show HN: PyTorch on Java (github.com/haifengl)
2 points by pdsminer 3 days ago | hide | past | favorite | discuss
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.
 help



Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact

Search: