matrainer>=0.1.0a1
confluid>=0.3.1
recordstream>=0.1.0a2
loggair>=0.2.0
datasets>=2.0.0
pillow
tqdm

[dev]
build>=1.0.0
twine>=5.0.0
black<25.0.0,>=24.0.0
isort<6.0.0,>=5.13.0
flake8<8.0.0,>=6.0.0
mypy<2.0.0,>=1.0.0
pytest<9.0.0,>=7.0.0
pytest-cov<7.0.0,>=4.0.0

[fastai]
fastai>=2.7
fastcore<2
fastprogress<1.1
torch
timm>=1.0.0
recordstream[torch]>=0.1.0a2
matrainer[mlflow,tensorboard,torch]>=0.1.0a1

[huggingface]
transformers>=4.40
accelerate>=0.30

[jax]
jax>=0.4
flax>=0.8
optax>=0.2

[keras]
keras>=3.0
torch
recordstream[keras]>=0.1.0a2
matrainer[keras]>=0.1.0a1

[lightning]
torch
torchmetrics
pytorch-lightning
timm>=1.0.0
matrainer[lightning,mlflow,tensorboard]>=0.1.0a1

[mlx]

[mlx:sys_platform == "darwin" and platform_machine == "arm64"]
mlx>=0.15

[tensorflow]
keras>=3.0
tensorflow>=2.16
recordstream[keras]>=0.1.0a2
matrainer[keras]>=0.1.0a1

[torch]
torch
torchmetrics
timm>=1.0.0
recordstream[torch]>=0.1.0a2
matrainer[mlflow,tensorboard,torch]>=0.1.0a1
