ocaml-torch
OCaml bindings for PyTorch
AI summary
Tensor library
Bindings for PyTorch's tensor library in OCaml for GPU acceleration and automatic differentiation
- stars
- 416
- forks
- 38
- watching
- 14
- awesome lists
- 2
Similar projects
Found by comparing what the projects do, not just their names.
ML framework binding
OCaml bindings for a popular machine learning framework
NLP transformer model
Implementing BERT-like NLP models in OCaml using PyTorch bindings and pre-trained weights from popular sources.
Deep learning library
Provides a CUDA backend for the PyTorch deep learning framework
Linear algebra library
An OCaml interface to high-performance linear algebra libraries (BLAS/LAPACK) for numerical computations.
Batching library
A PyTorch module for dynamic batching and optimized computation on deep neural networks
PyTorch toolkit
A collection of utility functions and tools for building deep learning models with PyTorch
Python OCaml binder
OCaml bindings for Python
Tensor library
Provides a PyTorch interface for working with NumPy arrays and tensors.
Reinforcement Learning framework
Replication of Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks in PyTorch for reinforcement learning tasks
PyTorch C++ implementation
A C++ implementation of PyTorch tutorials
DR framework
A PyTorch-based framework for dimensionality reduction and manifold learning
Pipeline Parallelism Library
A PyTorch-based library for efficient training of large neural networks using pipeline parallelism and automatic recomputation of gradients.
ML toolkit
A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development.
Neural simulator
A PyTorch implementation of a neural network potential for molecular simulations
Optimization library
An efficient library for differentiable optimization built on top of PyTorch.
Featured in 2 awesome lists
Each link jumps to the spot where the list mentions ocaml-torch.
