giles 07cae101ad Use JAX for fused pipeline fallback on CPU instead of GPUFrame path ...

When CUDA fused kernels aren't available, the fused-pipeline primitive
now uses JAX ops (jax_rotate, jax_scale, jax_shift_hue, etc.) instead
of falling back to one-by-one CuPy/GPUFrame operations. Legacy GPUFrame
path retained as last resort when JAX is also unavailable.

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

2026-02-25 15:35:13 +00:00

..

effects

Import L1 (celery) as l1/

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

primitive_libs

Use JAX for fused pipeline fallback on CPU instead of GPUFrame path

2026-02-25 15:35:13 +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