flâneur

LaurentMazare/ocaml-torch: OCaml bindings for PyTorch

github.com · 602 words · saved by 1 readers

OCaml bindings for PyTorch

Warning Development for this repo has moved to https://github.com/janestreet/torch . ocaml-torch provides some ocaml bindings for the PyTorch tensor library. This brings to OCaml NumPy-like tensor computations with GPU acceleration and tape-based automatic differentiation. These bindings use the PyTorch C++ API and are mostly automatically generated. The current GitHub tip and the opam package v0.7 corresponds to PyTorch v2.0.0. On Linux note that you will need the PyTorch version using the cxx11 abi cpu version, cuda 11.7 version. Opam Installation The opam package can be installed…

saved by

related reading