Metadata-Version: 2.4
Name: mlrunx-integrations
Version: 0.1.1
Summary: MLRunX integrations for popular ML frameworks
Author: MLRunX Team
License: MIT
Keywords: huggingface,hydra,ml,mlops,optuna,pytorch-lightning
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.10
Requires-Dist: mlrunx>=0.1.1
Provides-Extra: all
Requires-Dist: datasets>=2.14; extra == 'all'
Requires-Dist: hydra-core>=1.3; extra == 'all'
Requires-Dist: optuna>=3.0; extra == 'all'
Requires-Dist: pytorch-lightning>=2.0; extra == 'all'
Requires-Dist: transformers>=4.35; extra == 'all'
Provides-Extra: dev
Requires-Dist: pytest-asyncio>=0.24; extra == 'dev'
Requires-Dist: pytest>=8.0; extra == 'dev'
Provides-Extra: huggingface
Requires-Dist: datasets>=2.14; extra == 'huggingface'
Requires-Dist: transformers>=4.35; extra == 'huggingface'
Provides-Extra: hydra
Requires-Dist: hydra-core>=1.3; extra == 'hydra'
Provides-Extra: lightning
Requires-Dist: pytorch-lightning>=2.0; extra == 'lightning'
Provides-Extra: optuna
Requires-Dist: optuna>=3.0; extra == 'optuna'
Description-Content-Type: text/markdown

# MLRunX Integrations

Framework integrations for MLRunX.

## Supported Frameworks

- **PyTorch Lightning** - `MLRunXLogger`
- **HuggingFace Transformers** - `MLRunXCallback`
- **Optuna** - `MLRunXOptunaCallback`
- **Hydra** - `MLRunXHydraCallback`

## Installation

```bash
# Install with all integrations
pip install "mlrunx-integrations[all]"

# Or install specific integrations
pip install "mlrunx-integrations[lightning]"
pip install "mlrunx-integrations[transformers]"
pip install "mlrunx-integrations[optuna]"
pip install "mlrunx-integrations[hydra]"
```

## Usage

### PyTorch Lightning

```python
from mlrunx_integrations import MLRunXLogger
from pytorch_lightning import Trainer

trainer = Trainer(logger=MLRunXLogger(project="my-project"))
trainer.fit(model)
```

### HuggingFace Transformers

```python
from mlrunx_integrations import MLRunXCallback
from transformers import Trainer

trainer = Trainer(...)
trainer.add_callback(MLRunXCallback(project="my-project"))
trainer.train()
```

### Optuna

```python
from mlrunx_integrations import MLRunXOptunaCallback
import optuna

study = optuna.create_study()
study.optimize(
    objective,
    callbacks=[MLRunXOptunaCallback(project="my-project")]
)
```
