PyPI page
Home page
Author:
None
Summary:
A framework for training and evaluating foundation models, using the MEDS ecosystem for data processing and PyTorch Lightning for training.
Latest version:
0.4.0
Required dependencies:
dotenv
|
filelock
|
hydra-core
|
ipykernel
|
lightning
|
matplotlib
|
meds
|
meds-torch-data
|
meds-transforms
|
numpy
|
polars
|
pyarrow
|
pytest
|
scikit-learn
|
torch
|
torchmetrics
|
transformers
|
wandb
Downloads last day:
13
Downloads last week:
628
Downloads last month:
1,351