package rune

  1. Overview
  2. No Docs
Functional transformations for Nx arrays

Install 

dune-project
 Dependency

github.com Edit opam file Versions (4)

Authors

  1. Thibaut Mattio

Maintainers

  1. Thibaut Mattio

Sources

raven-1.0.0.alpha3.tbz
sha256=96d35ce03dfbebd2313657273e24c2e2d20f9e6c7825b8518b69bd1d6ed5870f
sha512=90c5053731d4108f37c19430e45456063e872b04b8a1bbad064c356e1b18e69222de8bfcf4ec14757e71f18164ec6e4630ba770dbcb1291665de5418827d1465

Description

Automatic differentiation and vectorizing maps for Nx tensors. Reverse-mode AD (grad, vjp), forward-mode AD (jvp), vmap, and gradient checking, built on OCaml 5 effect handlers.

Tags

automatic-differentiation machine-learning deep-learning optimization

Published: 17 Mar 2026

Dependencies (4)

  1. nx = version
  2. dune-configurator build
  3. dune >= "3.21"
  4. ocaml >= "5.2.0"

Dev Dependencies (3)

  1. odoc with-doc
  2. mdx with-test
  3. windtrap with-test

Used by (2)

  1. kaun >= "1.0.0~alpha3"
  2. raven >= "1.0.0~alpha3"

Conflicts

None