jaxKAN Documentation

Welcome to jaxKAN, a JAX implementation of Kolmogorov-Arnold Networks.

This documentation provides the information necessary to help you get started with jaxKAN. The About page offers further information about the package, while the Installation page details how to set it up. A complete API Reference is available for exploring the functionality of jaxKAN, and Tutorials include ready-to-run example code to guide you through practical applications. For developers interested in contributing, the Contributing page outlines how to get involved, and contributions are highly encouraged. Finally, for researchers or enthusiasts who have found jaxKAN useful in their projects, the Citing page provides instructions on how to properly reference the package in their work.

The source code for jaxKAN can be found in the jaxKAN GitHub Repository.

Contents:

• About
• Installation
• API Reference
• Tutorials
• Contributing
• Citing