torch

PyTorch bindings for OCaml
Description

The ocaml-torch project provides some OCaml bindings for the PyTorch library. This brings to OCaml NumPy-like tensor computations with GPU acceleration and tape-based automatic differentiation.

Install
Documentation github.com Apache-2.0 License Edit opam file
Published
08 Nov 2022
Authors
Maintainers
Sources
0.17.tar.gz
md5=c64a4ae1c677f9e2c2528bec6b70e733 sha512=2617754f8e5d127758ce66519683886d660ea9747659d6a3d58d0eb2e299ba9e3da7e5a559dc784bf36cdad477c59394f6ff93657378c572e17596bde52f4537
Dependencies
stdio
sexplib
ppx_sexp_conv
ppx_expect
ppx_custom_printf
ocaml-compiler-libs
ocaml >= "4.08"
npy
libtorch >= "1.13.0" & < "1.14.0"
dune-configurator
dune >= "1.3.0"
ctypes-foreign
ctypes >= "0.11"
cmdliner
base >= "v0.11.0"
Reverse Dependencies