from setuptools import setup

setup(
    name='dp-audit',
    version='1.0',
    description='JAX DP auditing',
    url='',
    author='',
    packages=['src', 'audit_one_run'],
    install_requires=[
        'torch==2.6.0',
        'torchvision==0.21.0',
        'transformers==4.48.2',
    ],
)