Organisationsahrefsocannl45cd1e ()debian-12-5.3+flambda_opam-2.4

debian-12-5.3+flambda_opam-2.4

Logs

Show full logs
2025-08-14 09:14.12: New job: test ahrefs/ocannl https://github.com/ahrefs/ocannl.git#refs/heads/master (45cd1e0995de26f6b4d72eb42fe5f7c58b46eaaa) (linux-x86_64:debian-12-5.3+flambda_opam-2.4)
Base: ocaml/opam:debian-12-ocaml-5.3-flambda@sha256:bc87a3a3e7ab52284be6c3db2016fad5b20704dd9d74b10ac786ae3a7af56f68
Opam project build

To reproduce locally:

git clone --recursive "https://github.com/ahrefs/ocannl.git" -b "master" && cd "ocannl" && git reset --hard 45cd1e09
cat > Dockerfile <<'END-OF-DOCKERFILE'
FROM ocaml/opam:debian-12-ocaml-5.3-flambda@sha256:bc87a3a3e7ab52284be6c3db2016fad5b20704dd9d74b10ac786ae3a7af56f68
# debian-12-5.3+flambda_opam-2.4
USER 1000:1000
ENV CLICOLOR_FORCE="1"
ENV OPAMCOLOR="always"
WORKDIR /src
RUN sudo ln -f /usr/bin/opam-2.4 /usr/bin/opam
RUN opam init --reinit -ni
RUN uname -rs && opam exec -- ocaml -version && opam --version
WORKDIR /src
RUN sudo chown opam /src
RUN cd ~/opam-repository && (git cat-file -e e77940d8ff8706305602d0d643bf01c77beef2e7 || git fetch origin master) && git reset -q --hard e77940d8ff8706305602d0d643bf01c77beef2e7 && git log --no-decorate -n1 --oneline && opam update -u
COPY --chown=1000:1000 neural_nets_lib.opam arrayjit.opam ./
RUN opam pin add -yn neural_nets_lib.dev './' && \
    opam pin add -yn arrayjit.dev './'
RUN echo '(lang dune 3.0)' > './dune-project'
ENV DEPS="angstrom.0.16.1 astring.0.8.5 backoff.0.1.1 base.v0.17.3 base-bigarray.base base-bytes.base base-domains.base base-effects.base base-nnp.base base-threads.base base-unix.base bigarray-compat.1.1.0 bigstringaf.0.10.0 camlp-streams.5.0.1 camlzip.1.13 cmdliner.1.3.0 conf-libcurl.2 conf-libffi.2.0.0 conf-pkg-config.4 conf-zlib.1 cppo.1.8.0 csexp.1.5.2 csv.2.4 ctypes.0.23.0 ctypes-foreign.0.23.0 curl.0.10.0 dune.3.19.1 dune-configurator.3.19.1 dune-private-libs.3.19.1 dune-site.3.19.1 dyn.3.19.1 fieldslib.v0.17.0 fmt.0.11.0 integers.0.7.0 jane-street-headers.v0.17.0 jst-config.v0.17.0 logs.0.9.0 mdx.2.5.0 mtime.2.1.0 multicore-magic.2.3.1 num.1.6 ocaml.5.3.0 ocaml-compiler.5.3.0 ocaml-compiler-libs.v0.17.0 ocaml-config.3 ocaml-syntax-shims.1.0.0 ocaml-variants.5.3.0+options ocaml-version.4.0.1 ocaml_intrinsics_kernel.v0.17.1 ocamlbuild.0.16.1 ocamlfind.1.9.8 ordering.3.19.1 parsexp.v0.17.0 pp.2.0.0 pprint.20230830 ppx_assert.v0.17.0 ppx_base.v0.17.0 ppx_cold.v0.17.0 ppx_compare.v0.17.0 ppx_derivers.1.2.1 ppx_deriving.6.1.1 ppx_enumerate.v0.17.0 ppx_expect.v0.17.3 ppx_fields_conv.v0.17.0 ppx_globalize.v0.17.2 ppx_hash.v0.17.0 ppx_here.v0.17.0 ppx_inline_test.v0.17.1 ppx_minidebug.2.3.0 ppx_optcomp.v0.17.1 ppx_sexp_conv.v0.17.1 ppx_string.v0.17.0 ppx_variants_conv.v0.17.1 ppxlib.0.36.0 ppxlib_jane.v0.17.4 printbox.0.12 printbox-ext-plot.0.12 printbox-html.0.12 printbox-md.0.12 printbox-text.0.12 ptime.1.2.0 re.1.13.2 result.1.5 saturn_lockfree.0.5.0 seq.base sexplib.v0.17.0 sexplib0.v0.17.0 stdio.v0.17.0 stdlib-shims.0.3.0 stdune.3.19.1 thread-local-storage.0.2 time_now.v0.17.0 topkg.1.1.0 tyxml.4.6.0 uucp.16.0.0 uutf.1.0.4 variantslib.v0.17.0"
ENV CI="true"
ENV OCAMLCI="true"
RUN opam update --depexts && opam install --cli=2.4 --depext-only -y neural_nets_lib.dev arrayjit.dev $DEPS
RUN opam install $DEPS
COPY --chown=1000:1000 . /src
RUN opam exec -- dune build @install @check @runtest && rm -rf _build

END-OF-DOCKERFILE
docker build .
END-REPRO-BLOCK

2025-08-14 09:14.12: Using cache hint "ahrefs/ocannl-ocaml/opam:debian-12-ocaml-5.3-flambda@sha256:bc87a3a3e7ab52284be6c3db2016fad5b20704dd9d74b10ac786ae3a7af56f68-debian-12-5.3+flambda_opam-2.4-132475af3e6ddd44ad76e37f5d1f298b"
2025-08-14 09:14.12: Using OBuilder spec:
((from ocaml/opam:debian-12-ocaml-5.3-flambda@sha256:bc87a3a3e7ab52284be6c3db2016fad5b20704dd9d74b10ac786ae3a7af56f68)
 (comment debian-12-5.3+flambda_opam-2.4)
 (user (uid 1000) (gid 1000))
 (env CLICOLOR_FORCE 1)
 (env OPAMCOLOR always)
 (workdir /src)
 (run (shell "sudo ln -f /usr/bin/opam-2.4 /usr/bin/opam"))
 (run (shell "opam init --reinit -ni"))
 (run (shell "uname -rs && opam exec -- ocaml -version && opam --version"))
 (workdir /src)
 (run (shell "sudo chown opam /src"))
 (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
      (network host)
      (shell "cd ~/opam-repository && (git cat-file -e e77940d8ff8706305602d0d643bf01c77beef2e7 || git fetch origin master) && git reset -q --hard e77940d8ff8706305602d0d643bf01c77beef2e7 && git log --no-decorate -n1 --oneline && opam update -u"))
 (copy (src neural_nets_lib.opam arrayjit.opam) (dst ./))
 (run (network host)
      (shell  "opam pin add -yn neural_nets_lib.dev './' && \
             \nopam pin add -yn arrayjit.dev './'"))
 (run (network host)
      (shell "echo '(lang dune 3.0)' > './dune-project'"))
 (env DEPS "angstrom.0.16.1 astring.0.8.5 backoff.0.1.1 base.v0.17.3 base-bigarray.base base-bytes.base base-domains.base base-effects.base base-nnp.base base-threads.base base-unix.base bigarray-compat.1.1.0 bigstringaf.0.10.0 camlp-streams.5.0.1 camlzip.1.13 cmdliner.1.3.0 conf-libcurl.2 conf-libffi.2.0.0 conf-pkg-config.4 conf-zlib.1 cppo.1.8.0 csexp.1.5.2 csv.2.4 ctypes.0.23.0 ctypes-foreign.0.23.0 curl.0.10.0 dune.3.19.1 dune-configurator.3.19.1 dune-private-libs.3.19.1 dune-site.3.19.1 dyn.3.19.1 fieldslib.v0.17.0 fmt.0.11.0 integers.0.7.0 jane-street-headers.v0.17.0 jst-config.v0.17.0 logs.0.9.0 mdx.2.5.0 mtime.2.1.0 multicore-magic.2.3.1 num.1.6 ocaml.5.3.0 ocaml-compiler.5.3.0 ocaml-compiler-libs.v0.17.0 ocaml-config.3 ocaml-syntax-shims.1.0.0 ocaml-variants.5.3.0+options ocaml-version.4.0.1 ocaml_intrinsics_kernel.v0.17.1 ocamlbuild.0.16.1 ocamlfind.1.9.8 ordering.3.19.1 parsexp.v0.17.0 pp.2.0.0 pprint.20230830 ppx_assert.v0.17.0 ppx_base.v0.17.0 ppx_cold.v0.17.0 ppx_compare.v0.17.0 ppx_derivers.1.2.1 ppx_deriving.6.1.1 ppx_enumerate.v0.17.0 ppx_expect.v0.17.3 ppx_fields_conv.v0.17.0 ppx_globalize.v0.17.2 ppx_hash.v0.17.0 ppx_here.v0.17.0 ppx_inline_test.v0.17.1 ppx_minidebug.2.3.0 ppx_optcomp.v0.17.1 ppx_sexp_conv.v0.17.1 ppx_string.v0.17.0 ppx_variants_conv.v0.17.1 ppxlib.0.36.0 ppxlib_jane.v0.17.4 printbox.0.12 printbox-ext-plot.0.12 printbox-html.0.12 printbox-md.0.12 printbox-text.0.12 ptime.1.2.0 re.1.13.2 result.1.5 saturn_lockfree.0.5.0 seq.base sexplib.v0.17.0 sexplib0.v0.17.0 stdio.v0.17.0 stdlib-shims.0.3.0 stdune.3.19.1 thread-local-storage.0.2 time_now.v0.17.0 topkg.1.1.0 tyxml.4.6.0 uucp.16.0.0 uutf.1.0.4 variantslib.v0.17.0")
 (env CI true)
 (env OCAMLCI true)
 (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
      (network host)
      (shell "opam update --depexts && opam install --cli=2.4 --depext-only -y neural_nets_lib.dev arrayjit.dev $DEPS"))
 (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
      (network host)
      (shell "opam install $DEPS"))
 (copy (src .) (dst /src))
 (run (shell "opam exec -- dune build @install @check @runtest && rm -rf _build"))
)

2025-08-14 09:14.12: Waiting for resource in pool OCluster
2025-08-14 09:14.12: Waiting for worker…
2025-08-14 09:14.13: Got resource from pool OCluster
Building on bremusa.ocamllabs.io
All commits already cached
HEAD is now at 45cd1e09 Introduce a new IR variant: `Constant_bits` for int64 constants, by Claude Claude found out that the discrepancy between sync_cc and metal backends was due to metal casting a float-single to uint4x32, while pure C was casting a double. This should go away with using Constant_bits / Tensor.bits for Fetch Embed_self_id and Tensor.get_random_seed

(from ocaml/opam:debian-12-ocaml-5.3-flambda@sha256:bc87a3a3e7ab52284be6c3db2016fad5b20704dd9d74b10ac786ae3a7af56f68)
2025-08-14 09:14.15 ---> using "8f19063995d2e91975b673384143fa4316c06cf525acaad5ddc9dfe32e879b21" from cache

/: (comment debian-12-5.3+flambda_opam-2.4)

/: (user (uid 1000) (gid 1000))

/: (env CLICOLOR_FORCE 1)

/: (env OPAMCOLOR always)

/: (workdir /src)

/src: (run (shell "sudo ln -f /usr/bin/opam-2.4 /usr/bin/opam"))
2025-08-14 09:14.15 ---> using "3964d961645aac4701995e7afa81c4230383d448cff22e07a607fd337d9459ea" from cache

/src: (run (shell "opam init --reinit -ni"))
Configuring from /home/opam/.opamrc and then from built-in defaults.
Checking for available remotes: rsync and local, git.
  - you won't be able to use mercurial repositories unless you install the hg command on your system.
  - you won't be able to use darcs repositories unless you install the darcs command on your system.

Continue? [Y/n] y
This version of opam requires an update to the layout of /home/opam/.opam from version 2.0 to version 2.2, which can't be reverted.
You may want to back it up before going further.

Format upgrade done.

<><> Updating repositories ><><><><><><><><><><><><><><><><><><><><><><><><><><>
[default] Initialised
2025-08-14 09:14.15 ---> using "5aebcfda9685f3a603e14fa984661649d5b4c24cad7ac53b2dbd7993b0191857" from cache

/src: (run (shell "uname -rs && opam exec -- ocaml -version && opam --version"))
Linux 6.8.0-63-generic
The OCaml toplevel, version 5.3.0
2.4.1
2025-08-14 09:14.15 ---> using "b3767d8466038a203e474e97e3c72b7e65c6cd3e74da7a70330f79ca1443edbb" from cache

/src: (workdir /src)

/src: (run (shell "sudo chown opam /src"))
2025-08-14 09:14.15 ---> using "3777998cb18ca2c1bd519ed60e51da93fddf8559566beec5a9ce7afd99330774" from cache

/src: (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
           (network host)
           (shell "cd ~/opam-repository && (git cat-file -e e77940d8ff8706305602d0d643bf01c77beef2e7 || git fetch origin master) && git reset -q --hard e77940d8ff8706305602d0d643bf01c77beef2e7 && git log --no-decorate -n1 --oneline && opam update -u"))
e77940d8ff Merge pull request #28255 from dbuenzli/b0-publish-fmt.0.11.0

<><> Updating package repositories ><><><><><><><><><><><><><><><><><><><><><><>
[default] synchronised from git+file:///home/opam/opam-repository

Already up-to-date.
Nothing to do.
# To update the current shell environment, run: eval $(opam env)
2025-08-14 09:14.15 ---> using "63b31d405cf8a512f06ec83086bb2087cf3a01b407c89d327ccfecbb2baa0e5a" from cache

/src: (copy (src neural_nets_lib.opam arrayjit.opam) (dst ./))
2025-08-14 09:14.15 ---> using "2d0faef7c38b3b7928d9e5d425bc401536047ef00db7c3d8993a8a1b358edd4f" from cache

/src: (run (network host)
           (shell  "opam pin add -yn neural_nets_lib.dev './' && \
                  \nopam pin add -yn arrayjit.dev './'"))
[neural_nets_lib.dev] synchronised (file:///src)
neural_nets_lib is now pinned to file:///src (version dev)
[arrayjit.dev] synchronised (file:///src)
arrayjit is now pinned to file:///src (version dev)
2025-08-14 09:14.15 ---> using "6c2ab594ad659ceacf9e4b8bcc4a86c7c471fc620454908adc5cb5b056b4696a" from cache

/src: (run (network host)
           (shell "echo '(lang dune 3.0)' > './dune-project'"))
2025-08-14 09:14.15 ---> using "08254241e4455cfa54ce1c982394b67c3f9e43e8d83d366fa9e9b3d26f40875d" from cache

/src: (env DEPS "angstrom.0.16.1 astring.0.8.5 backoff.0.1.1 base.v0.17.3 base-bigarray.base base-bytes.base base-domains.base base-effects.base base-nnp.base base-threads.base base-unix.base bigarray-compat.1.1.0 bigstringaf.0.10.0 camlp-streams.5.0.1 camlzip.1.13 cmdliner.1.3.0 conf-libcurl.2 conf-libffi.2.0.0 conf-pkg-config.4 conf-zlib.1 cppo.1.8.0 csexp.1.5.2 csv.2.4 ctypes.0.23.0 ctypes-foreign.0.23.0 curl.0.10.0 dune.3.19.1 dune-configurator.3.19.1 dune-private-libs.3.19.1 dune-site.3.19.1 dyn.3.19.1 fieldslib.v0.17.0 fmt.0.11.0 integers.0.7.0 jane-street-headers.v0.17.0 jst-config.v0.17.0 logs.0.9.0 mdx.2.5.0 mtime.2.1.0 multicore-magic.2.3.1 num.1.6 ocaml.5.3.0 ocaml-compiler.5.3.0 ocaml-compiler-libs.v0.17.0 ocaml-config.3 ocaml-syntax-shims.1.0.0 ocaml-variants.5.3.0+options ocaml-version.4.0.1 ocaml_intrinsics_kernel.v0.17.1 ocamlbuild.0.16.1 ocamlfind.1.9.8 ordering.3.19.1 parsexp.v0.17.0 pp.2.0.0 pprint.20230830 ppx_assert.v0.17.0 ppx_base.v0.17.0 ppx_cold.v0.17.0 ppx_compare.v0.17.0 ppx_derivers.1.2.1 ppx_deriving.6.1.1 ppx_enumerate.v0.17.0 ppx_expect.v0.17.3 ppx_fields_conv.v0.17.0 ppx_globalize.v0.17.2 ppx_hash.v0.17.0 ppx_here.v0.17.0 ppx_inline_test.v0.17.1 ppx_minidebug.2.3.0 ppx_optcomp.v0.17.1 ppx_sexp_conv.v0.17.1 ppx_string.v0.17.0 ppx_variants_conv.v0.17.1 ppxlib.0.36.0 ppxlib_jane.v0.17.4 printbox.0.12 printbox-ext-plot.0.12 printbox-html.0.12 printbox-md.0.12 printbox-text.0.12 ptime.1.2.0 re.1.13.2 result.1.5 saturn_lockfree.0.5.0 seq.base sexplib.v0.17.0 sexplib0.v0.17.0 stdio.v0.17.0 stdlib-shims.0.3.0 stdune.3.19.1 thread-local-storage.0.2 time_now.v0.17.0 topkg.1.1.0 tyxml.4.6.0 uucp.16.0.0 uutf.1.0.4 variantslib.v0.17.0")

/src: (env CI true)

/src: (env OCAMLCI true)

/src: (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
           (network host)
           (shell "opam update --depexts && opam install --cli=2.4 --depext-only -y neural_nets_lib.dev arrayjit.dev $DEPS"))
+ /usr/bin/sudo "apt-get" "update"
- Get:1 http://deb.debian.org/debian bookworm InRelease [151 kB]
- Get:2 http://deb.debian.org/debian bookworm-updates InRelease [55.4 kB]
- Get:3 http://deb.debian.org/debian-security bookworm-security InRelease [48.0 kB]
- Get:4 http://deb.debian.org/debian bookworm-updates/main amd64 Packages.diff/Index [21.8 kB]
- Ign:4 http://deb.debian.org/debian bookworm-updates/main amd64 Packages.diff/Index
- Get:5 http://deb.debian.org/debian bookworm-updates/main amd64 Packages [6924 B]
- Get:6 http://deb.debian.org/debian-security bookworm-security/main amd64 Packages [274 kB]
- Fetched 557 kB in 1s (1051 kB/s)
- Reading package lists...
- 

<><> Synchronising pinned packages ><><><><><><><><><><><><><><><><><><><><><><>
[arrayjit.dev] synchronised (file:///src)
[neural_nets_lib.dev] synchronised (file:///src)

[NOTE] Package ocaml-variants is already installed (current version is 5.3.0+options).
[NOTE] Package ocaml-config is already installed (current version is 3).
[NOTE] Package ocaml-compiler is already installed (current version is 5.3.0).
[NOTE] Package ocaml is already installed (current version is 5.3.0).
[NOTE] Package base-unix is already installed (current version is base).
[NOTE] Package base-threads is already installed (current version is base).
[NOTE] Package base-nnp is already installed (current version is base).
[NOTE] Package base-effects is already installed (current version is base).
[NOTE] Package base-domains is already installed (current version is base).
[NOTE] Package base-bigarray is already installed (current version is base).

The following system packages will first need to be installed:
    libcurl4-gnutls-dev libffi-dev pkg-config zlib1g-dev

<><> Handling external dependencies <><><><><><><><><><><><><><><><><><><><><><>

+ /usr/bin/sudo "apt-get" "install" "-qq" "-yy" "libcurl4-gnutls-dev" "libffi-dev" "pkg-config" "zlib1g-dev"
- debconf: delaying package configuration, since apt-utils is not installed
- Selecting previously unselected package libcurl4-gnutls-dev:amd64.
- (Reading database ... 
(Reading database ... 5%
(Reading database ... 10%
(Reading database ... 15%
(Reading database ... 20%
(Reading database ... 25%
(Reading database ... 30%
(Reading database ... 35%
(Reading database ... 40%
(Reading database ... 45%
(Reading database ... 50%
(Reading database ... 55%
(Reading database ... 60%
(Reading database ... 65%
(Reading database ... 70%
(Reading database ... 75%
(Reading database ... 80%
(Reading database ... 85%
(Reading database ... 90%
(Reading database ... 95%
(Reading database ... 100%
(Reading database ... 18778 files and directories currently installed.)
- Preparing to unpack .../0-libcurl4-gnutls-dev_7.88.1-10+deb12u12_amd64.deb ...
- Unpacking libcurl4-gnutls-dev:amd64 (7.88.1-10+deb12u12) ...
- Selecting previously unselected package libffi-dev:amd64.
- Preparing to unpack .../1-libffi-dev_3.4.4-1_amd64.deb ...
- Unpacking libffi-dev:amd64 (3.4.4-1) ...
- Selecting previously unselected package libpkgconf3:amd64.
- Preparing to unpack .../2-libpkgconf3_1.8.1-1_amd64.deb ...
- Unpacking libpkgconf3:amd64 (1.8.1-1) ...
- Selecting previously unselected package pkgconf-bin.
- Preparing to unpack .../3-pkgconf-bin_1.8.1-1_amd64.deb ...
- Unpacking pkgconf-bin (1.8.1-1) ...
- Selecting previously unselected package pkgconf:amd64.
- Preparing to unpack .../4-pkgconf_1.8.1-1_amd64.deb ...
- Unpacking pkgconf:amd64 (1.8.1-1) ...
- Selecting previously unselected package pkg-config:amd64.
- Preparing to unpack .../5-pkg-config_1.8.1-1_amd64.deb ...
- Unpacking pkg-config:amd64 (1.8.1-1) ...
- Selecting previously unselected package zlib1g-dev:amd64.
- Preparing to unpack .../6-zlib1g-dev_1%3a1.2.13.dfsg-1_amd64.deb ...
- Unpacking zlib1g-dev:amd64 (1:1.2.13.dfsg-1) ...
- Setting up libcurl4-gnutls-dev:amd64 (7.88.1-10+deb12u12) ...
- Setting up libffi-dev:amd64 (3.4.4-1) ...
- Setting up libpkgconf3:amd64 (1.8.1-1) ...
- Setting up pkgconf-bin (1.8.1-1) ...
- Setting up zlib1g-dev:amd64 (1:1.2.13.dfsg-1) ...
- Setting up pkgconf:amd64 (1.8.1-1) ...
- Setting up pkg-config:amd64 (1.8.1-1) ...
- Processing triggers for libc-bin (2.36-9+deb12u10) ...
2025-08-14 09:14.15 ---> using "01a0bc33ee314b71f22700bd6686aea6d06a02e7c6dedc62941cf37400eb4125" from cache

/src: (run (cache (opam-archives (target /home/opam/.opam/download-cache)))
           (network host)
           (shell "opam install $DEPS"))
[NOTE] Package ocaml-variants is already installed (current version is 5.3.0+options).
[NOTE] Package ocaml-config is already installed (current version is 3).
[NOTE] Package ocaml-compiler is already installed (current version is 5.3.0).
[NOTE] Package ocaml is already installed (current version is 5.3.0).
[NOTE] Package base-unix is already installed (current version is base).
[NOTE] Package base-threads is already installed (current version is base).
[NOTE] Package base-nnp is already installed (current version is base).
[NOTE] Package base-effects is already installed (current version is base).
[NOTE] Package base-domains is already installed (current version is base).
[NOTE] Package base-bigarray is already installed (current version is base).
The following actions will be performed:
=== install 87 packages
  - install angstrom                0.16.1
  - install astring                 0.8.5
  - install backoff                 0.1.1
  - install base                    v0.17.3
  - install base-bytes              base
  - install bigarray-compat         1.1.0
  - install bigstringaf             0.10.0
  - install camlp-streams           5.0.1
  - install camlzip                 1.13
  - install cmdliner                1.3.0
  - install conf-libcurl            2
  - install conf-libffi             2.0.0
  - install conf-pkg-config         4
  - install conf-zlib               1
  - install cppo                    1.8.0
  - install csexp                   1.5.2
  - install csv                     2.4
  - install ctypes                  0.23.0
  - install ctypes-foreign          0.23.0
  - install curl                    0.10.0
  - install dune                    3.19.1
  - install dune-configurator       3.19.1
  - install dune-private-libs       3.19.1
  - install dune-site               3.19.1
  - install dyn                     3.19.1
  - install fieldslib               v0.17.0
  - install fmt                     0.11.0
  - install integers                0.7.0
  - install jane-street-headers     v0.17.0
  - install jst-config              v0.17.0
  - install logs                    0.9.0
  - install mdx                     2.5.0
  - install mtime                   2.1.0
  - install multicore-magic         2.3.1
  - install num                     1.6
  - install ocaml-compiler-libs     v0.17.0
  - install ocaml-syntax-shims      1.0.0
  - install ocaml-version           4.0.1
  - install ocaml_intrinsics_kernel v0.17.1
  - install ocamlbuild              0.16.1
  - install ocamlfind               1.9.8
  - install ordering                3.19.1
  - install parsexp                 v0.17.0
  - install pp                      2.0.0
  - install pprint                  20230830
  - install ppx_assert              v0.17.0
  - install ppx_base                v0.17.0
  - install ppx_cold                v0.17.0
  - install ppx_compare             v0.17.0
  - install ppx_derivers            1.2.1
  - install ppx_deriving            6.1.1
  - install ppx_enumerate           v0.17.0
  - install ppx_expect              v0.17.3
  - install ppx_fields_conv         v0.17.0
  - install ppx_globalize           v0.17.2
  - install ppx_hash                v0.17.0
  - install ppx_here                v0.17.0
  - install ppx_inline_test         v0.17.1
  - install ppx_minidebug           2.3.0
  - install ppx_optcomp             v0.17.1
  - install ppx_sexp_conv           v0.17.1
  - install ppx_string              v0.17.0
  - install ppx_variants_conv       v0.17.1
  - install ppxlib                  0.36.0
  - install ppxlib_jane             v0.17.4
  - install printbox                0.12
  - install printbox-ext-plot       0.12
  - install printbox-html           0.12
  - install printbox-md             0.12
  - install printbox-text           0.12
  - install ptime                   1.2.0
  - install re                      1.13.2
  - install result                  1.5
  - install saturn_lockfree         0.5.0
  - install seq                     base
  - install sexplib                 v0.17.0
  - install sexplib0                v0.17.0
  - install stdio                   v0.17.0
  - install stdlib-shims            0.3.0
  - install stdune                  3.19.1
  - install thread-local-storage    0.2
  - install time_now                v0.17.0
  - install topkg                   1.1.0
  - install tyxml                   4.6.0
  - install uucp                    16.0.0
  - install uutf                    1.0.4
  - install variantslib             v0.17.0

<><> Processing actions <><><><><><><><><><><><><><><><><><><><><><><><><><><><>
-> retrieved angstrom.0.16.1  (cached)
-> retrieved astring.0.8.5  (cached)
-> retrieved backoff.0.1.1  (cached)
-> retrieved bigarray-compat.1.1.0  (cached)
-> retrieved bigstringaf.0.10.0  (cached)
-> retrieved base.v0.17.3  (cached)
-> retrieved camlp-streams.5.0.1  (cached)
-> retrieved camlzip.1.13  (cached)
-> installed conf-libcurl.2
-> installed conf-pkg-config.4
-> retrieved cmdliner.1.3.0  (cached)
-> retrieved cppo.1.8.0  (cached)
-> retrieved csexp.1.5.2  (cached)
-> installed conf-libffi.2.0.0
-> installed conf-zlib.1
-> retrieved csv.2.4  (cached)
-> retrieved ctypes.0.23.0, ctypes-foreign.0.23.0  (cached)
-> retrieved curl.0.10.0  (cached)
-> retrieved fieldslib.v0.17.0  (cached)
-> retrieved fmt.0.11.0  (cached)
-> retrieved integers.0.7.0  (cached)
-> retrieved jane-street-headers.v0.17.0  (cached)
-> retrieved jst-config.v0.17.0  (cached)
-> retrieved logs.0.9.0  (cached)
-> retrieved mtime.2.1.0  (cached)
-> retrieved multicore-magic.2.3.1  (cached)
-> retrieved mdx.2.5.0  (cached)
-> retrieved num.1.6  (cached)
-> retrieved ocaml-compiler-libs.v0.17.0  (cached)
-> retrieved ocaml-syntax-shims.1.0.0  (cached)
-> retrieved ocaml-version.4.0.1  (cached)
-> retrieved ocaml_intrinsics_kernel.v0.17.1  (cached)
-> retrieved ocamlbuild.0.16.1  (cached)
-> retrieved ocamlfind.1.9.8  (cached)
-> retrieved parsexp.v0.17.0  (cached)
-> retrieved pp.2.0.0  (cached)
-> retrieved pprint.20230830  (cached)
-> retrieved ppx_assert.v0.17.0  (cached)
-> retrieved ppx_base.v0.17.0  (cached)
-> retrieved ppx_cold.v0.17.0  (cached)
-> retrieved ppx_compare.v0.17.0  (cached)
-> retrieved ppx_derivers.1.2.1  (cached)
-> retrieved dune.3.19.1, dune-configurator.3.19.1, dune-private-libs.3.19.1, dune-site.3.19.1, dyn.3.19.1, ordering.3.19.1, stdune.3.19.1  (cached)
-> retrieved ppx_deriving.6.1.1  (cached)
-> retrieved ppx_enumerate.v0.17.0  (cached)
-> retrieved ppx_expect.v0.17.3  (cached)
-> retrieved ppx_fields_conv.v0.17.0  (cached)
-> retrieved ppx_globalize.v0.17.2  (cached)
-> retrieved ppx_hash.v0.17.0  (cached)
-> retrieved ppx_here.v0.17.0  (cached)
-> retrieved ppx_inline_test.v0.17.1  (cached)
-> retrieved ppx_optcomp.v0.17.1  (cached)
-> installed cmdliner.1.3.0
-> installed num.1.6
-> retrieved ppx_minidebug.2.3.0  (cached)
-> retrieved ppx_sexp_conv.v0.17.1  (cached)
-> retrieved ppx_string.v0.17.0  (cached)
-> retrieved ppx_variants_conv.v0.17.1  (cached)
-> retrieved ppxlib_jane.v0.17.4  (cached)
-> retrieved ptime.1.2.0  (cached)
-> retrieved ppxlib.0.36.0  (cached)
-> retrieved re.1.13.2  (cached)
-> retrieved result.1.5  (cached)
-> retrieved seq.base  (cached)
-> installed seq.base
-> retrieved sexplib.v0.17.0  (cached)
-> retrieved sexplib0.v0.17.0  (cached)
-> retrieved saturn_lockfree.0.5.0  (cached)
-> retrieved printbox.0.12, printbox-ext-plot.0.12, printbox-html.0.12, printbox-md.0.12, printbox-text.0.12  (cached)
-> retrieved stdio.v0.17.0  (cached)
-> retrieved stdlib-shims.0.3.0  (cached)
-> retrieved thread-local-storage.0.2  (cached)
-> retrieved time_now.v0.17.0  (cached)
-> retrieved topkg.1.1.0  (cached)
-> retrieved tyxml.4.6.0  (cached)
-> retrieved uutf.1.0.4  (cached)
-> retrieved variantslib.v0.17.0  (cached)
-> retrieved uucp.16.0.0  (cached)
-> installed ocamlfind.1.9.8
-> installed base-bytes.base
-> installed camlzip.1.13
-> installed ocamlbuild.0.16.1
-> installed topkg.1.1.0
-> installed mtime.2.1.0
-> installed uutf.1.0.4
-> installed fmt.0.11.0
-> installed ptime.1.2.0
-> installed astring.0.8.5
-> installed logs.0.9.0
-> installed dune.3.19.1
-> installed backoff.0.1.1
-> installed bigarray-compat.1.1.0
-> installed camlp-streams.5.0.1
-> installed csexp.1.5.2
-> installed cppo.1.8.0
-> installed csv.2.4
-> installed jane-street-headers.v0.17.0
-> installed multicore-magic.2.3.1
-> installed ocaml-compiler-libs.v0.17.0
-> installed ocaml-syntax-shims.1.0.0
-> installed ocaml-version.4.0.1
-> installed ocaml_intrinsics_kernel.v0.17.1
-> installed pp.2.0.0
-> installed pprint.20230830
-> installed ppx_derivers.1.2.1
-> installed printbox.0.12
-> installed re.1.13.2
-> installed result.1.5
-> installed sexplib0.v0.17.0
-> installed stdlib-shims.0.3.0
-> installed thread-local-storage.0.2
-> installed saturn_lockfree.0.5.0
-> installed integers.0.7.0
-> installed ordering.3.19.1
-> installed parsexp.v0.17.0
-> installed dune-configurator.3.19.1
-> installed dyn.3.19.1
-> installed sexplib.v0.17.0
-> installed bigstringaf.0.10.0
-> installed mdx.2.5.0
-> installed angstrom.0.16.1
-> installed curl.0.10.0
-> installed tyxml.4.6.0
-> installed printbox-html.0.12
-> installed stdune.3.19.1
-> installed uucp.16.0.0
-> installed ctypes.0.23.0
-> installed printbox-text.0.12
-> installed printbox-md.0.12
-> installed base.v0.17.3
-> installed printbox-ext-plot.0.12
-> installed dune-private-libs.3.19.1
-> installed fieldslib.v0.17.0
-> installed stdio.v0.17.0
-> installed variantslib.v0.17.0
-> installed ctypes-foreign.0.23.0
-> installed dune-site.3.19.1
-> installed ppxlib.0.36.0
-> installed ppx_optcomp.v0.17.1
-> installed ppxlib_jane.v0.17.4
-> installed ppx_cold.v0.17.0
-> installed ppx_here.v0.17.0
-> installed ppx_variants_conv.v0.17.1
-> installed ppx_fields_conv.v0.17.0
-> installed ppx_globalize.v0.17.2
-> installed ppx_enumerate.v0.17.0
-> installed ppx_deriving.6.1.1
-> installed ppx_compare.v0.17.0
-> installed ppx_sexp_conv.v0.17.1
-> installed ppx_hash.v0.17.0
-> installed ppx_assert.v0.17.0
-> installed ppx_minidebug.2.3.0
-> installed ppx_base.v0.17.0
-> installed jst-config.v0.17.0
-> installed ppx_string.v0.17.0
-> installed time_now.v0.17.0
-> installed ppx_inline_test.v0.17.1
-> installed ppx_expect.v0.17.3
Done.
# To update the current shell environment, run: eval $(opam env)
2025-08-14 09:14.15 ---> using "59f43b81e48adac6abe6cd91a0c36b11352f9254e0345efe4f665e4f4d8f8556" from cache

/src: (copy (src .) (dst /src))
2025-08-14 09:14.15 ---> saved as "4685f8c941d98ff8076bca090df9cc988f4cabfb2bf9a057f0ee5067911e4068"

/src: (run (shell "opam exec -- dune build @install @check @runtest && rm -rf _build"))
File "arrayjit/test/dune", line 6, characters 16-24:
6 |   (pps ppx_here ppx_jane)))
                    ^^^^^^^^
Error: Library "ppx_jane" not found.
-> required by
   _build/default/arrayjit/test/.merlin-conf/exe-test_numerical_types
-> required by alias arrayjit/test/check
(cd _build/default/test/operations/config && ../../../arrayjit/bin/read_config.exe --read=backend)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Wrote value of 'backend' to ocannl_backend.txt
(cd _build/default/test/ppx && ./test_ppx_op.exe)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/default/test/operations && ./threefry4x32_demo.exe --ocannl_output_prec_in_ll_files=true --ocannl_output_debug_files_in_build_directory=true)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Generated 48 uniform random numbers:
  [0]: 0.205322
  [1]: 0.188110
  [2]: 0.220215
  [3]: 0.073975
  [4]: 0.374023
  [5]: 0.494141
  [6]: 0.410400
  [7]: 0.102295
  [8]: 0.160400
  [9]: 0.635254
  [10]: 0.657715
  [11]: 0.580566
  [12]: 0.698730
  [13]: 0.979492
  [14]: 0.033386
  [15]: 0.711426
  [16]: 0.600098
  [17]: 0.772461
  [18]: 0.133789
  [19]: 0.799316
  [20]: 0.664062
  [21]: 0.045898
  [22]: 0.072632
  [23]: 0.610352
  [24]: 0.855469
  [25]: 0.286377
  [26]: 0.113281
  [27]: 0.690430
  [28]: 0.365479
  [29]: 0.236328
  [30]: 0.400146
  [31]: 0.354248
  [32]: 0.851562
  [33]: 0.613281
  [34]: 0.859863
  [35]: 0.624023
  [36]: 0.302246
  [37]: 0.005325
  [38]: 0.874023
  [39]: 0.461914
  [40]: 0.002930
  [41]: 0.811035
  [42]: 0.319824
  [43]: 0.088684
  [44]: 0.597656
  [45]: 0.608887
  [46]: 0.076538
  [47]: 0.784668
(cd _build/default/test/operations && ./test_threefry_precision.exe)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Threefry4x32 result precision: uint4x32
Uniform result precision: single
Retrieving commandline, environment, or config file variable ocannl_backend
Found sync_cc, in the config file
Retrieving commandline, environment, or config file variable ocannl_cd_ident_style
Not found, using default heuristic
Retrieving commandline, environment, or config file variable ocannl_ll_ident_style
Not found, using default heuristic
Retrieving commandline, environment, or config file variable ocannl_output_prec_in_ll_files
Not found, using default false
Retrieving commandline, environment, or config file variable ocannl_inline_complex_computations
Not found, using default true
Retrieving commandline, environment, or config file variable ocannl_prefer_backend_uniformity
Found true, in the config file
Retrieving commandline, environment, or config file variable ocannl_debug_log_to_stream_files
Not found, using default false
Retrieving commandline, environment, or config file variable ocannl_stack_threshold_in_bytes
Not found, using default 16384
Retrieving commandline, environment, or config file variable ocannl_output_dlls_in_build_directory
Not found, using default false
Retrieving commandline, environment, or config file variable ocannl_cc_backend_optimization_level
Not found, using default 3
Retrieving commandline, environment, or config file variable ocannl_cc_backend_compiler_command
Not found, using default gcc
Retrieving commandline, environment, or config file variable ocannl_cc_backend_verify_codesign
Not found, using default false
Compilation successful!
Actual tensor precision in context: uint4x32
(cd _build/default/test/operations && ./top_down_prec.exe --ocannl_output_prec_in_ll_files=true --ocannl_output_debug_files_in_build_directory=true)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
                      #8 *._d
                       8.00
                      #9 grad_*._d Virt/30
                      <void>
             #6 + Virt/152                 │#4 c non-emb
             <void>                        │ 2.00
             #7 grad_+ Virt/30             │#5 grad_c Local/26030
             <void>                        │<void>
#0 a non-emb         │#2 b non-emb         │
 2.00                │ 2.00                │
#1 grad_a Local/26030│#3 grad_b Local/26030│
<void>               │<void>               │
(cd _build/default/test/ppx && ./test_ppx_op_expected.exe)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations && .operations_tutorials.inline-tests/inline-test-runner.exe inline-test-runner operations_tutorials -partition 'Found 0, in the config file' -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations && .operations_tutorials.inline-tests/inline-test-runner.exe inline-test-runner operations_tutorials -partition 'Retrieving commandline, environment, or config file variable ocannl_log_level' -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/6961c661d4f43a4e96007f52b730c822/default/test/einsum && .einsum_tutorials.inline-tests/inline-test-runner.exe inline-test-runner einsum_tutorials -partition 'Retrieving commandline, environment, or config file variable ocannl_log_level' -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/6961c661d4f43a4e96007f52b730c822/default/test/einsum && .einsum_tutorials.inline-tests/inline-test-runner.exe inline-test-runner einsum_tutorials -partition 'Found 0, in the config file' -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations && .operations_tutorials.inline-tests/inline-test-runner.exe inline-test-runner operations_tutorials -partition test_threefry4x32.ml -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/default/test/operations && ./micrograd_demo_logging.exe --ocannl_debug_backend=text --ocannl_log_file_stem=micrograd_demo_logging --ocannl_log_main_domain_to_stdout=false --ocannl_debug_log_to_stream_files=true)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
Retrieving commandline, environment, or config file variable ocannl_backend
Found sync_cc, in the config file
Retrieving commandline, environment, or config file variable ocannl_never_capture_stdout
Not found, using default false
Retrieving commandline, environment, or config file variable ocannl_cd_ident_style
Not found, using default heuristic
Retrieving commandline, environment, or config file variable ocannl_ll_ident_style
Not found, using default heuristic
Retrieving commandline, environment, or config file variable ocannl_output_prec_in_ll_files
Not found, using default false
Retrieving commandline, environment, or config file variable ocannl_inline_complex_computations
Not found, using default true
Retrieving commandline, environment, or config file variable ocannl_prefer_backend_uniformity
Found true, in the config file
Retrieving commandline, environment, or config file variable ocannl_debug_log_to_stream_files
Found true, commandline --ocannl_debug_log_to_stream_files=true
Retrieving commandline, environment, or config file variable ocannl_stack_threshold_in_bytes
Not found, using default 16384
Retrieving commandline, environment, or config file variable ocannl_output_dlls_in_build_directory
Not found, using default false
Retrieving commandline, environment, or config file variable ocannl_cc_backend_optimization_level
Not found, using default 3
Retrieving commandline, environment, or config file variable ocannl_cc_backend_compiler_command
Not found, using default gcc
Retrieving commandline, environment, or config file variable ocannl_cc_backend_verify_codesign
Not found, using default false
HERE: test/operations/micrograd_demo_logging.ml:35:21
┌────────────────────┐
│[70]: +_g shape 0:1 │
│┌┬─────────┐        │
│││axis 0   │        │
│├┼─────────┤        │
│││ 2.47e+1 │        │
│└┴─────────┘        │
└────────────────────┘
HERE: test/operations/micrograd_demo_logging.ml:36:21
┌─────────────────┐
│[0]: a shape 0:1 │
│┌┬───────┐       │
│││axis 0 │       │
│├┼───────┤       │
│││ -4.00 │       │
│└┴───────┘       │
└─────────────────┘
┌────────────────────────┐
│[0]: a shape 0:1  grad_a│
│┌┬─────────┐            │
│││axis 0   │            │
│├┼─────────┤            │
│││ 1.38e+2 │            │
│└┴─────────┘            │
└────────────────────────┘
HERE: test/operations/micrograd_demo_logging.ml:37:21
┌─────────────────┐
│[2]: b shape 0:1 │
│┌┬──────┐        │
│││axis 0│        │
│├┼──────┤        │
│││ 2.00 │        │
│└┴──────┘        │
└─────────────────┘
┌────────────────────────┐
│[2]: b shape 0:1  grad_b│
│┌┬─────────┐            │
│││axis 0   │            │
│├┼─────────┤            │
│││ 6.45e+2 │            │
│└┴─────────┘            │
└────────────────────────┘
File "test/einsum/moons_demo_variant.expected", line 1, characters 0-0:
/usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum/moons_demo_variant.expected _build/default/test/einsum/moons_demo_variant.exe.output
diff --git a/_build/default/test/einsum/moons_demo_variant.expected b/_build/default/test/einsum/moons_demo_variant.exe.output
index dec6b08..0d56a97 100644
--- a/_build/default/test/einsum/moons_demo_variant.expected
+++ b/_build/default/test/einsum/moons_demo_variant.exe.output
@@ -42,6 +42,7 @@ n41 grad_relu_margin_loss as relu_margin_loss.grad: Virt/151; single prec 10x1;
 n42 10 as _10: Virt/40; single prec 1; mem in bytes: <not-in-yet>
 n43 => as n43: Local/9046; single prec 1; mem in bytes: <not-in-yet>
 n44 grad_=> as n43.grad: Virt/40; single prec 1; mem in bytes: <not-in-yet>
+n45 /._scalar_loss as scalar_loss: Host&stream/412410; single prec 1; mem in bytes: <not-in-yet>
 n46 grad_/._scalar_loss as scalar_loss.grad: Virt/40; single prec 1; mem in bytes: <not-in-yet>
 n47 1 as _1: Virt/40; single prec 1; mem in bytes: <not-in-yet>
 n48 200 as _200: Virt/40; single prec 1; mem in bytes: <not-in-yet>
@@ -52,6 +53,7 @@ n52 *. as n52: Virt/40; single prec 1; mem in bytes: <not-in-yet>
 n53 - as n53: Virt/152; single prec 1; mem in bytes: <not-in-yet>
 n54 0.1 as n54: Virt/40; single prec 1; mem in bytes: <not-in-yet>
 n55 *. as n55: Virt/152; single prec 1; mem in bytes: <not-in-yet>
+n56 /._learning_rate as learning_rate: Host&stream/412410; single prec 1; mem in bytes: <not-in-yet>
 n57 sgd_delta_b1 as sgd_delta_b1: Virt/15; single prec 16; mem in bytes: <not-in-yet>
 n58 sgd_momentum_b1 as sgd_momentum_b1: unknown; single prec <not-in-yet>; mem in bytes: <not-in-yet>
 n59 0.0001 as n59: Virt/40; single prec 1; mem in bytes: <not-in-yet>
(cd _build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations && .operations_tutorials.inline-tests/inline-test-runner.exe inline-test-runner operations_tutorials -partition zero2hero_1of7.ml -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
(cd _build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations && .operations_tutorials.inline-tests/inline-test-runner.exe inline-test-runner operations_tutorials -partition primitive_ops.ml -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
File "test/einsum/einsum_trivia_exec.expected", line 1, characters 0-0:
/usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum/einsum_trivia_exec.expected _build/default/test/einsum/einsum_trivia_exec.exe.output
diff --git a/_build/default/test/einsum/einsum_trivia_exec.expected b/_build/default/test/einsum/einsum_trivia_exec.exe.output
index 7b623db..dac7ec4 100644
--- a/_build/default/test/einsum/einsum_trivia_exec.expected
+++ b/_build/default/test/einsum/einsum_trivia_exec.exe.output
@@ -657,159 +657,159 @@ HERE: test/einsum/einsum_trivia_exec.ml:229:21
 │└──────┴────────────────────────────────────┘│
 └─────────────────────────────────────────────┘
 HERE: test/einsum/einsum_trivia_exec.ml:254:21
-┌────────────────────────────────┐
-│[9]: ;=>+ shape 0:4,1:5,2:6,3:3 │
-│┌──────┬──────────────────┐     │
-││0 @ 0 │axis 3            │     │
-│├──────┼──────────────────┤     │
-││0 @ 1 │ 0.00  0.00  0.00 │     │
-││axis 2│ 0.00  0.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 0.00  0.00  4.00 │     │
-││      │ 0.00  0.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││1 @ 1 │ 0.00  1.00  0.00 │     │
-││axis 2│ 0.00  1.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 0.00  1.00  4.00 │     │
-││      │ 0.00  1.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││2 @ 1 │ 0.00  2.00  0.00 │     │
-││axis 2│ 0.00  2.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 0.00  2.00  4.00 │     │
-││      │ 0.00  2.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││3 @ 1 │ 0.00  3.00  0.00 │     │
-││axis 2│ 0.00  3.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 0.00  3.00  4.00 │     │
-││      │ 0.00  3.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││4 @ 1 │ 0.00  4.00  0.00 │     │
-││axis 2│ 0.00  4.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 0.00  4.00  4.00 │     │
-││      │ 0.00  4.00  5.00 │     │
-│└──────┴──────────────────┘     │
-├────────────────────────────────┤
-│┌──────┬──────────────────┐     │
-││1 @ 0 │axis 3            │     │
-│├──────┼──────────────────┤     │
-││0 @ 1 │ 1.00  0.00  0.00 │     │
-││axis 2│ 1.00  0.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 1.00  0.00  4.00 │     │
-││      │ 1.00  0.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││1 @ 1 │ 1.00  1.00  0.00 │     │
-││axis 2│ 1.00  1.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 1.00  1.00  4.00 │     │
-││      │ 1.00  1.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││2 @ 1 │ 1.00  2.00  0.00 │     │
-││axis 2│ 1.00  2.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 1.00  2.00  4.00 │     │
-││      │ 1.00  2.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││3 @ 1 │ 1.00  3.00  0.00 │     │
-││axis 2│ 1.00  3.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 1.00  3.00  4.00 │     │
-││      │ 1.00  3.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││4 @ 1 │ 1.00  4.00  0.00 │     │
-││axis 2│ 1.00  4.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 1.00  4.00  4.00 │     │
-││      │ 1.00  4.00  5.00 │     │
-│└──────┴──────────────────┘     │
-├────────────────────────────────┤
-│┌──────┬──────────────────┐     │
-││2 @ 0 │axis 3            │     │
-│├──────┼──────────────────┤     │
-││0 @ 1 │ 2.00  0.00  0.00 │     │
-││axis 2│ 2.00  0.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 2.00  0.00  4.00 │     │
-││      │ 2.00  0.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││1 @ 1 │ 2.00  1.00  0.00 │     │
-││axis 2│ 2.00  1.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 2.00  1.00  4.00 │     │
-││      │ 2.00  1.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││2 @ 1 │ 2.00  2.00  0.00 │     │
-││axis 2│ 2.00  2.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 2.00  2.00  4.00 │     │
-││      │ 2.00  2.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││3 @ 1 │ 2.00  3.00  0.00 │     │
-││axis 2│ 2.00  3.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 2.00  3.00  4.00 │     │
-││      │ 2.00  3.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││4 @ 1 │ 2.00  4.00  0.00 │     │
-││axis 2│ 2.00  4.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 2.00  4.00  4.00 │     │
-││      │ 2.00  4.00  5.00 │     │
-│└──────┴──────────────────┘     │
-├────────────────────────────────┤
-│┌──────┬──────────────────┐     │
-││3 @ 0 │axis 3            │     │
-│├──────┼──────────────────┤     │
-││0 @ 1 │ 3.00  0.00  0.00 │     │
-││axis 2│ 3.00  0.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 3.00  0.00  4.00 │     │
-││      │ 3.00  0.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││1 @ 1 │ 3.00  1.00  0.00 │     │
-││axis 2│ 3.00  1.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 3.00  1.00  4.00 │     │
-││      │ 3.00  1.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││2 @ 1 │ 3.00  2.00  0.00 │     │
-││axis 2│ 3.00  2.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 3.00  2.00  4.00 │     │
-││      │ 3.00  2.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││3 @ 1 │ 3.00  3.00  0.00 │     │
-││axis 2│ 3.00  3.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 3.00  3.00  4.00 │     │
-││      │ 3.00  3.00  5.00 │     │
-│├──────┼──────────────────┤     │
-││4 @ 1 │ 3.00  4.00  0.00 │     │
-││axis 2│ 3.00  4.00  1.00 │     │
-││      │ ...   ...   ...  │     │
-││      │ 3.00  4.00  4.00 │     │
-││      │ 3.00  4.00  5.00 │     │
-│└──────┴──────────────────┘     │
-└────────────────────────────────┘
+┌───────────────────────────────────┐
+│[9]: ;=>+ shape 0:4,1:5,2:6,3:3    │
+│┌──────┬──────────────────────────┐│
+││0 @ 0 │axis 3                    ││
+│├──────┼──────────────────────────┤│
+││0 @ 1 │ 5.68e+29  7.00e-45  0.00 ││
+││axis 2│ 0.00      -5.01e-3  1.00 ││
+││      │ ...       ...       ...  ││
+││      │ -5.01e-3  4.08e-41  4.00 ││
+││      │ 4.08e-41  -5.01e-3  5.00 ││
+│├──────┼──────────────────────────┤│
+││1 @ 1 │ 5.68e+29  1.00     0.00  ││
+││axis 2│ 0.00      9.94e-1  1.00  ││
+││      │ ...       ...      ...   ││
+││      │ -5.01e-3  1.00     4.00  ││
+││      │ 4.08e-41  9.94e-1  5.00  ││
+│├──────┼──────────────────────────┤│
+││2 @ 1 │ 5.68e+29  2.00  0.00     ││
+││axis 2│ 0.00      1.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ -5.01e-3  2.00  4.00     ││
+││      │ 4.08e-41  1.99  5.00     ││
+│├──────┼──────────────────────────┤│
+││3 @ 1 │ 5.68e+29  3.00  0.00     ││
+││axis 2│ 0.00      2.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ -5.01e-3  3.00  4.00     ││
+││      │ 4.08e-41  2.99  5.00     ││
+│├──────┼──────────────────────────┤│
+││4 @ 1 │ 5.68e+29  4.00  0.00     ││
+││axis 2│ 0.00      3.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ -5.01e-3  4.00  4.00     ││
+││      │ 4.08e-41  3.99  5.00     ││
+│└──────┴──────────────────────────┘│
+├───────────────────────────────────┤
+│┌──────┬──────────────────────────┐│
+││1 @ 0 │axis 3                    ││
+│├──────┼──────────────────────────┤│
+││0 @ 1 │ 5.68e+29  7.00e-45  0.00 ││
+││axis 2│ 1.00      -5.01e-3  1.00 ││
+││      │ ...       ...       ...  ││
+││      │ 9.94e-1   4.08e-41  4.00 ││
+││      │ 1.00      -5.01e-3  5.00 ││
+│├──────┼──────────────────────────┤│
+││1 @ 1 │ 5.68e+29  1.00     0.00  ││
+││axis 2│ 1.00      9.94e-1  1.00  ││
+││      │ ...       ...      ...   ││
+││      │ 9.94e-1   1.00     4.00  ││
+││      │ 1.00      9.94e-1  5.00  ││
+│├──────┼──────────────────────────┤│
+││2 @ 1 │ 5.68e+29  2.00  0.00     ││
+││axis 2│ 1.00      1.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 9.94e-1   2.00  4.00     ││
+││      │ 1.00      1.99  5.00     ││
+│├──────┼──────────────────────────┤│
+││3 @ 1 │ 5.68e+29  3.00  0.00     ││
+││axis 2│ 1.00      2.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 9.94e-1   3.00  4.00     ││
+││      │ 1.00      2.99  5.00     ││
+│├──────┼──────────────────────────┤│
+││4 @ 1 │ 5.68e+29  4.00  0.00     ││
+││axis 2│ 1.00      3.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 9.94e-1   4.00  4.00     ││
+││      │ 1.00      3.99  5.00     ││
+│└──────┴──────────────────────────┘│
+├───────────────────────────────────┤
+│┌──────┬──────────────────────────┐│
+││2 @ 0 │axis 3                    ││
+│├──────┼──────────────────────────┤│
+││0 @ 1 │ 5.68e+29  7.00e-45  0.00 ││
+││axis 2│ 2.00      -5.01e-3  1.00 ││
+││      │ ...       ...       ...  ││
+││      │ 1.99      4.08e-41  4.00 ││
+││      │ 2.00      -5.01e-3  5.00 ││
+│├──────┼──────────────────────────┤│
+││1 @ 1 │ 5.68e+29  1.00     0.00  ││
+││axis 2│ 2.00      9.94e-1  1.00  ││
+││      │ ...       ...      ...   ││
+││      │ 1.99      1.00     4.00  ││
+││      │ 2.00      9.94e-1  5.00  ││
+│├──────┼──────────────────────────┤│
+││2 @ 1 │ 5.68e+29  2.00  0.00     ││
+││axis 2│ 2.00      1.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 1.99      2.00  4.00     ││
+││      │ 2.00      1.99  5.00     ││
+│├──────┼──────────────────────────┤│
+││3 @ 1 │ 5.68e+29  3.00  0.00     ││
+││axis 2│ 2.00      2.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 1.99      3.00  4.00     ││
+││      │ 2.00      2.99  5.00     ││
+│├──────┼──────────────────────────┤│
+││4 @ 1 │ 5.68e+29  4.00  0.00     ││
+││axis 2│ 2.00      3.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 1.99      4.00  4.00     ││
+││      │ 2.00      3.99  5.00     ││
+│└──────┴──────────────────────────┘│
+├───────────────────────────────────┤
+│┌──────┬──────────────────────────┐│
+││3 @ 0 │axis 3                    ││
+│├──────┼──────────────────────────┤│
+││0 @ 1 │ 5.68e+29  7.00e-45  0.00 ││
+││axis 2│ 3.00      -5.01e-3  1.00 ││
+││      │ ...       ...       ...  ││
+││      │ 2.99      4.08e-41  4.00 ││
+││      │ 3.00      -5.01e-3  5.00 ││
+│├──────┼──────────────────────────┤│
+││1 @ 1 │ 5.68e+29  1.00     0.00  ││
+││axis 2│ 3.00      9.94e-1  1.00  ││
+││      │ ...       ...      ...   ││
+││      │ 2.99      1.00     4.00  ││
+││      │ 3.00      9.94e-1  5.00  ││
+│├──────┼──────────────────────────┤│
+││2 @ 1 │ 5.68e+29  2.00  0.00     ││
+││axis 2│ 3.00      1.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 2.99      2.00  4.00     ││
+││      │ 3.00      1.99  5.00     ││
+│├──────┼──────────────────────────┤│
+││3 @ 1 │ 5.68e+29  3.00  0.00     ││
+││axis 2│ 3.00      2.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 2.99      3.00  4.00     ││
+││      │ 3.00      2.99  5.00     ││
+│├──────┼──────────────────────────┤│
+││4 @ 1 │ 5.68e+29  4.00  0.00     ││
+││axis 2│ 3.00      3.99  1.00     ││
+││      │ ...       ...   ...      ││
+││      │ 2.99      4.00  4.00     ││
+││      │ 3.00      3.99  5.00     ││
+│└──────┴──────────────────────────┘│
+└───────────────────────────────────┘
 HERE: test/einsum/einsum_trivia_exec.ml:255:21
 [1]: =>_ti shape 0:4,1:3  <not-hosted>
 HERE: test/einsum/einsum_trivia_exec.ml:256:21
-┌───────────────────────────┐
-│[7]: =>_tk shape 0:6,1:3   │
-│┌──────┬──────────────────┐│
-││      │axis 1            ││
-│├──────┼──────────────────┤│
-││axis 0│ 0.00  0.00  0.00 ││
-││      │ 0.00  0.00  1.00 ││
-││      │ ...   ...   ...  ││
-││      │ 0.00  0.00  4.00 ││
-││      │ 0.00  0.00  5.00 ││
-│└──────┴──────────────────┘│
-└───────────────────────────┘
+┌───────────────────────────────────┐
+│[7]: =>_tk shape 0:6,1:3           │
+│┌──────┬──────────────────────────┐│
+││      │axis 1                    ││
+│├──────┼──────────────────────────┤│
+││axis 0│ 5.68e+29  7.00e-45  0.00 ││
+││      │ 0.00      -5.01e-3  1.00 ││
+││      │ ...       ...       ...  ││
+││      │ -5.01e-3  4.08e-41  4.00 ││
+││      │ 4.08e-41  -5.01e-3  5.00 ││
+│└──────┴──────────────────────────┘│
+└───────────────────────────────────┘
 HERE: test/einsum/einsum_trivia_exec.ml:278:21
 ┌─────────────────────────────────────────────────────────────────┐
 │[2]: ;=>_c shape 0:3|1:4,2:2                                     │
(cd _build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations && .operations_tutorials.inline-tests/inline-test-runner.exe inline-test-runner operations_tutorials -partition hello_world_op.ml -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
File "test/operations/zero2hero_1of7.ml", line 1, characters 0-0:
/usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/operations/zero2hero_1of7.ml _build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations/zero2hero_1of7.ml.corrected
diff --git a/_build/default/test/operations/zero2hero_1of7.ml b/_build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations/zero2hero_1of7.ml.corrected
index ffc9c4b..5a455b3 100644
--- a/_build/default/test/operations/zero2hero_1of7.ml
+++ b/_build/.sandbox/25ea90858fdc8382afd7cce8a958480c/default/test/operations/zero2hero_1of7.ml.corrected
@@ -273,21 +273,21 @@ let%expect_test "Simple gradients hosted" =
   [%expect
     {|
                        #12 *._l
-                        0.00
+                        -3.28e-23
                        #13 grad_*._l
-                        0.00
+                        -3.28e-23
                  #8 +_d                   │#10 f non-emb
-                  0.00                    │ -2.00
+                  -3.28e-23               │ -2.00
                  #9 grad_+_d              │#11 grad_f
-                  0.00                    │ 0.00
+                  -3.28e-23               │ -3.28e-23
           #4 *._e            │#6 c non-emb│
-           0.00              │ 1.00e+1    │
+           3.02e-20          │ 1.00e+1    │
           #5 grad_*._e       │#7 grad_c   │
-           0.00              │ 0.00       │
+           -3.28e-23         │ -3.28e-23  │
     #0 a non-emb│#2 b non-emb│            │
      2.00       │ -3.00      │            │
     #1 grad_a   │#3 grad_b   │            │
-     0.00       │ 0.00       │            │
+     -1.75e+31  │ -1.64e+31  │            │
     |}];
   (* Do not update the params: all values and gradients will be at initial points, which are
      specified in the tensor in the brackets. *)
@@ -403,21 +403,21 @@ let%expect_test "Simple gradients virtual" =
   [%expect
     {|
                    #12 *._l
-                    0.00
+                    -3.28e-23
                    #13 grad_*._l Virt/40
                    <void>
              #8 +_d Virt/152              │#10 f non-emb
              <void>                       │ -2.00
              #9 grad_+_d Virt/151         │#11 grad_f
-             <void>                       │ 0.00
+             <void>                       │ -3.28e-23
       #4 *._e Virt/152       │#6 c non-emb│
       <void>                 │ 1.00e+1    │
       #5 grad_*._e Virt/151  │#7 grad_c   │
-      <void>                 │ 0.00       │
+      <void>                 │ -3.28e-23  │
     #0 a non-emb│#2 b non-emb│            │
      2.00       │ -3.00      │            │
     #1 grad_a   │#3 grad_b   │            │
-     0.00       │ 0.00       │            │
+     -1.68e+31  │ -1.66e+31  │            │
     |}];
   (* Do not update the params: all values and gradients will be at initial points, which are
      specified in the tensor in the brackets. *)
(cd _build/.sandbox/6961c661d4f43a4e96007f52b730c822/default/test/einsum && .einsum_tutorials.inline-tests/inline-test-runner.exe inline-test-runner einsum_tutorials -partition einsum_trivia.ml -source-tree-root ../.. -diff-cmd -)
Retrieving commandline, environment, or config file variable ocannl_log_level
Found 0, in the config file
File "test/einsum/einsum_trivia.ml", line 1, characters 0-0:
/usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/einsum/einsum_trivia.ml _build/.sandbox/6961c661d4f43a4e96007f52b730c822/default/test/einsum/einsum_trivia.ml.corrected
diff --git a/_build/default/test/einsum/einsum_trivia.ml b/_build/.sandbox/6961c661d4f43a4e96007f52b730c822/default/test/einsum/einsum_trivia.ml.corrected
index 2bcae70..465ab7e 100644
--- a/_build/default/test/einsum/einsum_trivia.ml
+++ b/_build/.sandbox/6961c661d4f43a4e96007f52b730c822/default/test/einsum/einsum_trivia.ml.corrected
@@ -976,144 +976,144 @@ let%expect_test "outer_sum simulating axis concatenation" =
   [%expect
     {|
     HERE: test/einsum/einsum_trivia.ml:975:21
-    ┌────────────────────────────────┐
-    │[9]: ;=>+ shape 0:4,1:5,2:6,3:3 │
-    │┌──────┬──────────────────┐     │
-    ││0 @ 0 │axis 3            │     │
-    │├──────┼──────────────────┤     │
-    ││0 @ 1 │ 0.00  0.00  0.00 │     │
-    ││axis 2│ 0.00  0.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 0.00  0.00  4.00 │     │
-    ││      │ 0.00  0.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││1 @ 1 │ 0.00  1.00  0.00 │     │
-    ││axis 2│ 0.00  1.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 0.00  1.00  4.00 │     │
-    ││      │ 0.00  1.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││2 @ 1 │ 0.00  2.00  0.00 │     │
-    ││axis 2│ 0.00  2.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 0.00  2.00  4.00 │     │
-    ││      │ 0.00  2.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││3 @ 1 │ 0.00  3.00  0.00 │     │
-    ││axis 2│ 0.00  3.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 0.00  3.00  4.00 │     │
-    ││      │ 0.00  3.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││4 @ 1 │ 0.00  4.00  0.00 │     │
-    ││axis 2│ 0.00  4.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 0.00  4.00  4.00 │     │
-    ││      │ 0.00  4.00  5.00 │     │
-    │└──────┴──────────────────┘     │
-    ├────────────────────────────────┤
-    │┌──────┬──────────────────┐     │
-    ││1 @ 0 │axis 3            │     │
-    │├──────┼──────────────────┤     │
-    ││0 @ 1 │ 1.00  0.00  0.00 │     │
-    ││axis 2│ 1.00  0.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 1.00  0.00  4.00 │     │
-    ││      │ 1.00  0.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││1 @ 1 │ 1.00  1.00  0.00 │     │
-    ││axis 2│ 1.00  1.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 1.00  1.00  4.00 │     │
-    ││      │ 1.00  1.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││2 @ 1 │ 1.00  2.00  0.00 │     │
-    ││axis 2│ 1.00  2.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 1.00  2.00  4.00 │     │
-    ││      │ 1.00  2.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││3 @ 1 │ 1.00  3.00  0.00 │     │
-    ││axis 2│ 1.00  3.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 1.00  3.00  4.00 │     │
-    ││      │ 1.00  3.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││4 @ 1 │ 1.00  4.00  0.00 │     │
-    ││axis 2│ 1.00  4.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 1.00  4.00  4.00 │     │
-    ││      │ 1.00  4.00  5.00 │     │
-    │└──────┴──────────────────┘     │
-    ├────────────────────────────────┤
-    │┌──────┬──────────────────┐     │
-    ││2 @ 0 │axis 3            │     │
-    │├──────┼──────────────────┤     │
-    ││0 @ 1 │ 2.00  0.00  0.00 │     │
-    ││axis 2│ 2.00  0.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 2.00  0.00  4.00 │     │
-    ││      │ 2.00  0.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││1 @ 1 │ 2.00  1.00  0.00 │     │
-    ││axis 2│ 2.00  1.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 2.00  1.00  4.00 │     │
-    ││      │ 2.00  1.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││2 @ 1 │ 2.00  2.00  0.00 │     │
-    ││axis 2│ 2.00  2.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 2.00  2.00  4.00 │     │
-    ││      │ 2.00  2.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││3 @ 1 │ 2.00  3.00  0.00 │     │
-    ││axis 2│ 2.00  3.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 2.00  3.00  4.00 │     │
-    ││      │ 2.00  3.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││4 @ 1 │ 2.00  4.00  0.00 │     │
-    ││axis 2│ 2.00  4.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 2.00  4.00  4.00 │     │
-    ││      │ 2.00  4.00  5.00 │     │
-    │└──────┴──────────────────┘     │
-    ├────────────────────────────────┤
-    │┌──────┬──────────────────┐     │
-    ││3 @ 0 │axis 3            │     │
-    │├──────┼──────────────────┤     │
-    ││0 @ 1 │ 3.00  0.00  0.00 │     │
-    ││axis 2│ 3.00  0.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 3.00  0.00  4.00 │     │
-    ││      │ 3.00  0.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││1 @ 1 │ 3.00  1.00  0.00 │     │
-    ││axis 2│ 3.00  1.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 3.00  1.00  4.00 │     │
-    ││      │ 3.00  1.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││2 @ 1 │ 3.00  2.00  0.00 │     │
-    ││axis 2│ 3.00  2.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 3.00  2.00  4.00 │     │
-    ││      │ 3.00  2.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││3 @ 1 │ 3.00  3.00  0.00 │     │
-    ││axis 2│ 3.00  3.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 3.00  3.00  4.00 │     │
-    ││      │ 3.00  3.00  5.00 │     │
-    │├──────┼──────────────────┤     │
-    ││4 @ 1 │ 3.00  4.00  0.00 │     │
-    ││axis 2│ 3.00  4.00  1.00 │     │
-    ││      │ ...   ...   ...  │     │
-    ││      │ 3.00  4.00  4.00 │     │
-    ││      │ 3.00  4.00  5.00 │     │
-    │└──────┴──────────────────┘     │
-    └────────────────────────────────┘
+    ┌────────────────────────────────────┐
+    │[9]: ;=>+ shape 0:4,1:5,2:6,3:3     │
+    │┌──────┬───────────────────────────┐│
+    ││0 @ 0 │axis 3                     ││
+    │├──────┼───────────────────────────┤│
+    ││0 @ 1 │ -4.69e+31  3.30e-41  0.00 ││
+    ││axis 2│ 0.00       1.93e+3   1.00 ││
+    ││      │ ...        ...       ...  ││
+    ││      │ 2.82e+3    3.04e+3   4.00 ││
+    ││      │ 0.00       0.00      5.00 ││
+    │├──────┼───────────────────────────┤│
+    ││1 @ 1 │ -4.69e+31  1.00     0.00  ││
+    ││axis 2│ 0.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 0.00       1.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││2 @ 1 │ -4.69e+31  2.00     0.00  ││
+    ││axis 2│ 0.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 0.00       2.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││3 @ 1 │ -4.69e+31  3.00     0.00  ││
+    ││axis 2│ 0.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 0.00       3.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││4 @ 1 │ -4.69e+31  4.00     0.00  ││
+    ││axis 2│ 0.00       1.94e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 0.00       4.00     5.00  ││
+    │└──────┴───────────────────────────┘│
+    ├────────────────────────────────────┤
+    │┌──────┬───────────────────────────┐│
+    ││1 @ 0 │axis 3                     ││
+    │├──────┼───────────────────────────┤│
+    ││0 @ 1 │ -4.69e+31  3.30e-41  0.00 ││
+    ││axis 2│ 1.00       1.93e+3   1.00 ││
+    ││      │ ...        ...       ...  ││
+    ││      │ 2.82e+3    3.04e+3   4.00 ││
+    ││      │ 1.00       0.00      5.00 ││
+    │├──────┼───────────────────────────┤│
+    ││1 @ 1 │ -4.69e+31  1.00     0.00  ││
+    ││axis 2│ 1.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 1.00       1.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││2 @ 1 │ -4.69e+31  2.00     0.00  ││
+    ││axis 2│ 1.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 1.00       2.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││3 @ 1 │ -4.69e+31  3.00     0.00  ││
+    ││axis 2│ 1.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 1.00       3.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││4 @ 1 │ -4.69e+31  4.00     0.00  ││
+    ││axis 2│ 1.00       1.94e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 1.00       4.00     5.00  ││
+    │└──────┴───────────────────────────┘│
+    ├────────────────────────────────────┤
+    │┌──────┬───────────────────────────┐│
+    ││2 @ 0 │axis 3                     ││
+    │├──────┼───────────────────────────┤│
+    ││0 @ 1 │ -4.69e+31  3.30e-41  0.00 ││
+    ││axis 2│ 2.00       1.93e+3   1.00 ││
+    ││      │ ...        ...       ...  ││
+    ││      │ 2.82e+3    3.04e+3   4.00 ││
+    ││      │ 2.00       0.00      5.00 ││
+    │├──────┼───────────────────────────┤│
+    ││1 @ 1 │ -4.69e+31  1.00     0.00  ││
+    ││axis 2│ 2.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 2.00       1.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││2 @ 1 │ -4.69e+31  2.00     0.00  ││
+    ││axis 2│ 2.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 2.00       2.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││3 @ 1 │ -4.69e+31  3.00     0.00  ││
+    ││axis 2│ 2.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 2.00       3.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││4 @ 1 │ -4.69e+31  4.00     0.00  ││
+    ││axis 2│ 2.00       1.94e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 2.00       4.00     5.00  ││
+    │└──────┴───────────────────────────┘│
+    ├────────────────────────────────────┤
+    │┌──────┬───────────────────────────┐│
+    ││3 @ 0 │axis 3                     ││
+    │├──────┼───────────────────────────┤│
+    ││0 @ 1 │ -4.69e+31  3.30e-41  0.00 ││
+    ││axis 2│ 3.00       1.93e+3   1.00 ││
+    ││      │ ...        ...       ...  ││
+    ││      │ 2.82e+3    3.04e+3   4.00 ││
+    ││      │ 3.00       0.00      5.00 ││
+    │├──────┼───────────────────────────┤│
+    ││1 @ 1 │ -4.69e+31  1.00     0.00  ││
+    ││axis 2│ 3.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 3.00       1.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││2 @ 1 │ -4.69e+31  2.00     0.00  ││
+    ││axis 2│ 3.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 3.00       2.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││3 @ 1 │ -4.69e+31  3.00     0.00  ││
+    ││axis 2│ 3.00       1.93e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 3.00       3.00     5.00  ││
+    │├──────┼───────────────────────────┤│
+    ││4 @ 1 │ -4.69e+31  4.00     0.00  ││
+    ││axis 2│ 3.00       1.94e+3  1.00  ││
+    ││      │ ...        ...      ...   ││
+    ││      │ 2.82e+3    3.04e+3  4.00  ││
+    ││      │ 3.00       4.00     5.00  ││
+    │└──────┴───────────────────────────┘│
+    └────────────────────────────────────┘
     |}];
   Train.printf ~here:[%here] ~with_code:false ~with_grad:false ti;
   [%expect
@@ -1125,18 +1125,18 @@ let%expect_test "outer_sum simulating axis concatenation" =
   [%expect
     {|
     HERE: test/einsum/einsum_trivia.ml:1124:21
-    ┌───────────────────────────┐
-    │[7]: =>_tk shape 0:6,1:3   │
-    │┌──────┬──────────────────┐│
-    ││      │axis 1            ││
-    │├──────┼──────────────────┤│
-    ││axis 0│ 0.00  0.00  0.00 ││
-    ││      │ 0.00  0.00  1.00 ││
-    ││      │ ...   ...   ...  ││
-    ││      │ 0.00  0.00  4.00 ││
-    ││      │ 0.00  0.00  5.00 ││
-    │└──────┴──────────────────┘│
-    └───────────────────────────┘
+    ┌────────────────────────────────────┐
+    │[7]: =>_tk shape 0:6,1:3            │
+    │┌──────┬───────────────────────────┐│
+    ││      │axis 1                     ││
+    │├──────┼───────────────────────────┤│
+    ││axis 0│ -4.69e+31  3.30e-41  0.00 ││
+    ││      │ 0.00       1.93e+3   1.00 ││
+    ││      │ ...        ...       ...  ││
+    ││      │ 2.82e+3    3.04e+3   4.00 ││
+    ││      │ 0.00       0.00      5.00 ││
+    │└──────┴───────────────────────────┘│
+    └────────────────────────────────────┘
     |}]
 
 let%expect_test "einsum with a leftmost input axis preserved as output axis" =
File "test/training/moons_demo_mini.expected", line 1, characters 0-0:
/usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/training/moons_demo_mini.expected _build/default/test/training/moons_demo_mini.exe.output
diff --git a/_build/default/test/training/moons_demo_mini.expected b/_build/default/test/training/moons_demo_mini.exe.output
index ab15b52..9f3da73 100644
--- a/_build/default/test/training/moons_demo_mini.expected
+++ b/_build/default/test/training/moons_demo_mini.exe.output
@@ -3,18 +3,18 @@ Found 0, in the config file
 Half-moons scatterplot and decision boundary:
 ┌─────────────────────────┐
 │*********#***************│
-│***###########********...│
-│**#############***.......│
+│***###########***********│
+│**#############******....│
 │*######%%*######.....%%%.│
-│#####**%%%..#####....%%%%│
+│#####**%%%*.#####....%%%%│
 │####***%%%%..####...%%%%.│
 │###***.%%%%...###..%%%%%.│
 │###*....%%%%%%%##.%%%%%..│
-│***......%%%%%%%%%%%%%...│
-│**.........%%%%%%%%%%....│
+│**.......%%%%%%%%%%%%%...│
+│*..........%%%%%%%%%%....│
 └─────────────────────────┘Loss:
 ┌─────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ 2.66│-                                                                                                   │
+│ 3.12│-                                                                                                   │
 │     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
@@ -23,8 +23,8 @@ Half-moons scatterplot and decision boundary:
 │     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
-│     │ -                                                                                                  │
 │     │                                                                                                    │
+│     │ -                                                                                                  │
 │     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
@@ -33,9 +33,11 @@ Half-moons scatterplot and decision boundary:
 │     │                                                                                                    │
 │     │                                                                                                    │
 │l    │                                                                                                    │
-│o    │  -                                                                                                 │
-│s    │   -                                                                                                │
+│o    │                                                                                                    │
 │s    │                                                                                                    │
+│s    │  -                                                                                                 │
+│     │   -                                                                                                │
+│     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
@@ -45,48 +47,45 @@ Half-moons scatterplot and decision boundary:
 │     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
+│     │                                                                                                    │
 │     │      -                                                                                             │
 │     │                                                                                                    │
-│     │       -  - -                                                                                       │
-│     │                -                                                                                   │
-│     │        -  -                                                                                        │
-│     │             - -  - -  -        -  -                                                            -   │
-│     │                                 -                                                                  │
-│     │                 -       ---        ---           -              -                 -                │
-│ 0.00│                     --     - --        ---- ----  --- ---- ----  --- ---- ---- ---  ---- ---- - -- │
+│     │          --                                      -                                                 │
+│     │       -                        -  -                                               -            -   │
+│     │        -   -          -         -    -      -  -   -     - -    - -        -                       │
+│ 0.00│             - ---- ---  ---- --    --  ----  --   - - ---   ---  - - ---- - -- ---  ---- ---- - -- │
 ├─────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
 │     │0.00                                                                                         7.90e+1│
 │     │                                                step                                                │
 └─────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘Log-loss, for better visibility:
 ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ 9.79e-1 │-                                                                                                   │
+│ 1.13    │-                                                                                                   │
 │         │ -                                                                                                  │
 │         │                                                                                                    │
 │         │  --                                                                                                │
-│         │     -                                                                                              │
 │         │                                                                                                    │
-│         │      -                                                                                             │
-│         │       -  - -   -                                                                                   │
-│         │        -  -                                                                                        │
-│         │             - -  - -  -        -  -                                                                │
-│         │                                 -                                                              -   │
-│         │                                     -            -                                -                │
-│         │                 -                  -                                                               │
-│         │                         ---          -                          -                                  │
-│         │                     -        -            - -  -  -   -                                            │
-│         │                                                          -                 -         -             │
-│l        │                      -                        -             -      -                   -  - -      │
-│o        │                                                                   -                                │
-│g        │                               -                              -                                     │
-│         │                                              -     -                        -            -      -  │
-│l        │                                        - -                    -                -                   │
-│o        │                                                         -                                          │
-│s        │                                                                      -                             │
-│s        │                                                        -   -          -                            │
+│         │     -                                                                                              │
 │         │                                                                                                    │
 │         │                                                                                                    │
+│         │      -                                                                                             │
+│         │          --                                      -                                                 │
+│         │       -                        -  -                                                                │
+│         │            -                         -                          -                 -            -   │
+│         │                       -         -           -  -   -              -        -                       │
+│         │        -        -                                        - -          -                            │
+│         │                -                   -         -                                            - -   -  │
+│         │             -             -   -          --           -     -      - -       -       -             │
+│l        │                    --   --                    -              -         -       -       -           │
+│o        │               -              -      -                  -      -                     -    -         │
+│g        │                                                   -     -                                          │
+│         │                                        -                                              -            │
+│l        │                      -                                                    -                        │
+│o        │                                                                                                    │
+│s        │                                                                                                    │
+│s        │                                                                         -                          │
+│         │                            -                        -                                              │
 │         │                                                                                                    │
-│         │                                                                                       -            │
+│         │                  -                                                                                 │
 │         │                                                                                                    │
 │         │                                                                                                    │
 │         │                                                                                                    │
@@ -95,10 +94,11 @@ Half-moons scatterplot and decision boundary:
 │         │                                                                                                    │
 │         │                                                                                                    │
 │         │                                                                                                    │
+│         │                                                                                  -                 │
 │         │                                                                                                    │
 │         │                                                                                                    │
 │         │                                                                                                    │
-│ -1.00e+1│                            -            -           -            -     -- -  -  --  -      -  -  - │
+│ -1.00e+1│                                         -                        -          -   -          -  -  - │
 ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
 │         │0.00                                                                                         7.90e+1│
 │         │                                                step                                                │
File "test/training/moons_demo_parallel.expected", line 1, characters 0-0:
/usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/training/moons_demo_parallel.expected _build/default/test/training/moons_demo_parallel.exe.output
diff --git a/_build/default/test/training/moons_demo_parallel.expected b/_build/default/test/training/moons_demo_parallel.exe.output
index 26ca5c7..6f779fc 100644
--- a/_build/default/test/training/moons_demo_parallel.expected
+++ b/_build/default/test/training/moons_demo_parallel.exe.output
@@ -1,6 +1,6 @@
 Retrieving commandline, environment, or config file variable ocannl_log_level
 Found 0, in the config file
-Epoch loss: 10.801180
+Epoch loss: 11.252870
 
 Half-moons scatterplot and decision boundary:
 ┌────────────────────────────────────────────────────────────────────────────────────────────────────┐
@@ -9,28 +9,28 @@ Half-moons scatterplot and decision boundary:
 │*******************#******#*#**#**##**##*#**#***#***************************************************│
 │***************#********#***#**#**#**#*#*##**##**##*************************************************│
 │*************#****#*###**#*##*#*##***##*#######******#**********************************************│
-│*****************#*#****#***#***#*#*#*********#*##*##*********************************************..│
-│************#***##*###***#*#********#**#***##*#***##***#***************************************.....│
-│**********#***#****#####*#***###******#*#*#***#***#***#*##**#********************************.......│
-│***********####*#***#****#*#*****##****#**********##*************************************...........│
-│*********#****#***#**##*#*#*#*******#**##*#*##*##*#*#**#**##**************************..............│
-│**********#**#*###**#**********************#***#***#**#*##***#*********************.................│
-│***#**#****###*##*#*##************%************##*##**#*#*#*********************....................│
-│***#**#**#*#***###**#*#**********%******%********#***#****#****#*************.......................│
-│********#**###***#******************************#***#*#**#***###*********...............%.%....%....│
-│**#********####*#**#**********%******%*%************#**##***##***#****............................%.│
-│****##*****##****#************%******%****************#**#*#*#*##*#.............................%...│
-│***#*#**#***#*#****************%*%*****%*************#*##*#*#**..#...........................%.%.%..│
-│*#*#**#**#*#***#***************%*%*%%*%*****************###*....#.###...................%.%..%...%.%│
-│****##**##******************************%****************###.##..#.#.....................%%%..%.%%..│
-│***#***##*##*****************%*%%*%%%*%*************.....#.#.####.#.#...................%%%%.%..%...│
-│#*#*#*#**#*#*****************%****%*%*%%%******...........#.............................%....%.%.%..│
-│##*##***#**#*****************%***%%*%%**%*.%..................#...#.#............................%%.│
-│*#**#******#*********************%**%...%%..................##...#.###...............%.%%.%..%......│
-│#***#*##**#*********************.%...%...%.......................#..#.................%.%%.%%%..%%%.│
-│*#*##***##*****************......%.%%%.%....................###..#.##..................%.%%.........│
-│******#*#*************............%%.%.%%%%.........................#............%.%..%%..%..%......│
-│*****#****#******.....................%.%.%...................#....#..............%.%..%..%.%..%%%..│
+│*****************#*#****#***#***#*#*#*********#*##*##***********************************************│
+│************#***##*###***#*#********#**#***##*#***##***#********************************************│
+│**********#***#****#####*#***###******#*#*#***#***#***#*##**#***************************************│
+│***********####*#***#****#*#*****##****#**********##************************************************│
+│*********#****#***#**##*#*#*#*******#**##*#*##*##*#*#**#**##**************************************..│
+│**********#**#*###**#**********************#***#***#**#*##***#*******************************.......│
+│***#**#****###*##*#*##************%************##*##**#*#*#*****************************............│
+│***#**#**#*#***###**#*#**********%******%********#***#****#****#*******************.................│
+│********#**###***#******************************#***#*#**#***###**************..........%.%....%....│
+│**#********####*#**#**********%******%*%************#**##***##***#********........................%.│
+│****##*****##****#************%******%****************#**#*#*#*##*#**...........................%...│
+│***#*#**#***#*#****************%*%*****%*************#*##*#*#***.#...........................%.%.%..│
+│*#*#**#**#*#***#***************%*%*%%*%*****************###.....#.###...................%.%..%...%.%│
+│****##**##******************************%*************...###.##..#.#.....................%%%..%.%%..│
+│***#***##*##*****************%*%%*%%%*%**********........#.#.####.#.#...................%%%%.%..%...│
+│#*#*#*#**#*#*****************%****%*%*%%%****.............#.............................%....%.%.%..│
+│##*##***#**#*****************%***%%*%%**%..%..................#...#.#............................%%.│
+│*#**#******#*********************%*.%...%%..................##...#.###...............%.%%.%..%......│
+│#***#*##**#*******************...%...%...%.......................#..#.................%.%%.%%%..%%%.│
+│*#*##***##***************........%.%%%.%....................###..#.##..................%.%%.........│
+│******#*#************.............%%.%.%%%%.........................#............%.%..%%..%..%......│
+│*****#****#*****......................%.%.%...................#....#..............%.%..%..%.%..%%%..│
 │*****##****.....................%...%...%%.%%..%...................#............%....%%%.%%..%......│
 │******............................%..%.%%.%%.%..%%.%%...........#.##..............%%%%.........%....│
 │*..................................%%..%...%..%%%%..........................%%%.....%.%..%.%........│
@@ -47,53 +47,53 @@ Half-moons scatterplot and decision boundary:
 └────────────────────────────────────────────────────────────────────────────────────────────────────┘
 Batch Log-loss:
 ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ 3.84    │-                                                                                                   │
-│         │                                                                                                    │
+│ 3.79    │-                                                                                                   │
 │         │                                                                                                    │
 │         │-                                                                                                   │
 │         │                                                                                                    │
 │         │                                                                                                    │
-│         │ - -                --                                                                              │
-│         │  -  ---- ---- ----   -- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │
-│         │ -                               - ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │
-│         │------------------- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │
-│         │-         -        - -- ---- ---- ----  ---- ---- ---- ---- ---- ----                               │
-│         │-   -- - - -- ----                     ---- ---                       ---- ---- ---- ---- ---- ---  │
-│         │---   -            --   -                      - ---- ---  --                                     - │
-│b        │                     --  ---                             -   -- - -- - -- --- -                     │
-│a        │- -  -                       -- -                  - -   --   ------ ------ -- ---------------------│
-│t        │ -     -                 --    -  ----      -  ---  - - -  - -      -        -                     -│
-│c        │  - - -- -- ---      ---- ---- ---- ----  --- -- - -  -  -   -  -    -                              │
-│h        │ -  -  --  --   ---   - -  --- -    -   --      -   --  - --  --  ---  ---- ---- ---- ----          │
-│         │   -  - -- -  ------ -            -  -   -                                                 ---- ----│
-│l        │   -     - - -- - - -       --      - --  - -  - - -      -                 -                       │
-│o        │      ---        -             ---       - -- - --  --- -- - ---- -- -       -         -    -       │
-│g        │   --       -        -    -                                         -  ---- -  -    - - -- - -    --│
-│         │                   -       -              -   -                      -        -  ---          - --  │
-│l        │                        -                    --    -                                             -  │
-│o        │-                      -                       -    -                                               │
-│s        │                      -                           -                                                 │
-│s        │ -  -                                    -       -     -                                     -     -│
-│         │        -       -                           -      -  -  -                                    - -   │
-│         │                                       -            -       -                                       │
-│         │                                                             -                                    - │
-│         │                                          -    -                     -                              │
-│         │              -                                                                                     │
-│         │                                                -          -                                        │
-│         │            -                                           -                                           │
-│         │                                           -                                                        │
-│         │                                                     -                                              │
+│         │                                                                                                    │
+│         │ -                  --   ----  - - -- - ---- -  -   -    -                                          │
+│         │      --- ---- ----   --      - -    -        --  -- - -- - ---- ---- ---- ---- ---- ---- ---- ---- │
+│         │  -- -                 -  - -  - -  - -   -  ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │
+│         │------------------------ ---- ---- ---- --------- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- │
+│         │ - ----- ---- ---       - -  - -  - -  -  -  --- - --- --    --- ---- ---- ---- ---- ----           │
+│         │ -   -             ----  - -  -                   --  ---- ---- ---- -- -                 ---- ---- │
+│         │ --- -           -      - -     -  - -  --  -                          -  ---- ---- ---- -          │
+│b        │  -  --                      -    -    -  -    -                                          ----------│
+│a        │         --                  - --   -         -  -    -    - -      -          - ---- ---- -    -- -│
+│t        │      ---  -   -                  -  - -  ---- - - ---  ---   --- -- - ---- ---                     │
+│c        │      -    --   - -   -   -- - ---- ---- ---  ----   -- -    - --   -- - -                          │
+│h        │   -   - - -- - --      -             -     -   -     -   --  - ---  -  - - -- - ---- -- - --    -  │
+│         │         -   --    - -- -  --                       -    -   -    --    -               -    -- - --│
+│l        │        -    -   - -  --   -- -- -  -             -                   --      -                     │
+│o        │             -     --   -                   -                 -           -                         │
+│g        │              - --             --                  -   - -  -                -                      │
+│         │      - -        -     -   -     -  -                                                               │
+│l        │              -           -                          -             -       --            -          │
+│o        │                  -  -      -                    -                              --                  │
+│s        │                                -           -                       -                               │
+│s        │       -                                      -              -         -            -               │
+│         │                    -              -                           -         -        -  -              │
+│         │                                     -                                                              │
+│         │                   -                                                        -                       │
+│         │                                                                                 -                  │
+│         │                                                        -         -       -    -                    │
+│         │                                                                                                    │
+│         │                                                                                   -                │
+│         │                                                                                                    │
+│         │                                                                                                    │
 │         │                                                                                                    │
 │         │                                                                                                    │
-│         │                                    -                                                               │
-│ -1.04e+1│ ---------------------------------------------------------------------------------------------------│
+│         │                                                                                                    │
+│ -1.04e+1│  --  ----------------------------------------------------------------------------------------------│
 ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
 │         │0.00                                                                                         1.59e+3│
 │         │                                                step                                                │
 └─────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘
 Epoch Log-loss:
 ┌─────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ 4.38│-                                                                                                   │
+│ 4.61│-                                                                                                   │
 │     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
@@ -111,28 +111,28 @@ Epoch Log-loss:
 │o    │                                                                                                    │
 │c    │                                                                                                    │
 │h    │                                                                                                    │
-│     │                                                                                                    │
-│l    │                                                                                                    │
+│     │   -                                                                                                │
+│l    │ -                                                                                                  │
 │o    │                                                                                                    │
 │g    │                                                                                                    │
-│     │                                                                                                    │
+│     │  -                                                                                                 │
 │l    │                                                                                                    │
 │o    │                                                                                                    │
 │s    │                                                                                                    │
 │s    │                                                                                                    │
 │     │                                                                                                    │
+│     │     -                                                                                              │
+│     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
 │     │                                                                                                    │
-│     │ -                                                                                                  │
 │     │                                                                                                    │
-│     │   -      -                                                                                         │
-│     │  -  ----  --- ---- --   --                                                                         │
-│     │                      --   -- ----                                                                  │
-│     │                                   ---- ---- ---                                                    │
-│     │                                                - ---- ---- ---  -                                  │
-│     │                                                               -  --- ---- ---  ---                 │
-│ 2.37│                                                                              -    - ---- ---- ---- │
+│     │      --- -                                                                                         │
+│     │           --- -- - ---  ---- -  - -     -                                                          │
+│     │                 -     -       --   --- - -- ----   -                                               │
+│     │                                                  -- -  --- - -    -                                │
+│     │                                                       -     - - -- - ---- ----  -                  │
+│ 2.42│                                                                                - -- ---- ---- ---- │
 ├─────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
 │     │0.00                                                                                         7.90e+1│
 │     │                                                step                                                │
File "test/training/moons_demo.expected", line 1, characters 0-0:
/usr/bin/git --no-pager diff --no-index --color=always -u _build/default/test/training/moons_demo.expected _build/default/test/training/moons_demo.exe.output
diff --git a/_build/default/test/training/moons_demo.expected b/_build/default/test/training/moons_demo.exe.output
index a39bfc5..4c38f33 100644
--- a/_build/default/test/training/moons_demo.expected
+++ b/_build/default/test/training/moons_demo.exe.output
@@ -2,49 +2,49 @@ Retrieving commandline, environment, or config file variable ocannl_log_level
 Found 0, in the config file
 Half-moons scatterplot and decision boundary:
 ┌────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│****************************************#***********************************************************│
-│********************************##*#*******#********************************************************│
-│*******************#******#*#**#**##**##*#**#***#************************************...............│
-│***************#********#***#**#**#**#*#*##**##**##*****************************....................│
-│*************#****#*###**#*##*#*##***##*#######******#**********************........................│
-│*****************#*#****#***#***#*#*#*********#*##*##*******************............................│
-│************#***##*###***#*#********#**#***##*#***##***#*************...............................│
-│**********#***#****#####*#***###******#*#*#***#***#***#*##**#****...................................│
-│***********####*#***#****#*#*****##****#**********##**********......................................│
-│*********#****#***#**##*#*#*#*......#**##*#*##*##*#*#**#**##........................................│
-│**********#**#*###**#*******...........****#***#***#**#*##...#......................................│
-│***#**#****###*##*#*##*****.......%......******##*##**#.#.#.........................................│
-│***#**#**#*#***###**#*#**........%......%....****#***#....#....#....................................│
-│********#**###***#******........................#***#*#..#...###........................%.%....%....│
-│**#********####*#**#***.......%......%.%..........**#**##...##...#................................%.│
-│****##*****##****#*****.......%......%..............**#**#.#.#.##.#.............................%...│
-│***#*#**#***#*#*******.........%.%.....%.............#*##*#*#....#...........................%.%.%..│
-│*#*#**#**#*#***#******.........%.%.%%.%................*###****.#.###...................%.%..%...%.%│
-│****##**##***********...................%................###*##*.#.#.....................%%%..%.%%..│
-│***#***##*##********.........%.%%.%%%.%..................#.#*####.#.#...................%%%%.%..%...│
-│#*#*#*#**#*#********.........%....%.%.%%%.................#...**........................%....%.%.%..│
-│##*##***#**#*******..........%...%%.%%..%..%..................#...#.#............................%%.│
-│*#**#******#*******..............%..%...%%..................##...#.###...............%.%%.%..%......│
-│#***#*##**#*******...............%...%...%.......................#..#.................%.%%.%%%..%%%.│
-│*#*##***##*******................%.%%%.%....................###..#.##..................%.%%.........│
-│******#*#********.................%%.%.%%%%.........................#............%.%..%%..%..%......│
-│*****#****#*****......................%.%.%...................#....#..............%.%..%..%.%..%%%..│
-│*****##*********................%...%...%%.%%..%...................#............%....%%%.%%..%......│
-│***************...................%..%.%%.%%.%..%%.%%...........#.##..............%%%%.........%....│
-│***************....................%%..%...%..%%%%..........................%%%.....%.%..%.%........│
-│**************........................%..%.%..%..%....................%..%..%%%%.%.%%%%%...%%.......│
-│*************.......................%%...%%%.%%%%.%.%.%%.....%...........%.%%%%%.%....%.%...........│
-│*************...........................%..%.......%.%%%....%%.%..%%%.%%..%.%.%.%..%................│
-│************............................%...%%..%....%..%%.%%.%.%....%....%%%..%.%%..%.%............│
-│************...............................%%....%..%.%%%..%%%....%%.%...%%...%%...%%...%%..........│
-│***********..................................%.%..%..%%%%%.%....%%...%%.%..%..%....%................│
-│***********................................%............%..%%.....%%..%.%.%....%....................│
-│**********.........................................%...%.%..%.%.......%%...%........................│
-│**********..............................................%%%.....%.%..%..%.....%..%..................│
-│*********.............................................%....%%%%.%.%.%....%..........................│
+│****************************************#****************************************************.......│
+│********************************##*#*******#************************************************........│
+│*******************#******#*#**#**##**##*#**#***#******************************************.........│
+│***************#********#***#**#**#**#*#*##**##**##**************************************...........│
+│*************#****#*###**#*##*#*##***##*#######******#**********************************............│
+│*****************#*#****#***#***#*#*#*********#*##*##**********************************.............│
+│************#***##*###***#*#********#**#***##*#***##***#******************************..............│
+│**********#***#****#####*#***###******#*#*#***#***#***#*##**#************************...............│
+│***********####*#***#****#*#*****##****#**********##********************************................│
+│*********#****#***#**##*#*#*#*****..#**##*#*##*##*#*#**#**##***********************.................│
+│**********#**#*###**#************........**#***#***#**#*##***#********************..................│
+│***#**#****###*##*#*##*********...%........****##*##**#*#*#**********************...................│
+│***#**#**#*#***###**#*#******....%......%...*****#***#****#****#*****************...................│
+│********#**###***#**********.................***#***#*#**#***###****************........%.%....%....│
+│**#********####*#**#*******...%......%.%.......*****#**##***##***#*************...................%.│
+│****##*****##****#********....%......%..........******#**#*#*#*##*#***********..................%...│
+│***#*#**#***#*#***********.....%.%.....%.........****#*##*#*#****#***********................%.%.%..│
+│*#*#**#**#*#***#*********......%.%.%%.%...........******###*****#*###*******............%.%..%...%.%│
+│****##**##**************................%...........*****###*##**#*#*******..............%%%..%.%%..│
+│***#***##*##***********......%.%%.%%%.%..............****#*#*####*#*#******.............%%%%.%..%...│
+│#*#*#*#**#*#***********......%....%.%.%%%.............****#***************..............%....%.%.%..│
+│##*##***#**#**********.......%...%%.%%..%..%...........*******#***#*#****........................%%.│
+│*#**#******#**********...........%..%...%%..............****##***#*###**.............%.%%.%..%......│
+│#***#*##**#**********............%...%...%................*******#**#**...............%.%%.%%%..%%%.│
+│*#*##***##***********............%.%%%.%...................*###**#*##*.................%.%%.........│
+│******#*#***********..............%%.%.%%%%.................********#............%.%..%%..%..%......│
+│*****#****#********...................%.%.%...................#****#*.............%.%..%..%.%..%%%..│
+│*****##************.............%...%...%%.%%..%................***#............%....%%%.%%..%......│
+│******************................%..%.%%.%%.%..%%.%%...........#.##..............%%%%.........%....│
+│******************.................%%..%...%..%%%%..........................%%%.....%.%..%.%........│
+│*****************.....................%..%.%..%..%....................%..%..%%%%.%.%%%%%...%%.......│
+│*****************...................%%...%%%.%%%%.%.%.%%.....%...........%.%%%%%.%....%.%...........│
+│****************........................%..%.......%.%%%....%%.%..%%%.%%..%.%.%.%..%................│
+│****************........................%...%%..%....%..%%.%%.%.%....%....%%%..%.%%..%.%............│
+│***************............................%%....%..%.%%%..%%%....%%.%...%%...%%...%%...%%..........│
+│***************..............................%.%..%..%%%%%.%....%%...%%.%..%..%....%................│
+│**************.............................%............%..%%.....%%..%.%.%....%....................│
+│**************.....................................%...%.%..%.%.......%%...%........................│
+│*************...........................................%%%.....%.%..%..%.....%..%..................│
+│*************.........................................%....%%%%.%.%.%....%..........................│
 └────────────────────────────────────────────────────────────────────────────────────────────────────┘Loss:
 ┌────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ 2.61e+1│-                                                                                                   │
+│ 2.66e+1│-                                                                                                   │
 │        │                                                                                                    │
 │        │                                                                                                    │
 │        │                                                                                                    │
@@ -74,59 +74,59 @@ Half-moons scatterplot and decision boundary:
 │        │                                                                                                    │
 │        │                                                                                                    │
 │        │                                                                                                    │
+│        │-                                                                                                   │
 │        │                                                                                                    │
 │        │                                                                                                    │
-│        │                                                                                                    │
-│        │                                                                                                    │
-│        │ --                   -                                                                             │
-│        │-     ---    - ---     -                                                                            │
-│        │   - - -   --  - -- -- - ----  -   ---    --       --                                               │
-│        │----  - - - --  - - - -- ---- - --    - --   ---- ---- ---- ---- ---   -                            │
-│        │-------------- ---- ---- ---- ---- --    --- ---- - -- ---- ---- ---- ---- ---- ---- ---- ---- ---- │
+│        │        -                                                                                           │
+│        │-  - --   ----                                                                                      │
+│        │ --    -                                                                                            │
+│        │   - -- -   -- ---            --                                                                    │
+│        │---  - -  --   ---- ---- ---- ---- -- -                                                             │
+│        │---- ----   -- ---- ---- ---- ---- ---  ---- --   -    -         -                                  │
 │ 0.00   │----------------------------------------------------------------------------------------------------│
 ├────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
 │        │0.00                                                                                         6.39e+3│
 │        │                                                step                                                │
 └────────┴────────────────────────────────────────────────────────────────────────────────────────────────────┘Log-loss, for better visibility:
 ┌─────────┬────────────────────────────────────────────────────────────────────────────────────────────────────┐
-│ 3.26    │-                                                                                                   │
-│         │                                                                                                    │
+│ 3.28    │-                                                                                                   │
 │         │                                                                                                    │
 │         │                                                                                                    │
 │         │                                                                                                    │
 │         │                                                                                                    │
-│         │ --                   -                                                                             │
-│         │-  -  ---   -- ---- -  -           -                                                                │
-│         │     ---  -- - ---   --- ---- ----  --- ---- ---- ---- -                                            │
-│         │---- ---- - -     - -  - ---- --                  ---  ---- ---- ---- ---- -- - ----    -   -    -  │
-│         │ --- --------- ---- -- - ---- - -- -       - ----    -  --- ---- ---- ---- ---- ---- ---- ---- ---- │
-│         │-------- ----  ---- ---- ---- --    ---  --         -                                               │
-│         │--   - -   --- ---  ---  -  -           -         -- -                                              │
-│         │--   --- - --  -  -   --  -   --              ---  --- ----    -  -                                 │
-│         │--- ----  - --   -   ---       -           - -    -- -      ---  - --  -                            │
-│         │--  -  - -    -  --  -      -  --               -   -  -            - - -- ---        -   -  -  - - │
-│l        │- - - -- -  -  ---  -  - -- -  --  -  ----- --   -  -   --  -   ---   - ------- ---- --------  - -  │
-│o        │- ---- ---- -- ------ - -  -   -- ----- -- -  ---------- -- ---- ------- ---- -- ------ - -- - -- - │
-│g        │  - --  ---- ------ - ----- -  ---- -   -- -- - ---  ---- ---- ----  -- --   - ------------- ----- -│
-│         │- - -- --  ------ -- -  ----  - -- --    --  -- -  - - ------  -    - - ---- -- -  - - -- - ---   --│
-│l        │---    --  - ---- - -  --- ---   -- - -   --  --- -------- -------    - -   - --- -----  - -------  │
-│o        │- - -  --  ----- - - ---- - --  - - -- --- -- -  ------   ---- ---- - ---- -           --        - -│
-│s        │-      --------- -  --  ----  ---- -- -  -- ----  -- ---- -   -- - -----   -- - -   -       -     --│
-│s        │    ----   - -------- --  -----   -    -- -  -  -  - --- --   -- -- -    ---  -      -  ---  - -  - │
-│         │  - --   ------ - - --   -- ----- -- --     -  ---- ----  --    -  ---     - - -  - -- -  --  ---   │
-│         │-- --   -  --      -  - -  - -  - ---       -  - -- -   -  --  -- - - - -    -  -  -  ----      -  -│
-│         │--- -- -   - - -       -  -        ----  ----   ---  ---        --  -- -  -    --   -- -  -  - -  --│
-│         │ -  -       -      ---   -   -  --  -                 -  - -     - -           -   -      -   -     │
-│         │    --  ---    -          - - -            -    -    --       --       -----       -              - │
-│         │  - -      -       -        - - - -              -            -         --       -       -          │
-│         │ -          -                 --                -       -   -      -     -    -             - -  -  │
-│         │  -          -   -                                     -                -                     -     │
-│         │   -      -                       -                            -                                    │
-│         │   -                                           -             -                                      │
-│         │    -               -                                 -                                             │
+│         │-                                                                                                   │
+│         │-  - -- -   --                                                                                      │
+│         │ --    -  --                                                                                        │
+│         │   - ----   -- ---   --  -- - --                                                                    │
+│         │---- --   -- - ---- ---- - -- ---- ----                                                             │
+│         │---- ----   -- ---- ---- ----  --  --   ---- --                                                     │
+│         │---- ----    - ---- ---- ---- ----   -   -        -    -         -                                  │
+│         │-  --- -- - --      -- -   --  -   -  -   -   --   -    -   ---   - -             -                 │
+│         │- ------- -  - -- - --   -      --           -      -     -        -    -   -                   -   │
+│         │--- -  --  - - ---    --  -                     - -- - - -    -            - --  -                  │
+│         │ -  -----  --- -    --   ---- --           -   -  - - - -   - --      --  --    -  - -  -- - -  ----│
+│l        │- - -   ----     -   -- --  -  -    --  ---   - - -- --- - --- - -   --  - -  -   -- - -   -  -  -  │
+│o        │- -----  --- -  --   -  -  -   - - ------   --- ----  --- ---   ---- -- -- --  - -  - -   -    -   -│
+│g        │-- --  ----- -  ---  --   - -   -    --  - -----  -- -     --- -----  -- - -- --      -  -- - -   --│
+│         │-  ------ ---  -- -  ------- --  - -- ---- - -- --   -- - -  -   -- - -- -     --  -   ---  -  -  - │
+│l        │-   ---------------- - -  -- --- ----  - ---- -  ---  --  -   --             -  -    -     -      - │
+│o        │- - ------  -- -     -    --- -- --   - --  -   -- -- -   - - - -                 -  -       -    --│
+│s        │ ----- - ---- ---- -- -   -------  -- --  -- --         -    -      - -          - -      -- ----   │
+│s        │-  -- -- --- -- -    --   -- ---- -  --   --- -    --      -    - - ---   -     -  --        -  --  │
+│         │ ----  -----   -    ----- -------- -- - -  ---    -        --   -  -                  - -      --   │
+│         │     --  -- - --    --     ------ --   --      - - -   -  -  -         -   -       - -              │
+│         │-   -- - -      -  ---  -- - --  --     -    -                   -            --     -            - │
+│         │-       - -    --  -  -   -- - -  - -        -      - --  -         -  -     -           -          │
+│         │      -  -    - -    -    - -  -- -  -         -             -                   -                  │
+│         │                   - -       -   --- -   -          -  -                          -                -│
+│         │  -             -  --         ---           -             - -                                       │
+│         │                  -       -- - -- -   -        -      -                                             │
+│         │  --                              -  -                                                              │
+│         │                                 - -                                 -                              │
+│         │                               -                                                                    │
+│         │                                                           -                                        │
+│         │                               -                                                                    │
 │         │                                                                                                    │
-│         │                            -                                                   -                   │
-│         │                                  -   -                                                             │
 │         │                                                                                                    │
 │ -1.00e+1│----------------------------------------------------------------------------------------------------│
 ├─────────┼────────────────────────────────────────────────────────────────────────────────────────────────────┤
"/usr/bin/env" "bash" "-c" "opam exec -- dune build @install @check @runtest && rm -rf _build" failed with exit status 1
2025-08-14 09:17.32: Job failed: Failed: Build failed