scaling-book 是一个全面的指南和资源库,专注于使用 JAX 扩展机器学习模型。它提供了详细的文档、实用的代码示例,涵盖了从基础到高级的扩展方法,并与现代机器学习框架和工具集成。该项目还支持社区驱动的更新和贡献。