## Multi AD Backend Support - [x] Add `backend` kwarg to `NeuroTabRegressor` / `NeuroTabClassifier` (`enzyme`, `zygote`, `mooncake`) - [x] Extend `device` kwarg to accept `cpu`, `gpu`, `reactant` - [x] Wire up Zygote extension - [x] Enzyme + Reactant path (`device=reactant`) - [x] Document device/backend compatibility ## TabReD Temporal Embeddings - [ ] Implement temporal embeddings per [Temporal embedding](https://arxiv.org/html/2502.20260v1) - [ ] Adapt constructor to support optional temporal embeddings - [ ] Ensure compatibility with existing embedding options (piecewise, linear, etc.) ## ModernNCA Core Model - [ ] Implement ModernNCA per [paper](https://openreview.net/pdf?id=JytL2MrlLT) - [ ] Benchmark against MLBenchmarks suite - [ ] Code + package documentation ## MLBenchmarks Datasets (TabReD) - [ ] Add ingestion recipes for the 3 TabReD datasets under `src/Datasets` - [ ] Add training scripts under `benchmarks/` - [ ] Benchmark against Python TabReD reference implementation
Multi AD Backend Support
backendkwarg toNeuroTabRegressor/NeuroTabClassifier(enzyme,zygote,mooncake)devicekwarg to acceptcpu,gpu,reactantdevice=reactant)TabReD Temporal Embeddings
ModernNCA Core Model
MLBenchmarks Datasets (TabReD)
src/Datasetsbenchmarks/