- 前言
- 函数, 多元函数
- 求导，偏导
- 求导的程序实现
- 手动实现
- 有限差分
- 符号推导
- 自动微分
- 前向模式
- 反向模
- ...
- 语法基础
- jnp, random, 精度
- grad
- jit
- vmap/pmap
- 用JAX的方式思考
- 名词表
- 性能问题
- 不同库之间比较
-
Notifications
You must be signed in to change notification settings - Fork 0
Roy-Kid/jax-in-practice
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published