Popular repositories Loading

1. taktiny taktiny Public

   A Deep Learning framework built on JAX, featuring OOP-style modeling, trainers, and native architectures spanning Transformers, Diffusion, and SSMs.

   Python 1

2. dextiny dextiny Public

   Abstract computation graph contracts and structural verification for JAX callables

   Python

3. opencl-pjrt opencl-pjrt Public

   OpenCL PJRT backend plugin for JAX enabling OpenCL GPU acceleration

   C++

4. vulkan-pjrt vulkan-pjrt Public

   Native Vulkan PJRT Plugin implementing the OpenXLA PJRT C API specification for acceleration of compute operations (JAX / XLA) on Vulkan-compatible GPUs

   C++

5. dialectiny dialectiny Public

   C++