FedJAX resembles the pseudo-code used to describe novel algorithms in academic papers.