Archived
1
0
Fork 0

gnu: Add julia-zygote.

* gnu/packages/julia-xyz.scm (julia-zygote): New variable.

Signed-off-by: Ludovic Courtès <ludo@gnu.org>
This commit is contained in:
zimoun 2021-03-25 02:05:16 +01:00 committed by Ludovic Courtès
parent 568cebc258
commit 13f7f06c49
No known key found for this signature in database
GPG key ID: 090B11993D9AEBB5

View file

@ -1221,3 +1221,38 @@ system.")
(description "Minimal package which enables to add custom gradients to (description "Minimal package which enables to add custom gradients to
Zygote, without depending on Zygote itself.") Zygote, without depending on Zygote itself.")
(license license:expat))) (license license:expat)))
(define-public julia-zygote
(package
(name "julia-zygote")
(version "0.6.4")
(source
(origin
(method git-fetch)
(uri (git-reference
(url "https://github.com/FluxML/Zygote.jl")
(commit (string-append "v" version))))
(file-name (git-file-name name version))
(sha256
(base32 "1h2ph0lsisbkh8y4xgwzgw9p5zi243q8zzp5gfh3zw9pzkx6a1rf"))))
(build-system julia-build-system)
(arguments
`(#:tests? #f)) ;require CUDA, not packaged yet
(propagated-inputs
`(("julia-abstractffs" ,julia-abstractffts)
("julia-chainrules" ,julia-chainrules)
("julia-diffrules" ,julia-diffrules)
("julia-fillarrays" ,julia-fillarrays)
("julia-forwarddiff" ,julia-forwarddiff)
("julia-irtools" ,julia-irtools)
("julia-macrotools" ,julia-macrotools)
("julia-nanmath" ,julia-nanmath)
("julia-requires" ,julia-requires)
("julia-specialfunctions" ,julia-specialfunctions)
("julia-zygote-rules" ,julia-zygoterules)))
(home-page "https://fluxml.ai/Zygote.jl")
(synopsis "Automatic differentiation in Julia")
(description "Zygote provides source-to-source automatic
differentiation (AD) in Julia, and is the next-generation AD system for the
Flux differentiable programming framework.")
(license license:expat)))