xla
Update Neuron initializations
#7952
Merged

Update Neuron initializations #7952

jeffhataws
jeffhataws Update setting of Neuron environments, NEURON_LIBRARY_PATH
3a02717c
jeffhataws Add some more defaults; lint fixes
9456c1d5
jeffhataws Allow torch-neuronx to override _internal/neuron.py
1afabb62
jeffhataws jeffhataws requested a review from will-cromar will-cromar 1 year ago
jeffhataws jeffhataws requested a review from aws-kingrj aws-kingrj 1 year ago
jeffhataws jeffhataws requested a review from zpcore zpcore 1 year ago
jeffhataws jeffhataws requested a review from JackCaoG JackCaoG 1 year ago
jeffhataws jeffhataws requested a review from amithrm amithrm 1 year ago
will-cromar
will-cromar commented on 2024-09-04
jeffhataws Move PJRT_DEVICE and NEURON_LIBRARY_PATH to other locations
f9a5a662
jeffhataws Move PJRT_DEVICE default into _maybe_select_default_device
b6b6f2ca
will-cromar
will-cromar approved these changes on 2024-09-05
aws-kingrj
aws-kingrj approved these changes on 2024-09-05
JackCaoG JackCaoG merged 901c3a32 into master 1 year ago
JackCaoG JackCaoG deleted the jeffhataws_update_neuron_initializations branch 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone