← all repositories

LaurentMazare/ocaml-torch

OCaml bindings for PyTorch enabling tensor computations with GPU support and automatic differentiation in OCaml.

442 stars OCaml ML Frameworks
ocaml-torch
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

ocaml-torch provides OCaml language bindings for the PyTorch tensor library, bringing NumPy-like tensor computations with GPU acceleration and tape-based automatic differentiation to the OCaml ecosystem. The bindings use the PyTorch C++ API and are mostly automatically generated, currently targeting PyTorch v2.0.0. Developers can install via opam and write ML programs using the torch library in OCaml.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.