hhlo
HHLO is a Haskell library and runtime for building, compiling, and executing machine-learning programs targeting StableHLO, the portable intermediate representation of the OpenXLA ecosystem. . Instead of replicating JAX's Python-based tracing infrastructure, HHLO generates StableHLO MLIR text directly from Haskell and compiles it to CPU or GPU via the PJRT plugin interface. . Key features: . * Type-safe EDSL with phantom-shape tensors (e.g. Tensor '[2,3] 'F32) * 50+ ops: matmul, conv2d, softmax, batch-norm, control flow, and more * CPU execution via PJRT CPU plugin * GPU execution via PJRT CUDA plugin with device enumeration * Multi-GPU concurrent inference scaling * No LLVM/MLIR build dependency — emits text and lets PJRT parse it . See the README for setup instructions and examples.
hackage.haskell.org
0.3.0.0
1 day ago
3
6 total
Links
| Registry | hackage.haskell.org |
| Source | Repository |
| JSON API | View JSON |
| CodeMeta | codemeta.json |
Package Details
| PURL |
pkg:hackage/hhlo
spec |
| License | MIT |
| First Release | 4 days ago |
| Last Synced | about 3 hours ago |