ocaml-torch
Tensor library
Bindings for PyTorch's tensor library in OCaml for GPU acceleration and automatic differentiation
OCaml bindings for PyTorch
413 stars
14 watching
38 forks
Language: OCaml
last commit: about 1 month ago
Linked from 2 awesome lists
artificial-intelligencedeep-learninggpumachine-learningneural-networkocamlpytorchtensor
bharathgs/awesome-pytorch-list
ocaml-community/awesome-ocamlRelated projects:
| Repository | Description | Stars |
|---|---|---|
| laurentmazare/tensorflow-ocaml | OCaml bindings for a popular machine learning framework | 283 |
| laurentmazare/ocaml-bert | Implementing BERT-like NLP models in OCaml using PyTorch bindings and pre-trained weights from popular sources. | 23 |
 torch/cutorch | Provides a CUDA backend for the PyTorch deep learning framework | 336 |
 mmottl/lacaml | An OCaml interface to widely used linear algebra libraries for high-performance numerical computations. | 128 |
 nearai/torchfold | A PyTorch module for dynamic batching and optimized computation on deep neural networks | 221 |
 vermeille/torchelie | A collection of utility functions and tools for building deep learning models with PyTorch | 111 |
 thierry-martinez/pyml | OCaml bindings for Python | 187 |
 wkentaro/pytorch-for-numpy-users | Provides a PyTorch interface for working with NumPy arrays and tensors. | 687 |
 tristandeleu/pytorch-maml-rl | Replication of Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks in PyTorch for reinforcement learning tasks | 827 |
 prabhuomkar/pytorch-cpp | A C++ implementation of PyTorch tutorials | 1,965 |
 torchdr/torchdr | A PyTorch-based library for dimensionality reduction and manifold learning using affinity matrices. | 68 |
 kakaobrain/torchgpipe | A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients. | 817 |
 bloodaxe/pytorch-toolbelt | A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. | 1,520 |
 aiqm/torchani | A PyTorch implementation of a neural network potential for molecular simulations | 464 |
 metaopt/torchopt | An efficient library for differentiable optimization built on top of PyTorch. | 544 |