LICENSE
README.md
pyproject.toml
src/diffmax/__init__.py
src/diffmax/_autograd.py
src/diffmax/_module.py
src/diffmax/_ops.py
src/diffmax/helpers.py
src/diffmax/monitoring.py
src/diffmax.egg-info/PKG-INFO
src/diffmax.egg-info/SOURCES.txt
src/diffmax.egg-info/dependency_links.txt
src/diffmax.egg-info/requires.txt
src/diffmax.egg-info/top_level.txt
src/diffmax/backends/__init__.py
src/diffmax/backends/cpu.py
src/diffmax/backends/cuda/__init__.py
src/diffmax/backends/cuda/csrc_impl.py
src/diffmax/backends/cuda/triton_impl.py
src/diffmax/backends/hip/__init__.py
src/diffmax/backends/hip/csrc_impl.py
src/diffmax/backends/hip/placeholder_impl.py
src/diffmax/backends/hip/tilelang_impl.py
src/diffmax/backends/hip/triton_impl.py
src/diffmax/backends/npu/__init__.py
src/diffmax/backends/npu/csrc_impl.py
src/diffmax/backends/npu/placeholder_impl.py
src/diffmax/backends/npu/tilelang_impl.py
src/diffmax/backends/npu/torch_npu_impl.py
tests/test_amp_compat.py
tests/test_autograd.py
tests/test_backend_placeholders.py
tests/test_compile.py
tests/test_cpu_backend.py
tests/test_cuda_backend.py
tests/test_cuda_selector.py
tests/test_helpers.py
tests/test_monitoring.py
tests/test_ops.py
tests/test_package.py
tests/test_semantic_correctness.py