Files
rustytorch/crates/models/rtx-vision
Omar SobhandClaude Sonnet 4.6 f487196367
CI / Format Check (push) Failing after 11s
CI / Test (macos-latest) (push) Has been skipped
CI / Test (ubuntu-latest) (push) Has been skipped
CI / Python Bindings (maturin) (macos-latest) (push) Has been skipped
CI / Python Bindings (maturin) (ubuntu-latest) (push) Has been skipped
CI / WASM Build + Size Check (push) Has been skipped
Documentation / Build User Guide (push) Successful in 9s
CI / Build CPU-Only (Explicit) (push) Failing after 1m10s
CI / CI Success (push) Failing after 0s
Documentation / Build API Documentation (push) Failing after 40s
Performance Benchmarks / Run Benchmarks (push) Successful in 7m53s
CI / Build (macos-latest) (push) Failing after 30s
CI / Build (ubuntu-latest) (push) Failing after 48s
CI / Distributed Training Tests (push) Has been skipped
CI / Clippy Check (push) Failing after 52s
feat(batch27): JEPA ViT bridge, WebDataset shard reading, training loop
Gap 2 — rtx-vision ViT bridge (jepa_vision_bridge.rs, 8 tests):
- ViT::forward_features(): patch reps without classification head
- ViT::encode_patch_indices(): shape-correct placeholder for GPU dispatch
- RtxVisionJepaEncoder implementing JepaEncoder (vision-bridge feature)
- From<&ViTConfig> for JepaViTConfig config conversion
- rtx-vision added as optional dep; vision-bridge feature gate

Gap 3 — WebDataset tar-shard reading (jepa_data.rs, +12 tests, 47 total):
- parse_tar_bytes(): pure stdlib tar parser (512-byte block format)
- read_webdataset_shard(): file reader with ShardLoadStats timing
- WebDatasetRecord: key, image_bytes, label, extension
- ShuffleBuffer: fixed-capacity reservoir sampling via LCG PRNG
- JepaDataPipeline::from_filesystem(): validates paths, loads shards, builds pipeline

Gap 5 — Training loop runner (jepa_runner.rs + examples/jepa_train.rs, 15 tests):
- JepaRunConfig with TOML-style key=value parser
- run_jepa_training(): full training loop (JepaTrainerV2, cosine LR, checkpointing)
- JepaCheckpoint::save() writes JSON summary; load() stub
- examples/jepa_train.rs: --config/--size/--steps/--dry-run CLI flags

Co-Authored-By: Claude Sonnet 4.6 <[email protected]>
2026-06-27 15:23:55 +00:00
..
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00
2026-03-04 00:08:42 +00:00