tsai

This version has 7 feature flags, 1 of them enabled by default.

default

• backend-ndarray (default)

backend-ndarray (default)

• tsai_analysis/backend-ndarray (default)
• tsai_core/backend-ndarray (default)
• tsai_data/backend-ndarray (default)
• tsai_explain/backend-ndarray (default)
• tsai_models/backend-ndarray (default)
• tsai_train/backend-ndarray (default)
• tsai_transforms/backend-ndarray (default)

backend-mlx

• burn-mlx

backend-tch

• tsai_analysis/backend-tch
• tsai_core/backend-tch
• tsai_data/backend-tch
• tsai_explain/backend-tch
• tsai_models/backend-tch
• tsai_train/backend-tch
• tsai_transforms/backend-tch

backend-wgpu

• burn-wgpu
• tsai_analysis/backend-wgpu
• tsai_core/backend-wgpu
• tsai_data/backend-wgpu
• tsai_explain/backend-wgpu
• tsai_models/backend-wgpu
• tsai_train/backend-wgpu
• tsai_transforms/backend-wgpu

burn-mlx

• dep:burn-mlx

burn-wgpu

• dep:burn-wgpu

wandb

• tsai_train/wandb