Enabling PyTorch on XLA Devices (e.g. Google TPU)
A Differentiable Data Pipeline Framework for JAX
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡
JAX bindings for the cuDecomp library
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
High-performance TensorFlow library for quantitative finance.
Felafax is building AI infra for non-NVIDIA GPUs
A fast transfer matrix method written in jax for modelling optical multilayer thin films
project_description
