-
### System Info
```shell
I used the official optimum neuron AMI for Trainum.
https://aws.amazon.com/marketplace/pp/prodview-gr3e6yiscria2
```
### Who can help?
_No response_
### Information
…
-
Build StableHLO programs rather than XLA programs, ~~then we can provide support for anything that accepts StableHLO, including XLA and IREE~~ removing our dependency on XLA. This may require rewritng…
-
## 🐛 Bug
Not sure if it is a bug or expected behavior, when doing `module.to(xm.xla_device())` it will create new parameter tensors instead of modify the tensor in place.
## To Reproduce
Runn…
-
## Fix the Op info test for `uniform .. var_mean`
1. Find the lines 333 to 337 of [test_ops.py](test/test_ops.py) and remove
`uniform` .. `var_mean` from `skip_list`
2. Run op_info test with `pyte…
qihqi updated
1 month ago
-
### Description
(Sorry if this is not the right place for this issue.)
I've recently got access to France's ADASTRA machine AMD-based HPC calculator.
It is not connected to the internet, so to…
-
When I am trying to use the example given to train the vicuna 7B model on colab. I am getting the following error.
```
2024-08-25 17:30:20.138324: E external/local_xla/xla/stream_executor/cuda/cuda…
-
### Description
I ran the Get Started code on the [Apple Accelerated JAX training on Mac page](https://developer.apple.com/metal/jax/), namely:
```python
python3 -m venv ~/jax-metal
source ~/jax…
-
I ran a few test jobs based on the recent [llama2-7B fine-tuning blog](https://www.philschmid.de/fine-tune-llama-7b-trainium#3-fine-tune-llama-on-aws-trainium-using-the-neurontrainer) using the latest…
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
Yes
### Source
source
### TensorFlow version
tf 2.17
### Custom code
Yes
### OS platform and distribution
Linux U…
x0w3n updated
1 month ago
-
### Description
The first time I encountered this error was run mult-node. Then after I run another code, single node also encountered this problem which was ok before. I think this error has s…