← All repos

ocaml-torch

OCaml bindings for PyTorch

artificial-intelligencedeep-learninggpumachine-learningneural-networkocamlpytorchtensor
Browse cluster: ROS 2 Rust Middleware & Tooling
544commits
9contributors
8languages

Tech stack & purpose

Ocaml-torch provides OCaml bindings for PyTorch, enabling machine learning and deep learning work in the OCaml language. The project spans multiple problem domains within machine learning, including neural network training, generative models, computer vision tasks, and model inference. The tech stack centers on OCaml as the host language, with bindings to PyTorch for tensor operations and neural network construction, plus integration with related libraries like torch_vision for image processing. Additional tools mentioned include dune (the OCaml build system) and Torch Script for model serialization and cross-language deployment.

Community & reference links

Languages

OCaml
52.2%
C++
24.3%
C
23.2%
Jupyter Notebook
0.2%
Makefile
0.0%
Python
0.0%
Standard ML
0.0%
Dune
0.0%

Contributors