Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
☆26Mar 18, 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:
- Ongoing research training transformer models at scale☆39Updated this week
- 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
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆34Feb 26, 2026Updated 3 weeks ago
- Build scripts for Qemu.js☆17Jan 26, 2019Updated 7 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆25Updated this week
- AMD’s C++ library for accelerating tensor primitives☆49Updated this week
- Development repository for the Triton language and compiler☆143Updated this week
- The AMD rocAL is designed to efficiently decode and process images and videos from a variety of storage formats and modify them through a…☆23Updated this week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆26Mar 11, 2026Updated last week
- A GPU-Accelerated In-Memory Key-Value Store (AWS-focused fork)☆28Nov 1, 2017Updated 8 years ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆11May 6, 2023Updated 2 years ago
- CMake modules used within the ROCm libraries☆73Mar 13, 2026Updated last week
- A human-friendly implementation of the iRobot Open Interface version 2 API.☆14May 14, 2016Updated 9 years ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Feb 11, 2026Updated last month
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆26Updated this week
- RISC-V System on Chip Builder☆12Sep 27, 2020Updated 5 years ago
- 8-bit CUDA functions for PyTorch☆70Sep 24, 2025Updated 5 months ago
- ☆12Mar 19, 2022Updated 4 years ago
- CUDA keyring packaging for Debian☆14Apr 14, 2023Updated 2 years ago
- Build gstreamer on Raspberry Pi 3☆14Nov 2, 2018Updated 7 years ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆88Mar 5, 2026Updated 2 weeks ago
- The official Snap package of the FreeCAD project☆13Mar 7, 2026Updated 2 weeks ago
- ☆12May 8, 2025Updated 10 months ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆83Mar 12, 2026Updated last week
- Misc documentation and specifications☆14Feb 26, 2022Updated 4 years ago
- High-Performance Linpack Benchmark adopted version for GPU backend☆12Sep 12, 2022Updated 3 years ago
- OpenRISC Conference Website☆16Aug 15, 2024Updated last year
- NVIDIA SocketCAN drivers for Bosch Time Triggered M_CAN and Tegra Hypervisor sec CAN☆12Dec 26, 2021Updated 4 years ago
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆12Jun 24, 2024Updated last year
- A heavy modification of the original c_uart_interface_example, works on ARM Cortex-M4 STM32F4 (as an offboard processor)☆11Jul 8, 2016Updated 9 years ago
- Notes on putting micropython on STM32F407VG bare board☆11Oct 7, 2019Updated 6 years ago
- ☆64Updated this week
- Series of CLI tools for Hyperswarm☆18Jun 18, 2022Updated 3 years ago
- Procyon is the brightest star in the constellation of Canis Minor. But it's also the name of my RISC-V out-of-order processor.☆12Apr 6, 2023Updated 2 years ago
- HIP Python Low-level Bindings☆35Feb 12, 2026Updated last month
- Sphinx themes ("stanford" and "neo-rtd") based on readthedocs.org☆10Jan 9, 2017Updated 9 years ago
- ☆14Nov 15, 2024Updated last year
- ☆17Nov 27, 2023Updated 2 years ago
- Little cpu in verilog.☆11May 20, 2023Updated 2 years ago