XLA_USE_BF16
XLA_DOWNCAST_BF16
Get Started
jax-neuronx
Use ML Frameworks
torch-neuronx
torch-neuron
LSTM
tensorflow-neuronx
tensorflow-neuron
Training Libraries
neuronx-distributed-training
neuronx-distributed
Inference Libraries
neuronx-distributed-inference
NxD Core Libraries
Developer Tools
Orchestrate and Deploy
Runtime & Collectives
Compilers
neuronx-cc
neuron-cc
Neuron Kernel Interface (NKI)
Other Content
transformers-neuronx
jit
This decorator compiles a top-level NKI function to run on NeuronDevices.
