brainjepa 0.2.6

Brain-JEPA fMRI Foundation Model — inference in Rust (RLX)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

name: CI

on:
  push:
    branches: [main]
  pull_request:
    branches: [main]

env:
  CARGO_TERM_COLOR: always

jobs:
  rlx:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4
      - uses: dtolnay/rust-toolchain@stable
      - uses: Swatinem/rust-cache@v2
      - name: RLX unit tests
        run: cargo test --features rlx-engine --test config --test masks_indices --test rlx_classification --test rlx_graph_compile --test rlx_predictor_compile --test rlx_weights_load --test rlx_device_errors
      - name: CSV export test
        run: cargo test --features rlx-engine --test csv_export

  parity:
    runs-on: macos-latest
    steps:
      - uses: actions/checkout@v4
      - uses: dtolnay/rust-toolchain@stable
      - uses: Swatinem/rust-cache@v2
      - name: Download weights (if missing)
        run: |
          if [ ! -f data/brainjepa.safetensors ]; then
            cargo run --release --bin download_weights --features hf-download || true
          fi
      - name: RLX cross-backend parity
        run: bash scripts/parity.sh