Top 3 Python Xla Projects

Xla

• Add a project

1. xla

   1 10 2,596 9.8 Python

   Enabling PyTorch on XLA Devices (e.g. Google TPU)

   

   Project mention: AI Model Optimization on AWS Inferentia and Trainium | dev.to | 2024-11-03

   As discussed in a previous post, Neuron Cores are treated as XLA devices and the torch-neuronx Python package implements the PyTorch/XLA API. Consequently, any optimization opportunities provided by the OpenXLA framework, and specifically those offered by the PyTorch/XLA API, can be leveraged on AWS Inferentia and Trainium. In this section we consider a few of these opportunities.

2. InfluxDB

   www.influxdata.com featured

   InfluxDB – Built for High-Performance Time Series Workloads. InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.

   

3. mpi4jax

   2 1 478 6.8 Python

   Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

   

4. extending-jax

   3 2 395 3.0 Python

   Extending JAX with custom C++ and CUDA code

   

Python Xla related posts

• Why Did Google Brain Exist?

  2 projects | news.ycombinator.com | 26 Apr 2023

• Extending JAX with custom C++ and CUDA code

  1 project | news.ycombinator.com | 12 Jan 2021

Index

Sponsored

InfluxDB – Built for High-Performance Time Series Workloads

InfluxDB 3 OSS is now GA. Transform, enrich, and act on time series data directly in the database. Automate critical tasks and eliminate the need to move data externally. Download now.

www.influxdata.com

Do not miss the trending Python projects with our weekly report!