tabicl-model 2.1.1

TabICL transformer model — column embedding, row interaction, ICL learning, KV cache.
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
37
38
39
40

//! Port of `tabicl._model` — the TabICL transformer.
//!
//! Python reference: `/Users/Shared/tabicl/src/tabicl/_model/`.
//!
//! Module map (matches the Python file names, minus the `.py`):
//!
//! | this crate         | Python file                | LOC (Python) | status     |
//! |--------------------|----------------------------|--------------|------------|
//! | `ssmax`            | `_model/ssmax.py`          | 216          | spec done  |
//! | `rope`             | `_model/rope.py`           | 418          | spec       |
//! | `layers`           | `_model/layers.py`         | 792          | placeholder|
//! | `attention`        | `_model/attention.py`      | 298          | placeholder|
//! | `encoders`         | `_model/encoders.py`       | 399          | placeholder|
//! | `embedding`        | `_model/embedding.py`      | 890          | placeholder|
//! | `interaction`      | `_model/interaction.py`    | 273          | placeholder|
//! | `learning`         | `_model/learning.py`       | 829          | placeholder|
//! | `kv_cache`         | `_model/kv_cache.py`       | 362          | placeholder|
//! | `inference`        | `_model/inference.py`      | 1412         | placeholder|
//! | `inference_config` | `_model/inference_config.py`| 371         | spec       |
//! | `tabicl`           | `_model/tabicl.py`         | 899          | placeholder|

pub mod attention;
pub mod autodiff;
pub mod embedding;
pub mod encoders;
pub mod gpu_attention;
pub mod inference;
pub mod inference_config;
pub mod interaction;
pub mod kv_cache;
pub mod layers;
pub mod learning;
pub mod rope;
pub mod ssmax;
pub mod state_dict;
pub mod tabicl;

pub use inference_config::InferenceConfig;
pub use state_dict::{StateDict, StateDictError};
pub use tabicl::TabICL;