-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathdune-project
123 lines (110 loc) · 2.47 KB
/
dune-project
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
(lang dune 3.11)
(using ctypes 0.3)
(name ocannl)
(version 0.5.1)
(generate_opam_files true)
(source
(github lukstafi/ocannl))
(authors "Lukasz Stafiniak")
(maintainers "Lukasz Stafiniak <[email protected]>")
(license "BSD-2-Clause")
(documentation https://github.com/lukstafi/ocannl/blob/master/README.md)
; We give up on npy / ocannl_npy for now.
(package
(name arrayjit)
(synopsis
"An array language compiler with multiple backends (CPU, CUDA), staged compilation")
(description
"The optimizing compiler sub-package of OCANNL. Use neural_nets_lib instead to also get: nice syntax, shape inference, backpropagation, optimizers.")
(depends
(ocaml
(>= 5.2.0))
dune
(base
(>= v0.17.0))
(ctypes
(>= 0.23))
(ctypes-foreign
(>= 0.23))
; opam 2.2.0 has with-dev-setup. Is it supported, what's the syntax?
; (ocamlformat
; (>= 0.27.0)
; :with-dev-setup)
(printbox
(>= 0.12))
(printbox-text
(>= 0.12))
stdio
sexplib
num
(saturn_lockfree
(>= 0.5.0))
ppxlib
ppx_compare
ppx_hash
ppx_here
ppx_sexp_conv
ppx_string
ppx_variants_conv
ppx_expect
(ppx_minidebug
(>= 2.0.3)))
(depopts
(cudajit
(>= 0.6.1))
(gccjit
(>= 0.3.2)))
(conflicts
(cudajit
(< 0.6.1))
(gccjit
(< 0.3.2)))
(tags
(deeplearning array jit CUDA)))
(package
(name neural_nets_lib)
(synopsis
"A from-scratch Deep Learning framework with an optimizing compiler, shape inference, concise syntax")
(description
"OCaml Compiles Algorithms for Neural Networks Learning is a compiled Deep Learning framework that puts emphasis on low-level backends (like tinygrad), shape inference, concise notation (ab)using PPX.")
(depends
(ocaml
(>= 5.2.0))
dune
(base
(>= v0.17.0))
; opam 2.2.0 has with-dev-setup. Is it supported, what's the syntax?
; (ocamlformat
; (>= 0.26.2)
; :with-dev-setup)
; Here we mean (arrayjit (= 0.4.1)), but we must be compatible with arrayjit.dev during installation.
(arrayjit
(>= 0.5.1))
(printbox
(>= 0.12))
(printbox-text
(>= 0.12))
(printbox-ext-plot
(>= 0.12))
(angstrom
(>= 0.15))
stdio
sexplib
num
time_now
ppxlib
ppx_compare
ppx_fields_conv
ppx_hash
ppx_here
ppx_sexp_conv
ppx_string
ppx_variants_conv
ppx_expect
(ppx_minidebug
(>= 2.0.3))
(odoc :with-doc)
(md2mld :with-doc))
(tags
(deeplearning tensor backprop jit gccjit CUDA)))
; See the complete stanza docs at https://dune.readthedocs.io/en/stable/dune-files.html#dune-project