Repository navigation
Commit bc36df3
README + tests: align with jax.experimental.jet's factorial_scaled flag
Recent jax.experimental.jet (PR jax-ml/jax#29676, merged 2025-07) accepts
factorial_scaled=False, which returns the same Taylor coefficients as
jet_array. Update the convention discussion in README accordingly, and
switch the comparison tests to use factorial_scaled=False directly
instead of post-hoc dividing standard-jet output by k!. The tests now
assert direct coefficient equality rather than equality-after-rescaling.
Also clarify the effective_order caveats. The hint applies to three
primitive categories with different behavior:
- linear / zero-derivative (add, sub, neg, broadcast, …): no effect,
already O(K) elementwise
- nonlinear without aware rule (sin, cos, sqrt, erf, …): silently ignored
- nonlinear with aware rule (exp, log1p, pow, tanh, …): convolution scan
short-circuits, entries above effective_order unspecified
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>1 parent 56375ba commit bc36df3
2 files changed
Lines changed: 45 additions & 36 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
80 | 80 | | |
81 | 81 | | |
82 | 82 | | |
83 | | - | |
84 | | - | |
85 | | - | |
86 | | - | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
| 87 | + | |
| 88 | + | |
| 89 | + | |
| 90 | + | |
| 91 | + | |
87 | 92 | | |
88 | 93 | | |
89 | 94 | | |
| |||
227 | 232 | | |
228 | 233 | | |
229 | 234 | | |
230 | | - | |
231 | | - | |
232 | | - | |
233 | | - | |
234 | | - | |
235 | | - | |
236 | | - | |
| 235 | + | |
| 236 | + | |
| 237 | + | |
| 238 | + | |
| 239 | + | |
| 240 | + | |
| 241 | + | |
| 242 | + | |
| 243 | + | |
| 244 | + | |
| 245 | + | |
| 246 | + | |
| 247 | + | |
| 248 | + | |
| 249 | + | |
| 250 | + | |
| 251 | + | |
| 252 | + | |
| 253 | + | |
| 254 | + | |
| 255 | + | |
237 | 256 | | |
238 | 257 | | |
239 | 258 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | 1 | | |
2 | 2 | | |
3 | | - | |
4 | | - | |
5 | | - | |
6 | | - | |
7 | | - | |
8 | | - | |
9 | | - | |
10 | | - | |
11 | | - | |
12 | | - | |
13 | | - | |
14 | | - | |
15 | | - | |
16 | | - | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
17 | 10 | | |
18 | 11 | | |
19 | 12 | | |
| |||
35 | 28 | | |
36 | 29 | | |
37 | 30 | | |
38 | | - | |
39 | | - | |
40 | | - | |
41 | | - | |
42 | | - | |
43 | 31 | | |
44 | 32 | | |
45 | 33 | | |
46 | 34 | | |
47 | 35 | | |
48 | 36 | | |
49 | 37 | | |
50 | | - | |
51 | | - | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
52 | 41 | | |
53 | 42 | | |
54 | 43 | | |
55 | 44 | | |
56 | 45 | | |
57 | | - | |
| 46 | + | |
58 | 47 | | |
59 | 48 | | |
60 | 49 | | |
| |||
92 | 81 | | |
93 | 82 | | |
94 | 83 | | |
95 | | - | |
96 | | - | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
97 | 87 | | |
98 | 88 | | |
99 | | - | |
| 89 | + | |
100 | 90 | | |
101 | 91 | | |
102 | 92 | | |
| |||
0 commit comments