dfs-lite

This repository contains a JAX implementation of the Deep Fusion System (DFS) Lite for large language models.