#

xla

Here are 30 public repositories matching this topic...

elixir-nx / nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

elixir gpu jit pytorch tensor numerical xla

Updated Feb 14, 2023
Elixir

pytorch / xla

Enabling PyTorch on Google TPU

deep-learning compiler pytorch xla

Updated Feb 16, 2023
C++

n2cholas / awesome-jax

JAX - A curated list of resources https://github.com/google/jax

machine-learning awesome deep-learning neural-network numpy autograd awesome-list jax xla

Updated Feb 4, 2023

get-started-with-JAX

gordicaleksa / get-started-with-JAX

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.

python machine-learning tutorial deep-learning jupyter numpy flax haiku lax jax xla optax learn-jax

Updated Jul 11, 2022
Jupyter Notebook

mpi4jax / mpi4jax

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

gpu mpi parallel-computing jit high-performance-computing jax xla

Updated Feb 9, 2023
Python

dfm / extending-jax

Extending JAX with custom C++ and CUDA code

Updated Jan 2, 2023
Python

JuliaTPU / XLA.jl

Julia on TPUs

machine-learning deep-learning julia-language tpu xla going-faster peanut-butter

Updated Dec 18, 2020
Julia

kamalkraj / ALBERT-TF2.0

ALBERT model Pretraining and Fine Tuning using TF2.0

classifier glue tf2 mlm albert squad machine-comprehension cola tensoflow fine-tuning xla multi-gpu-training tf-hub albert-tf2 weights-conversion

Updated Nov 21, 2022
Python

inoryy / tensorflow-optimized-wheels

TensorFlow wheels built for latest CUDA/CuDNN and enabled performance flags: SSE, AVX, FMA; XLA

python tensorflow cuda sse avx2 wheels cudnn tensorflow-gpu xla tensorflow-wheels

Updated Feb 12, 2020

HomebrewNLP / revlib

Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload

deep-learning pytorch tpu revnet xla deepspeed momentumnet

Updated Aug 6, 2022
Python

huanghuidmml / tfbert

基于tensorflow1.x的预训练模型调用，支持单机多卡、梯度累积，XLA加速，混合精度。可灵活训练、验证、预测。

tensorflow trainer albert bert electra mixed-precision xla ernie ernie-gram chinesebert

Updated Aug 16, 2021
Python

sax

flaport / sax

S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.

deep-learning simulation optimization autograd simulation-framework simulations circuit physics-simulation ann photonics s-parameters jax xla photonic-circuit photonic-optimization

Updated Jan 7, 2023
Jupyter Notebook

sseung0703 / TF2-jit-compile-on-multi-gpu

Tensorflow2 training code with jit compiling on multi-GPU.

xla tensorflow2 multi-gpu-training jit-compilation

Updated Jan 28, 2021
Python

onnx / onnx-xla

XLA integration of Open Neural Network Exchange (ONNX)

tensorflow deeplearning onnx xla

Updated Aug 17, 2018
C++

scala-network / scala-pool

Official scala pool repository

mining mining-pool xla cryptonote-nodejs-pool scalahq

Updated Jan 22, 2023
JavaScript

DifferentiableUniverseInitiative / jaxDecomp

JAX bindings for the NVIDIA cuDecomp library

hpc cuda jax xla

Updated Jan 29, 2023
Python

scala-network / StellitePay-API

DEPRECATED ⛔️

api scala xla scalapay

Updated Jan 10, 2019
PHP

mugithi / google-terraform-pytorch-tpu

Automated provisioner of a Google Cloud TPU environment for training in PyTorch

pytorch xla cloud-tpu training-enviroment

Updated Aug 11, 2020
HCL

gottingen / tf-reading

tensorflow code reading

tensorflow llvm-ir xla tensorflow2

Updated Jan 12, 2023
C++

jhn-nt / data-snax

Versatile Data Ingestion Pipelines for Jax

machine-learning deep-learning pipeline tensorflow numpy data-loader jax xla data-loaders

Updated Jan 21, 2023
Python

Improve this page

Add a description, image, and links to the xla topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the xla topic, visit your repo's landing page and select "manage topics."