Code

add dims for chunk and test_zygote#47

Code

Merged

CarloLucibello merged 13 commits intomainJuliaML/MLUtils.jl:mainfrom cl/chunkJuliaML/MLUtils.jl:cl/chunkCopy head branch name to clipboardFeb 10, 2022

+121-10

Commits

Commits on Feb 9, 2022

• add chunk dims and test_zygote

  CarloLucibello

  committed
  4a918bb

  Copy full SHA for 4a918bb

• cleanup

  CarloLucibello

  committed
  9ba0bf4

  Copy full SHA for 9ba0bf4

• cleanup

  CarloLucibello

  committed
  e41c872

  Copy full SHA for e41c872

• make chunk differentiable

  CarloLucibello

  committed
  df41460

  Copy full SHA for df41460

• check_inferred=false

  CarloLucibello

  committed
  f75e430

  Copy full SHA for f75e430

• define rule for chunk

  CarloLucibello

  committed
  507b636

  Copy full SHA for 507b636

• second order

  CarloLucibello

  committed
  06af88e

  Copy full SHA for 06af88e

• fix pullback

  CarloLucibello

  committed
  80658db

  Copy full SHA for 80658db

• Update src/utils.jl

  Show description for bb48f1c
  

  CarloLucibello

  and

  mcabbott

  authored
  bb48f1c

  Copy full SHA for bb48f1c

• Update utils.jl

  CarloLucibello

  authored
  269ec67

  Copy full SHA for 269ec67

Commits on Feb 10, 2022

• Update src/utils.jl

  Show description for 65bf57a
  

  CarloLucibello

  and

  mcabbott

  authored
  65bf57a

  Copy full SHA for 65bf57a

• remove collect

  CarloLucibello

  committed
  8788ca1

  Copy full SHA for 8788ca1

• cleanup

  CarloLucibello

  committed
  1c6a8c5

  Copy full SHA for 1c6a8c5