jax-development

【开发与工程】Write, debug, review, profile, or shard JAX numerical code. Use when the hard part is JAX tracing, autodiff, control flow, PRNGs, compilation, array placement, or runtime performance. Do not impose JAX on a NumPy-only or unrelated GPU task.

独立 GitHub 快照 · 自动分类 · 以原作者文档为准