• Andrey Filippov's avatar
    CLAUDE: export L2 (Layer2Net step) to TorchScript for native LibTorch · 7a351e9e
    Andrey Filippov authored
    Piece 1 of the native-JNA DNN path (no Python server). Adds export_l2_torchscript.py:
    wraps a trained Layer2Net's cell+head into a single-step L2Step module
    forward(x,h)->(h_new,det,vel) — exactly infer_server's per-scene recurrence
    (h=cell(x,h); det,vel=decode(h)) — so the C++ side just carries h and calls it
    per scene. Size-agnostic (circular pad + 1x1 head), runs on the full field.
    
    Validated: scripted==eager exact (0.0); C++ LibTorch (libtorch_probe/l2_probe)
    loads it on Blackwell CUDA and replays the recurrence with hidden-state match
    9.5e-7. Required disabling the TorchScript JIT fuser (nvrtc element-wise fusion
    fails on Blackwell -arch; production wants no runtime nvrtc) — folds into the
    native lib startup in piece 2.
    Co-Authored-By: 's avatarClaude Opus 4.8 (1M context) <noreply@anthropic.com>
    7a351e9e
Name
Last commit
Last update
.gitignore Loading commit data...
DESIGN.md Loading commit data...
DESIGN_v2_mf_hough.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
baselines.py Loading commit data...
build_combo3.py Loading commit data...
clean_eval_fwhm.py Loading commit data...
compare_dnn_truth.py Loading commit data...
dense_check.py Loading commit data...
diag_clean.py Loading commit data...
eval_mfs.py Loading commit data...
export_l2_torchscript.py Loading commit data...
export_onnx.py Loading commit data...
export_refine.py Loading commit data...
export_torchscript.py Loading commit data...
extract_B.py Loading commit data...
gap_eval.py Loading commit data...
gen_synth_cuas.py Loading commit data...
ghost_probe.py Loading commit data...
infer_server.py Loading commit data...
l1_samples.py Loading commit data...
l2_fp_analysis.py Loading commit data...
layer2.py Loading commit data...
layer2_data.py Loading commit data...
layer2_eval.py Loading commit data...
layer2_gapcheck.py Loading commit data...
layer2_train.py Loading commit data...
layer2_train_A.py Loading commit data...
layer2_train_P.py Loading commit data...
layer2p.py Loading commit data...
make_testvec.py Loading commit data...
model.py Loading commit data...
nettest.py Loading commit data...
partial_votes.py Loading commit data...
run_infer_local.sh Loading commit data...
run_infer_server.sh Loading commit data...
run_l2A.sh Loading commit data...
run_l2_chain.sh Loading commit data...
run_l2_dense.sh Loading commit data...
run_l2_seq.sh Loading commit data...
shake.py Loading commit data...
stage2.py Loading commit data...
stage2_eval3.py Loading commit data...
synth.py Loading commit data...
test_infer_client.py Loading commit data...
train.py Loading commit data...
velocity_bias.py Loading commit data...
viz_results.py Loading commit data...
viz_trainingdata.py Loading commit data...
vote_1d.py Loading commit data...