Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About

The Top 9 Python Xla Open Source Projects

Open source projects categorized as Python Xla
Get Started With Jax ⭐ 487
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.
most recent commit 5 months ago
Extending Jax ⭐ 293
Extending JAX with custom C++ and CUDA code
most recent commit 6 months ago
Albert Tf2.0 ⭐ 155
ALBERT model Pretraining and Fine Tuning using TF2.0
most recent commit 4 years ago
Tensorflow Optimized Wheels ⭐ 116
TensorFlow wheels built for latest CUDA/CuDNN and enabled performance flags: SSE, AVX, FMA; XLA
most recent commit 4 years ago
Revlib ⭐ 106
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
dependent packages 2total releases 30most recent commit 2 years agopypi revlib} Downloads
Tfbert ⭐ 42
基于tensorflow1.x的预训练模型调用，支持单机多卡、梯度累积，XLA加速，混合精度。可灵活
most recent commit 3 years ago
Jaxdecomp ⭐ 18
JAX bindings for the NVIDIA cuDecomp library
most recent commit 3 months ago
Onnx Xla ⭐ 11
XLA integration of Open Neural Network Exchange (ONNX)
most recent commit 6 years ago
Tf2 Jit Compile On Multi Gpu ⭐ 9
Tensorflow2 training code with jit compiling on multi-GPU.
most recent commit 3 years ago
Popular Programming Languages Categories
Python
Javascript
Java
Typescript
C Plus Plus
Php
Golang
C
C Sharp
Shell
Categories
Privacy | About | Terms | Follow Us On Twitter

Downloads, Dependent Repos, Dependent Packages, Total Releases, Latest Releases data powered by Libraries.io.

Copyright 2018-2024 Awesome Open Source.  All rights reserved.