mpi4jax
Array communicator
Enables JAX array communication over MPI without copying data
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python 
445 stars
11 watching
30 forks
Language: Python
last commit: 17 days ago
Linked from 1 awesome list
gpuhigh-performance-computingjaxjitmpiparallel-computingxla
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 461 |
 tigase/jaxmpp | An XMPP client library implementation of core and popular extensions in Java | 33 |
 jczic/esp32-mpy-jama | An integrated development environment and toolbox for MicroPython on ESP32 microcontrollers | 426 |
 jajberni/mcp342x_lopy | A MicroPython driver for the MCP342x ADC | 3 |
 hpi-swa/ipolyglot | A polyglot Jupyter kernel for GraalVM | 93 |
 davisyoshida/lorax | A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation. | 132 |
 mpdavis/python-jose | Provides a Python implementation of the JOSE technologies for encrypting and/or signing content. | 1,548 |
 jshilong/gpt4roi | Training and deploying large language models on computer vision tasks using region-of-interest inputs | 506 |
 ivoputzer/cli-args-parser-kata | Implementing a CLI arguments parser to process input in various formats | 5 |
 dfm/extending-jax | This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing | 378 |
 gpac/mp4box.js | A JavaScript library for processing MP4 files and providing file information, segmentation, and tracking functionality. | 1,993 |
| hpi-swa/trufflesqueak | A Squeak/Smalltalk VM and Polyglot Programming Environment for the GraalVM. | 285 |
 kongaskristjan/fire-hpp | Creates a CLI from a function signature with minimal boilerplate code | 446 |
 jposada202020/micropython_mma8452q | A MicroPython driver for interacting with an NXP MMA8452Q accelerometer | 0 |
 dino/dino | A modern XMPP chat client written in Vala using GTK+ | 2,232 |