No packages match

anvl - Accelerated Array Computing and Automatic Differentiation

Accelerated array computing and code transformations for R. Numerical programs operating on arrays can be just-in-time compiled to optimized executables via 'XLA' -- the same compiler that powers 'JAX' and 'TensorFlow' -- and run on CPU or NVIDIA GPU from the same source. Also provides reverse-mode automatic differentiation, returning the gradient of a function as another R function.

Last updated

array-computingautodiffgpu-accelerationxla

7.21 score 74 stars 1 dependents 51 scripts

pjrt - R Interface to PJRT

Provides an R interface to PJRT (Pluggable Jit RunTime), which compiles StableHLO and HLO programs with the XLA compiler and runs them on hardware backends such as CPU and GPU.

Last updated

deep-learninggpuxlaprotobufopenblascpp

5.46 score 8 stars 2 dependents 15 scripts

stablehlo - Write 'StableHLO' Programs

A low-level interface for building 'StableHLO' programs from R. 'StableHLO' is an operation set and portable intermediate representation for machine learning computations, developed within the 'OpenXLA' project and consumed by compilers such as 'XLA' <https://openxla.org/stablehlo>. The package implements the operations of the 'StableHLO' specification together with their type inference rules, so that output types are derived automatically and invalid programs are rejected with informative errors at construction time. The resulting programs can be compiled and executed on hardware backends such as CPU and GPU using the 'pjrt' package.

Last updated

deep-learninggpustablehloxla

5.41 score 8 stars 2 dependents 13 scripts

xlamisc - Helper Functions for 'r-xla'

Frequently used generics and helper functions used in 'anvl' and its companion packages 'stablehlo' and 'pjrt'. Comes with S3 generics to query tensors (shape, data type, device) and to convert them to R arrays and raw vectors, as well as an enum-style class for tensor element types.

Last updated

3.83 score 3 dependents 1 scripts

tengen - Tensor Generics

Generic functions and common objects for working with tensors.

Last updated

3.65 score 3 stars 1 dependents 1 scripts

xlamisc - Helper Functions for 'r-xla'

Frequently used generics and helper functions used in 'anvl' and its companion packages 'stablehlo' and 'pjrt'. Comes with S3 generics to query arrays (shape, data type, device) and to convert them to R arrays and raw vectors, as well as a class for data types.

Last updated

2.78 score 4 dependents 1 scripts

pjrt.cuda - CUDA Toolkit Libraries for 'pjrt'

Provides the CUDA 13.3 native libraries required by the CUDA PJRT plugin used by the 'pjrt' package (CUDA runtime, cuBLAS, cuDNN, CUPTI, NVRTC, cuFFT, cuSOLVER, cuSPARSE, NvJitLink, NCCL, and NVCC). The package itself is lightweight; the CUDA components are downloaded as NVIDIA's official redistributable binaries from PyPI into a user cache directory the first time the package is loaded.

Last updated

1.70 score