[project]
name = "i-jepa"
version = "0.1.0"
description = "Recreation of I-JEPA paper from scratch in PyTorch and JAX"
readme = "README.md"
requires-python = ">=3.13"
dependencies = [
    "datasets[vision]>=4.5.0",
    "jax>=0.9.0",
    "jax2onnx>=0.11.2",
    "numpy>=2.4.2",
    "pillow>=12.1.0",
    "requests>=2.32.5",
    "torch>=2.10.0",
    "torchvision>=0.25.0",
    "transformers>=5.0.0",
]