| [project] | |
| name = "flame" | |
| dynamic = ["version"] | |
| description = "A minimal training framework for scaling FLA models" | |
| readme = "README.md" | |
| authors = [ | |
| { name = "Songlin Yang", email = "yangsl66@mit.edu" }, | |
| { name = "Yu Zhang", email = "yzhang.cs@outlook.com" }, | |
| ] | |
| license = { file = "LICENSE" } | |
| classifiers = [ | |
| "Programming Language :: Python :: 3", | |
| "License :: OSI Approved :: MIT License", | |
| "Operating System :: OS Independent", | |
| "Topic :: Scientific/Engineering :: Artificial Intelligence", | |
| ] | |
| requires-python = ">=3.10" | |
| dependencies = [ | |
| 'torch>=2.5', | |
| 'torchdata', | |
| 'transformers>=4.45.0', | |
| 'triton>=3.0', | |
| 'datasets>=3.3.0', | |
| 'einops', | |
| 'ninja', | |
| 'wandb', | |
| 'tiktoken', | |
| 'tensorboard', | |
| ] | |
| [project.optional-dependencies] | |
| dev = ["pytest"] | |
| [project.urls] | |
| Homepage = "https://github.com/fla-org/flame" | |
| [build-system] | |
| requires = ["setuptools>=45", "wheel", "ninja", "torch"] | |
| [tool.isort] | |
| line_length = 127 | |
| multi_line_output = 3 | |