NyayaLLM: Indian Criminal Law AI

Navigating the recently overhauled Indian criminal justice system, which introduces the Bharatiya Nyaya Sanhita (BNS), Bharatiya Nagarik Suraksha Sanhita (BNSS), and Bharatiya Sakshya Adhiniyam (BSA), is a massive challenge for both legal professionals and everyday citizens. NyayaLLM solves this by providing a highly accurate, domain-specific AI assistant trained exclusively on these new legal frameworks. We leveraged the immense compute power of AMD MI300X instances to fine-tune a Llama 3.1 model, ensuring rapid iteration and high throughput during the training phase. By utilizing ROCm and Hugging Face libraries, we adapted the base model to understand complex legal jargon, procedural nuances, and the specific relationships between the old IPC/CrPC sections and the newly implemented BNS/BNSS codes. The resulting application allows users to query specific legal scenarios, understand compliance requirements, and draft preliminary legal summaries with significantly reduced hallucinations compared to standard, general-purpose models. Our ultimate goal with NyayaLLM is to democratize legal knowledge, increase productivity for law firms dealing with the transition, and ensure that the complexities of the new Indian criminal laws do not become a barrier to justice.