xla
Here are 44 public repositories matching this topic...
Enabling PyTorch on XLA Devices (e.g. Google TPU)
-
Updated
Aug 21, 2024 - C++
JAX - A curated list of resources https://github.com/google/jax
-
Updated
Jul 10, 2024
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
-
Updated
Nov 29, 2023 - Jupyter Notebook
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡
-
Updated
Aug 20, 2024 - Python
Julia on TPUs
-
Updated
Dec 18, 2020 - Julia
ALBERT model Pretraining and Fine Tuning using TF2.0
-
Updated
Mar 24, 2023 - Python
GoMLX: An Accelerated Machine Learning Framework For Go
-
Updated
Aug 20, 2024 - Go
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
-
Updated
Aug 6, 2022 - Python
TensorFlow wheels built for latest CUDA/CuDNN and enabled performance flags: SSE, AVX, FMA; XLA
-
Updated
Feb 12, 2020
S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.
-
Updated
Aug 14, 2024 - Jupyter Notebook
基于tensorflow1.x的预训练模型调用,支持单机多卡、梯度累积,XLA加速,混合精度。可灵活训练、验证、预测。
-
Updated
Aug 16, 2021 - Python
Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
-
Updated
Aug 27, 2023 - Jupyter Notebook
Official scala pool repository
-
Updated
Sep 30, 2023 - JavaScript
XLA integration of Open Neural Network Exchange (ONNX)
-
Updated
Aug 17, 2018 - C++
Tensorflow2 training code with jit compiling on multi-GPU.
-
Updated
Jan 28, 2021 - Python
PyTorch distributed training acceleration framework
-
Updated
Aug 20, 2024 - Python
Improve this page
Add a description, image, and links to the xla topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the xla topic, visit your repo's landing page and select "manage topics."