Awesome Open Source

Programming Languages

Search results for gpu jax

19 search results found

The Unified AI Framework

Training Operator ⭐ 1,447

Distributed ML Training and Fine-Tuning on Kubernetes

Transformerengine ⭐ 1,172

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.

A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine

Mpi4jax ⭐ 348

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

The versatile ocean simulator, in pure Python, powered by JAX.

Pyhpc Benchmarks ⭐ 290

A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python 🚀

Jaxfluids ⭐ 203

Differentiable Fluid Dynamics Package

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

JAX-based Spiking Neural Network framework

Differentiable optical models as parameterised neural networks in Jax using Zodiax

Python Suite for Advanced General Ensemble Simulations

Efficiently Composable Data Augmentation on the GPU with Jax

Fully JAX-compatible python implementation of the DC-EGM algorithm from Iskhakov, Jorgensen, Rust, and Schjerning (QE, 2017).

Scimlbenchmarksoutput ⭐ 13

SciML-Bench Benchmarks for Scientific Machine Learning (SciML), Physics-Informed Machine Learning (PIML), and Scientific AI Performance

Nvidia Gpu Ml Library Test ⭐ 11

Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being properly picked up

Doc Builder ⭐ 6

Simple Documentation Builder for Ivy Projects.

make your statistical research faster

Related Searches

Python Gpu (2,900)

C Plus Plus Gpu (1,847)

Gpu Nvidia (1,159)

Gpu Cpu (1,105)

Tensorflow Gpu (990)

Deep Learning Gpu (798)

Pytorch Gpu (613)

Machine Learning Gpu (423)

Gpu Parallel (248)

1-19 of 19 search results

Privacy | About | Terms | Follow Us On Twitter

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