Search results

Adopt suite-wide NKI simulator dispatch pattern (NKI 0.3.0 migration)

Status: Open.

#23 In trnsci/trnsparse;

· scttfrdmn opened on Apr 14, 2026

On-chip iterative solvers over BSR (SBUF-resident A)

chemistryQuantum-chemistry / scientific-computing use caseQuantum-chemistry / scientific-computing use case

enhancementNew feature or requestNew feature or request

neuronRequires AWS Neuron / Trainium hardwareRequires AWS Neuron / Trainium hardware

Status: Open.

#22 In trnsci/trnsparse;

· scttfrdmn opened on Apr 13, 2026

Block-sparse attention primitive

enhancementNew feature or requestNew feature or request

neuronRequires AWS Neuron / Trainium hardwareRequires AWS Neuron / Trainium hardware

Status: Open.

#21 In trnsci/trnsparse;

· scttfrdmn opened on Apr 13, 2026

Fused screen + matmul NKI kernel

enhancementNew feature or requestNew feature or request

neuronRequires AWS Neuron / Trainium hardwareRequires AWS Neuron / Trainium hardware

Status: Open.

#19 In trnsci/trnsparse;

· scttfrdmn opened on Apr 13, 2026

Phase 5 — trn2 DMA bandwidth exploitation for SpMM

phase-5-generationRoadmap: phase-5-generationRoadmap: phase-5-generation

Status: Open.

#17 In trnsci/trnsparse;

· scttfrdmn opened on Apr 13, 2026

Phase 4 — sharded sparse matrices across chips

phase-4-multichipRoadmap: phase-4-multichipRoadmap: phase-4-multichip

Status: Open.

#16 In trnsci/trnsparse;

· scttfrdmn opened on Apr 13, 2026

Phase 3 — nnz-bucketing SpMM + streaming large-sparse + NEFF cache

phase-3-perfRoadmap: phase-3-perfRoadmap: phase-3-perf

Status: Open.

#15 In trnsci/trnsparse;

· scttfrdmn opened on Apr 13, 2026

examples: PySCF integration sketch

enhancementNew feature or requestNew feature or request

Status: Open.

#13 In trnsci/trnsparse;

· scttfrdmn opened on Apr 13, 2026

examples/sparse_fock.py: wire up to trnblas for the Fock contraction

enhancementNew feature or requestNew feature or request

Status: Open.

#6 In trnsci/trnsparse;

· scttfrdmn opened on Apr 13, 2026