Lexa AI is a multilingual machine translation model, built as an NLP course project. It implements a small encoder-decoder transformer from scratch in PyTorch, with no pretrained weights, to translate English into Hindi and Telugu.
A single encoder handles English input while the decoder is shared across both target languages, switched by a beginning-of-sentence token (<s-hi> or <s-te>). The model was trained on the Opus100 en-hi and en-te subsets using a custom BERT-style tokenizer, and is exposed through a Streamlit interface with a provided Dockerfile.