Dataset from https://huggingface.co/datasets/Helsinki-NLP/opus_books/viewer/en-hu , there are a lot of other translation language datasets, you can get another you wish there.
pip install torch datasets jax
Change configure file in folder config
to your wish
python train.py --config_path ../config/config.json