giles 4c2e716558 Make JAX the primary fused-pipeline path for CPU/GPU parity ...

JAX via XLA produces identical output on CPU and GPU. Previously
CUDA hand-written kernels were preferred on GPU, causing visual
differences vs the JAX CPU fallback. Now JAX is always used first,
with legacy CuPy/GPUFrame as fallback only when JAX is unavailable.

Also adds comprehensive CLAUDE.md for the monorepo.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

2026-02-25 19:31:53 +00:00

..

effects

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00

primitive_libs

Make JAX the primary fused-pipeline path for CPU/GPU parity

2026-02-25 19:31:53 +00:00

__init__.py

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00

derived.sexp

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00

interpreter.py

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00

parser.py

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00

primitives.py

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00

test_interpreter.py

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00

wgsl_compiler.py

Import L1 (celery) as l1/

2026-02-24 23:07:19 +00:00