Reverse-mode algorithmic differentiation using effect handlers in OCaml 5

CommunityNews · 19 November 2022 15:56

effects-examples/algorithmic_differentiation.ml at master · ocaml-multicore/effects-examples.
Examples to illustrate the use of algebraic effects in Multicore OCaml - effects-examples/algorithmic_differentiation.ml at master · ocaml-multicore/effects-examples

Read in full here:

github.com

ocaml-multicore/effects-examples/blob/master/algorithmic_differentiation.ml

(* Reverse-mode Algorithmic differentiation using effect handlers.
   Adapted from https://twitter.com/tiarkrompf/status/963314799521222656.
   See https://openreview.net/forum?id=SJxJtYkPG for more information. *)
open Effect
open Effect.Deep

module F : sig
  type t

  val mk : float -> t
  val ( +. ) : t -> t -> t
  val ( *. ) : t -> t -> t
  val grad : (t -> t) -> float -> float
  val grad2 : (t * t -> t) -> float * float -> float * float
end = struct
  type t = { v : float; mutable d : float }

  let mk v = { v; d = 0.0 }

  type _ Effect.t += Add : t * t -> t Effect.t

This file has been truncated. show original

This thread was posted by one of our members via one of our news source trackers.

bot · 19 November 2022 15:56

Corresponding tweet for this thread:

Share link for this tweet.