Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About

The Top 19 Gpu Jax Open Source Projects

Open source projects categorized as Gpu Jax
Categories > Hardware > Gpu
Categories > Jax
Ivy ⭐ 13,904
The Unified AI Framework
dependent packages 7total releases 103most recent commit 3 months agopypi ivy} Downloads
Training Operator ⭐ 1,447
Distributed ML Training and Fine-Tuning on Kubernetes
dependent packages 28total releases 128most recent commit 4 days ago
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.
most recent commit 3 months ago
Mosec ⭐ 661
A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine
dependent packages 1total releases 40most recent commit 3 months agopypi mosec} Downloads
Mpi4jax ⭐ 348
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡
dependent packages 2total releases 61most recent commit 2 months agopypi mpi4jax} Downloads
Veros ⭐ 303
The versatile ocean simulator, in pure Python, powered by JAX.
most recent commit 5 days ago
Pyhpc Benchmarks ⭐ 290
A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python 🚀
most recent commit 4 months ago
Jaxfluids ⭐ 203
Differentiable Fluid Dynamics Package
most recent commit 4 months ago
Jwave ⭐ 108
A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
dependent packages 1total releases 5most recent commit 3 months agopypi jwave} Downloads
Symjax ⭐ 90
Documentation:
total releases 13most recent commit 3 years ago
Ra9 ⭐ 47
JAX-based Spiking Neural Network framework
total releases 23most recent commit 3 years ago
Dlux ⭐ 38
Differentiable optical models as parameterised neural networks in Jax using Zodiax
dependent packages 2total releases 17most recent commit 5 months agopypi dLux} Downloads
Pysages ⭐ 33
Python Suite for Advanced General Ensemble Simulations
most recent commit 3 months ago
Augmax ⭐ 18
Efficiently Composable Data Augmentation on the GPU with Jax
total releases 5most recent commit a year ago
Dcegm ⭐ 15
Fully JAX-compatible python implementation of the DC-EGM algorithm from Iskhakov, Jorgensen, Rust, and Schjerning (QE, 2017).
most recent commit 5 months ago
Scimlbenchmarksoutput ⭐ 13
SciML-Bench Benchmarks for Scientific Machine Learning (SciML), Physics-Informed Machine Learning (PIML), and Scientific AI Performance
most recent commit 3 months ago
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
most recent commit 3 months ago
Doc Builder ⭐ 6
Simple Documentation Builder for Ivy Projects.
most recent commit 3 months ago
Stamox ⭐ 6
make your statistical research faster
total releases 12most recent commit 10 months ago
The 6 Most Depended On Gpu Jax Open Source Projects
Training Operator ⭐ 1,447
Distributed ML Training and Fine-Tuning on Kubernetes
dependent packages 28total releases 128latest release November 01, 2023most recent commit 4 days ago
Ivy ⭐ 13,904
The Unified AI Framework
dependent packages 7total releases 103latest release October 11, 2023most recent commit 3 months agopypi ivy} Downloads
Mpi4jax ⭐ 348
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
dependent packages 2total releases 61latest release December 11, 2023most recent commit 2 months agopypi mpi4jax} Downloads
Dlux ⭐ 38
Differentiable optical models as parameterised neural networks in Jax using Zodiax
dependent packages 2total releases 17latest release October 04, 2023most recent commit 5 months agopypi dLux} Downloads
Jwave ⭐ 108
A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
dependent packages 1total releases 5latest release September 27, 2023most recent commit 3 months agopypi jwave} Downloads
Mosec ⭐ 661
A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine
dependent packages 1total releases 40latest release December 05, 2023most recent commit 3 months agopypi mosec} Downloads
The 10 Latest Releases In Gpu Jax Open Source Projects
Mpi4jax ⭐ 348
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
dependent packages 2total releases 61latest release December 11, 2023most recent commit 2 months agopypi mpi4jax} Downloads
Mosec ⭐ 661
A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine
dependent packages 1total releases 40latest release December 05, 2023most recent commit 3 months agopypi mosec} Downloads
Training Operator ⭐ 1,447
Distributed ML Training and Fine-Tuning on Kubernetes
dependent packages 28total releases 128latest release November 01, 2023most recent commit 4 days ago
Ivy ⭐ 13,904
The Unified AI Framework
dependent packages 7total releases 103latest release October 11, 2023most recent commit 3 months agopypi ivy} Downloads
Dlux ⭐ 38
Differentiable optical models as parameterised neural networks in Jax using Zodiax
dependent packages 2total releases 17latest release October 04, 2023most recent commit 5 months agopypi dLux} Downloads
Jwave ⭐ 108
A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
dependent packages 1total releases 5latest release September 27, 2023most recent commit 3 months agopypi jwave} Downloads
Stamox ⭐ 6
make your statistical research faster
total releases 12latest release November 10, 2022most recent commit 10 months ago
Augmax ⭐ 18
Efficiently Composable Data Augmentation on the GPU with Jax
total releases 5latest release September 26, 2021most recent commit a year ago
Ra9 ⭐ 47
JAX-based Spiking Neural Network framework
total releases 23latest release December 08, 2020most recent commit 3 years ago
Symjax ⭐ 90
Documentation:
total releases 13latest release November 11, 2020most recent commit 3 years ago
Popular Hardware Categories
Arduino
Raspberry Pi
Camera
Sensor
Gpu
Firmware
Usb
Keyboard
Cpu
Arm
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.