XLA_USE_BF16
XLA_DOWNCAST_BF16
Get Started
jax-neuronx
Orchestrate and Deploy
Use ML Frameworks
torch-neuronx
torch-neuron
LSTM
tensorflow-neuronx
tensorflow-neuron
Work with Neuron Libraries
neuronx-distributed-training
neuronx-distributed-inference
neuronx-distributed
Runtime & Collectives
Neuron Kernel Interface (NKI)
Compiler
neuronx-cc
neuron-cc
Legacy software and docs
tfloat32
32-bit floating-point number (1S,8E,10M)