File tree
40 files changed
+533
-751
lines changed- docs_nnx
- api_reference/flax.nnx
- guides
- examples/nnx_toy_examples
- flax/nnx
- bridge
- nn
- training
- transforms
- tests/nnx
- nn
40 files changed
+533
-751
lines changedLines changed: 0 additions & 9 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
4 | 4 |
| |
5 | 5 |
| |
6 | 6 |
| |
7 |
| - | |
8 |
| - | |
9 |
| - | |
10 |
| - | |
11 |
| - | |
12 |
| - | |
13 |
| - | |
14 |
| - | |
15 |
| - | |
16 | 7 |
| |
17 | 8 |
| |
18 | 9 |
| |
|
Lines changed: 0 additions & 2 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
18 | 18 |
| |
19 | 19 |
| |
20 | 20 |
| |
21 |
| - | |
22 |
| - | |
23 | 21 |
| |
24 | 22 |
| |
25 | 23 |
| |
|
Lines changed: 1 addition & 1 deletion
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
131 | 131 |
| |
132 | 132 |
| |
133 | 133 |
| |
134 |
| - | |
| 134 | + | |
135 | 135 |
| |
136 | 136 |
| |
137 | 137 |
| |
|
Lines changed: 1 addition & 1 deletion
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
82 | 82 |
| |
83 | 83 |
| |
84 | 84 |
| |
85 |
| - | |
| 85 | + | |
86 | 86 |
| |
87 | 87 |
| |
88 | 88 |
| |
|
Lines changed: 1 addition & 1 deletion
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
415 | 415 |
| |
416 | 416 |
| |
417 | 417 |
| |
418 |
| - | |
| 418 | + | |
419 | 419 |
| |
420 | 420 |
| |
421 | 421 |
| |
|
Lines changed: 1 addition & 1 deletion
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
652 | 652 |
| |
653 | 653 |
| |
654 | 654 |
| |
655 |
| - | |
| 655 | + | |
656 | 656 |
| |
657 | 657 |
| |
658 | 658 |
| |
|
Lines changed: 1 addition & 1 deletion
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
295 | 295 |
| |
296 | 296 |
| |
297 | 297 |
| |
298 |
| - | |
| 298 | + | |
299 | 299 |
| |
300 | 300 |
| |
301 | 301 |
| |
|
Lines changed: 2 additions & 2 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
199 | 199 |
| |
200 | 200 |
| |
201 | 201 |
| |
202 |
| - | |
| 202 | + | |
203 | 203 |
| |
204 |
| - | |
| 204 | + | |
205 | 205 |
| |
206 | 206 |
| |
207 | 207 |
| |
|
Lines changed: 8 additions & 8 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
56 | 56 |
| |
57 | 57 |
| |
58 | 58 |
| |
59 |
| - | |
60 |
| - | |
| 59 | + | |
| 60 | + | |
61 | 61 |
| |
62 | 62 |
| |
63 | 63 |
| |
| |||
74 | 74 |
| |
75 | 75 |
| |
76 | 76 |
| |
77 |
| - | |
78 |
| - | |
| 77 | + | |
| 78 | + | |
79 | 79 |
| |
80 | 80 |
| |
81 | 81 |
| |
| |||
102 | 102 |
| |
103 | 103 |
| |
104 | 104 |
| |
105 |
| - | |
106 |
| - | |
| 105 | + | |
| 106 | + | |
107 | 107 |
| |
108 | 108 |
| |
109 | 109 |
| |
| |||
118 | 118 |
| |
119 | 119 |
| |
120 | 120 |
| |
121 |
| - | |
122 |
| - | |
| 121 | + | |
| 122 | + | |
123 | 123 |
| |
124 | 124 |
| |
125 | 125 |
| |
|
Lines changed: 1 addition & 1 deletion
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
182 | 182 |
| |
183 | 183 |
| |
184 | 184 |
| |
185 |
| - | |
| 185 | + | |
186 | 186 |
| |
187 | 187 |
| |
188 | 188 |
| |
|
Lines changed: 7 additions & 7 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
276 | 276 |
| |
277 | 277 |
| |
278 | 278 |
| |
279 |
| - | |
| 279 | + | |
280 | 280 |
| |
281 | 281 |
| |
282 | 282 |
| |
| |||
337 | 337 |
| |
338 | 338 |
| |
339 | 339 |
| |
340 |
| - | |
| 340 | + | |
341 | 341 |
| |
342 | 342 |
| |
343 | 343 |
| |
| |||
706 | 706 |
| |
707 | 707 |
| |
708 | 708 |
| |
709 |
| - | |
| 709 | + | |
710 | 710 |
| |
711 | 711 |
| |
712 | 712 |
| |
| |||
784 | 784 |
| |
785 | 785 |
| |
786 | 786 |
| |
787 |
| - | |
| 787 | + | |
788 | 788 |
| |
789 | 789 |
| |
790 | 790 |
| |
| |||
850 | 850 |
| |
851 | 851 |
| |
852 | 852 |
| |
853 |
| - | |
| 853 | + | |
854 | 854 |
| |
855 | 855 |
| |
856 | 856 |
| |
| |||
1001 | 1001 |
| |
1002 | 1002 |
| |
1003 | 1003 |
| |
1004 |
| - | |
| 1004 | + | |
1005 | 1005 |
| |
1006 | 1006 |
| |
1007 | 1007 |
| |
| |||
1076 | 1076 |
| |
1077 | 1077 |
| |
1078 | 1078 |
| |
1079 |
| - | |
| 1079 | + | |
1080 | 1080 |
| |
1081 | 1081 |
| |
1082 | 1082 |
| |
|
Lines changed: 7 additions & 7 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
136 | 136 |
| |
137 | 137 |
| |
138 | 138 |
| |
139 |
| - | |
| 139 | + | |
140 | 140 |
| |
141 | 141 |
| |
142 | 142 |
| |
| |||
168 | 168 |
| |
169 | 169 |
| |
170 | 170 |
| |
171 |
| - | |
| 171 | + | |
172 | 172 |
| |
173 | 173 |
| |
174 | 174 |
| |
| |||
338 | 338 |
| |
339 | 339 |
| |
340 | 340 |
| |
341 |
| - | |
| 341 | + | |
342 | 342 |
| |
343 | 343 |
| |
344 | 344 |
| |
| |||
383 | 383 |
| |
384 | 384 |
| |
385 | 385 |
| |
386 |
| - | |
| 386 | + | |
387 | 387 |
| |
388 | 388 |
| |
389 | 389 |
| |
| |||
420 | 420 |
| |
421 | 421 |
| |
422 | 422 |
| |
423 |
| - | |
| 423 | + | |
424 | 424 |
| |
425 | 425 |
| |
426 | 426 |
| |
| |||
545 | 545 |
| |
546 | 546 |
| |
547 | 547 |
| |
548 |
| - | |
| 548 | + | |
549 | 549 |
| |
550 | 550 |
| |
551 | 551 |
| |
| |||
590 | 590 |
| |
591 | 591 |
| |
592 | 592 |
| |
593 |
| - | |
| 593 | + | |
594 | 594 |
| |
595 | 595 |
| |
596 | 596 |
| |
|
Lines changed: 1 addition & 4 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
37 | 37 |
| |
38 | 38 |
| |
39 | 39 |
| |
40 |
| - | |
41 |
| - | |
42 |
| - | |
43 |
| - | |
| 40 | + |
Lines changed: 24 additions & 15 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
71 | 71 |
| |
72 | 72 |
| |
73 | 73 |
| |
74 |
| - | |
| 74 | + | |
75 | 75 |
| |
76 | 76 |
| |
77 | 77 |
| |
78 | 78 |
| |
79 |
| - | |
| 79 | + | |
80 | 80 |
| |
81 | 81 |
| |
82 | 82 |
| |
83 | 83 |
| |
84 | 84 |
| |
85 | 85 |
| |
86 |
| - | |
87 |
| - | |
| 86 | + | |
| 87 | + | |
| 88 | + | |
| 89 | + | |
| 90 | + | |
88 | 91 |
| |
89 | 92 |
| |
90 |
| - | |
| 93 | + | |
91 | 94 |
| |
92 |
| - | |
| 95 | + | |
93 | 96 |
| |
94 | 97 |
| |
95 |
| - | |
| 98 | + | |
96 | 99 |
| |
97 |
| - | |
98 |
| - | |
99 |
| - | |
| 100 | + | |
| 101 | + | |
| 102 | + | |
| 103 | + | |
| 104 | + | |
| 105 | + | |
| 106 | + | |
| 107 | + | |
| 108 | + | |
100 | 109 |
| |
101 | 110 |
| |
102 | 111 |
| |
103 | 112 |
| |
104 | 113 |
| |
105 | 114 |
| |
106 |
| - | |
| 115 | + | |
107 | 116 |
| |
108 | 117 |
| |
109 | 118 |
| |
110 | 119 |
| |
111 |
| - | |
| 120 | + | |
112 | 121 |
| |
113 |
| - | |
| 122 | + | |
114 | 123 |
| |
115 | 124 |
| |
116 |
| - | |
| 125 | + | |
117 | 126 |
| |
118 | 127 |
| |
119 | 128 |
| |
| |||
137 | 146 |
| |
138 | 147 |
| |
139 | 148 |
| |
140 |
| - | |
| 149 | + | |
141 | 150 |
| |
142 | 151 |
| |
143 | 152 |
| |
|
Lines changed: 1 addition & 1 deletion
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
43 | 43 |
| |
44 | 44 |
| |
45 | 45 |
| |
| 46 | + | |
46 | 47 |
| |
47 | 48 |
| |
48 | 49 |
| |
| |||
161 | 162 |
| |
162 | 163 |
| |
163 | 164 |
| |
164 |
| - | |
165 | 165 |
| |
166 | 166 |
| |
167 | 167 |
| |
|
0 commit comments