-
### 🐛 Describe the bug
The following code is an example provided in the documentation of vmap:
```python
import torch
# Setup
N = 5
f = lambda x: x ** 2
x = torch.randn(N, requires_grad=True)…
-
(base) aqa6122@e5-cse-135-10:/scratch/aqa6122/lib/rational_kat_cu$ pip install -e .
Obtaining file:///scratch/aqa6122/lib/rational_kat_cu
Preparing metadata (setup.py) ... done
Installing collect…
-
Traceback (most recent call last):
File "main.py", line 136, in
main()
File "main.py", line 132, in main
atari_learn(env, task.env_id, num_timesteps=task.max_timesteps, double_dqn=dou…
-
https://gist.github.com/zou3519/b987e00a82c7e184b8896a5df7b0bfa9
Benchmarking two cases:
1. torch.ops.mylib.foo operator that has an Autograd key that takes unboxed inputs but a CPU key that boxes…
-
Wondering if anyone on the Bend team or in the community has started thinking about a neural network library that runs on Bend or more generally on the HVM. Is the Bend API mature enough for that righ…
-
### 🚀 The feature, motivation and pitch
I am trying to implement eager mode of PT2E quantization on CPU. Currently, the PT2E quantization on CPU is lowered to Inductor by `torch.compile`. The current…
-
Getting an internal assert failure:
```
E1120 16:27:42.446000 166428 site-packages/torch/_subclasses/fake_tensor.py:2301] [0/1_1] failed while attempting to run meta for aten._upsample_bilinear2d_aa…
-
I am trying to fine tune a model using FastLanguageModel, and I only want to train the last few layers. When I am passing the target module/layer, it is throwing error saying that it only allows "acce…
-
As you know most, if not all, user model functions are written in python, so I wanted to see if one can just hook in autograd for such cases. Caveat: "Autograd doesn't support assignment into arrays"…
-
Would be nice in the README to point Python folks to [autograd](https://github.com/hips/autograd), and I'm sure they'd be pumped to point Lua folks here.