arXiv:2502.03047v2 通知类型: replace-cross 摘要：遗传编程是一种受进化启发的优化算法，能够自动演化可解释的计算机程序的结构。遗传编程中的适应度评估面临着高计算要求的限制，限制了其在难题上的性能。因此，目前没有适用于广泛任务的高效遗传编程框架。为了解决这个问题，我们开发了Kozax，一个能够演化任意问题的符号表达式的遗传编程框架。我们使用JAX实现了Kozax，这是一个高性能和可扩展的机器学习框架，这使得适应度评估能够有效地扩展到大型种群或数据集中的GPU上。此外，Kozax提供了连续优化、自定义运算符定义和同时演化多个树的功能。我们展示了Kozax在发现自然定律方程、恢复隐藏动态变量的方程、演化控制策略和优化目标函数方面的成功应用。总的来说，Kozax提供了一个通用、快速和可扩展的库，用于在科学计算领域优化白盒解决方案。