mpi4jax
by mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python 
AI summary
Array communicator
Enables JAX array communication over MPI without copying data
- stars
- 453
- forks
- 30
- watching
- 11
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
XMPP client library
An XMPP client library implementation of core and popular extensions in Java
MicroPython IDE
An integrated development environment and toolbox for MicroPython on ESP32 microcontrollers
ADC driver
A MicroPython driver for the MCP342x ADC
Polyglot runtime
A polyglot Jupyter kernel for GraalVM
Memory optimizer
A JAX transform that simplifies the training of large language models by reducing memory usage through low-rank adaptation.
Content Encryption Library
Provides a Python implementation of the JOSE technologies for encrypting and/or signing content.
Region-of-Interest Training
Training and deploying large language models on computer vision tasks using region-of-interest inputs
CLI parser
Implementing a CLI arguments parser to process input in various formats
JAX wrapper
This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing
MP4 parser
A JavaScript library for processing MP4 files and providing file information, segmentation, and tracking functionality.
Virtual Machine
A Squeak/Smalltalk VM and Polyglot Programming Environment for the GraalVM.
CLI generator
Creates a CLI from a function signature with minimal boilerplate code
Accelerometer driver
A MicroPython driver for interacting with an NXP MMA8452Q accelerometer
dino/dino2.2K
XMPP client
A modern XMPP chat client written in Vala using GTK+
Featured in 1 awesome list
Each link jumps to the spot where the list mentions mpi4jax.
