PyPI page
Home page
Author:
Payal Chandak
Summary:
A framework for training and evaluating foundation models, using the MEDS ecosystem for data processing and PyTorch Lightning for training.
Latest version:
0.7.0
Required dependencies:
filelock
|
hydra-core
|
hydra-joblib-launcher
|
ipykernel
|
lightning
|
matplotlib
|
meds
|
meds-torch-data
|
meds-transforms
|
numpy
|
polars
|
pyarrow
|
pytest
|
scikit-learn
|
torch
|
torchdata
|
torchmetrics
|
transformers
|
wandb
Downloads last day:
2
Downloads last week:
33
Downloads last month:
63
7-day smoothing