Library Search
结果页

JAX 基础变换实战:grad, jit, vmap

在 元器件 / 技术栈 / 知识库 中找到 13 个匹配结果。

返回文档库
技术栈

技术栈

5 个
查看全部
JAX # JAX 自动微分 + JIT Hello World ## 目标 体验 JAX 的三大核心变换:`grad`(自动求导)、`jit`(编译加速)、`vmap`(自动向量化)。 ## 完整代码 ```python import jax import jax.numpy as jnp import time # ───────── 1. 基础:jax.numpy 与 NumPy 几乎一样… 3% PyTorch 1. 动态计算图相比静态图,性能上有损失吗?PyTorch 2.0 的 `torch.compile` 如何解决? 2. `optimizer.zero_grad()` 如果忘记调用会怎样? 3. `model.train()` 和 `model.eval()` 具体影响了哪些层的行为? 4. 为什么 `CrossEntropyLoss` 的输入不能经过 softmax? # PyTorch 进… 2% Fiber Fiber v3 引入了钩子系统,可以在请求生命周期的特定阶段执行逻辑: ```go app := fiber.New() // Hooks app.Hooks().OnListen(func(listenData fiber.ListenData) error { log.Printf("🚀 服务已启动在 %s", listenData.Addr) return nil }… 1% Sentinel ``` ┌──────────┐ │ CLOSED │ (正常) └─────┬────┘ │ 失败达到阈值 ┌─────┴────┐ │ OPEN │ (熔断) └─────┬────┘ │ 休眠时间到 ┌─────┴… 1% Amazon DynamoDB # Amazon DynamoDB 从零到实战:Serverless 后端 ## 1. 背景与概念 ### 1.1 DynamoDB 设计哲学 DynamoDB 继承自 Amazon 的 Dynamo 论文,核心理念: - **永远可写**(Always Writable):无主从切换 - **最终一致**:默认最终一致,可选强一致 - **单表设计**:不像 SQL 多表 JOIN,最佳… 1%
知识库

知识库

8 个
JAX 基础变换实战:grad, jit, vmap # JAX 自动微分 + JIT Hello World ## 目标 体验 JAX 的三大核心变换:`grad`(自动求导)、`jit`(编译加速)、`vmap`(自动向量化)。 ## 完整代码 ```python import jax import jax.numpy as jnp import time # ───────── 1. 基础:jax.numpy 与 NumPy 几乎一样… 3% JAX 纯函数式思维与神经网络 # JAX 入门教程:纯函数式思维与神经网络 ## 1. JAX 的哲学 JAX 不是另一个 TensorFlow 或 PyTorch。它的核心是**函数变换**:把普通的 Python/NumPy 函数,变换为可微分、可编译、可并行的版本。 ``` 你的函数 f ──► jax.grad(f) ──► 自动求导 ──► jax.jit(f) ──► XL… 3% 02-进阶实战-迁移学习与部署 # PyTorch 进阶实战 —— 迁移学习、混合精度与模型部署 ## 本章目标 - 使用预训练模型进行迁移学习(Fine-tuning) - 掌握混合精度训练(AMP)加速 - 模型导出:TorchScript → ONNX - 模型量化与推理优化 ## 1. 迁移学习(Fine-tuning) ### 1.1 使用 torchvision 预训练模型 ```python import… 2% 02-neural-network-mnist # ============================================================ # 5. 训练循环 # ============================================================ history = {"train_loss": [], "train_acc": [], "test_loss": [], "… 2% 02-进阶实战-测试-安全-部署 # Django 进阶实战 —— 测试、安全与部署 ## 本章目标 - 编写单元测试与集成测试 - 理解 Django 安全机制与最佳实践 - 使用 PostgreSQL + Gunicorn + Nginx 生产部署 - 掌握 Django 性能优化技巧 ## 1. 测试体系 ### 1.1 单元测试 ```python # blog/tests/test_models.py from… 1% Sentinel 流量治理入门:限流熔断实战 ``` ┌──────────┐ │ CLOSED │ (正常) └─────┬────┘ │ 失败达到阈值 ┌─────┴────┐ │ OPEN │ (熔断) └─────┬────┘ │ 休眠时间到 ┌─────┴… 1% Fiber 毕设实战 — Express 用户迁移到 Go 的最短路径 Fiber v3 引入了钩子系统,可以在请求生命周期的特定阶段执行逻辑: ```go app := fiber.New() // Hooks app.Hooks().OnListen(func(listenData fiber.ListenData) error { log.Printf("🚀 服务已启动在 %s", listenData.Addr) return nil }… 1% 01-tensor-basics # 回传到 CPU result_cpu = result.cpu() print(f"✅ 回传 CPU: {result_cpu.device}") # 直接创建在 GPU 上 gpu_tensor = torch.ones(4, 4, device="cuda") # =============================================… 1%