Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
☆32Nov 2, 2026Updated this week
Alternatives and similar repositories for jax
Users that are interested in jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Sight_FrameServer is the delivery method used by SIGHT for remote visualization. SIGHT is an exploratory visualization tool for large sca…☆10Feb 28, 2019Updated 7 years ago
- The AMD rocAL is designed to efficiently decode and process images and videos from a variety of storage formats and modify them through a…☆26Updated this week
- ☆11Oct 14, 2026Updated 3 weeks ago
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆34May 29, 2026Updated 5 months ago
- CMake modules used within the ROCm libraries☆78Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Scripts to automate building linux images for my emulator riscv_em☆16Jul 27, 2026Updated 3 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆24Oct 19, 2026Updated 2 weeks ago
- AMD’s C++ library for accelerating tensor primitives☆50Updated this week
- Hacking repo for UL3XS FPGA stuff.☆22Aug 9, 2024Updated 2 years ago
- A demo code for implementation of differentiable thermodynamic modeling in JAX.☆11Sep 18, 2021Updated 5 years ago
- Automatically exported from code.google.com/p/another-chebpy☆13Apr 28, 2015Updated 11 years ago
- Radix sort for floating point values☆10Jun 18, 2017Updated 9 years ago
- Development repository for the Triton language and compiler☆146Updated this week
- A GPU-Accelerated In-Memory Key-Value Store (AWS-focused fork)☆27Nov 1, 2017Updated 9 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆27Updated this week
- A list of links to useful code snippets for manim☆14Jul 27, 2024Updated 2 years ago
- Discretization of the Lippmann--Schwinger equation with periodic boundary conditions☆13Sep 26, 2026Updated last month
- Torch-MIGraphX integrates AMD's graph inference engine with the PyTorch ecosystem.☆22Updated this week
- Slurm SPANK plugin to ease setup of SSH tunnels and port forwarding☆12Mar 21, 2024Updated 2 years ago
- Examples using petsc4py, cython and petsc to solve Finite Volume Problems.☆14Sep 10, 2017Updated 9 years ago
- PNP+NS Solver in Python☆15Apr 19, 2018Updated 8 years ago
- Scripts to recover (accidentally) deleted files from ext3 partitions☆13Aug 16, 2017Updated 9 years ago
- The source documentation of FreeCAD, generated with 'make WebDoc'☆11Aug 2, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Library and tools to access the SGI X File System (XFS)☆13Oct 11, 2026Updated 3 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆27Updated this week
- CUDA keyring packaging for Debian☆14Apr 14, 2023Updated 3 years ago
- DEPRECATED: Development repository for Chef Cookbook xfs☆12Feb 8, 2016Updated 10 years ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆83May 28, 2026Updated 5 months ago
- ☆77Updated this week
- Regression test suite for xfs and other filesystems. The zfs branch has been updated to be compatible with ZFS.☆13Nov 4, 2015Updated 11 years ago
- Help system addon for FreeCAD.☆11Apr 1, 2026Updated 7 months ago
- Next generation library for iterative sparse solvers for ROCm platform☆104Updated this week
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆13Updated this week
- AMD ROCm Performance Primitives (RPP) library is a comprehensive high-performance computer vision library for AMD processors with HIP/Ope…☆74Oct 22, 2026Updated last week
- This repository is deprecated. New location is https://github.com/linux-can/socketcand☆10Aug 9, 2019Updated 7 years ago
- 8-bit CUDA functions for PyTorch☆71Aug 19, 2026Updated 2 months ago
- 程式人雜誌☆11Dec 23, 2023Updated 2 years ago
- A set of Dockerfiles for images that support Standard Diffusion on AMD 7900XTX Cards (gfx1100)☆10Jul 4, 2023Updated 3 years ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆83Updated this week