easydel jax kernels writen in triton for gpus and pallas for tpus
☆28Jul 11, 2026Updated last week
Alternatives and similar repositories for ejkernel
Users that are interested in ejkernel are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- SpecTrax is a JAX-native library for neural networks and graph learning, built for performance, composability and modularity.☆42Jul 13, 2026Updated last week
- A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/…☆34Mar 4, 2025Updated last year
- Xerxes, a highly advanced Persian AI assistant developed by InstinctAI, a cutting-edge AI startup. primary function is to assist users wi…☆14Apr 27, 2024Updated 2 years ago
- OST Collection: An AI-powered suite of models that predict the next word matches with remarkable accuracy (Text Generative Models). OST C…☆16Nov 16, 2023Updated 2 years ago
- (EasyDel Former) is a utility library designed to simplify and enhance the development in JAX☆33Jun 21, 2026Updated last month
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Agents for intelligence and coordination☆26Updated this week
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆369Updated this week
- self-hosted workspace for Google's Jules AI agent☆16Jul 4, 2026Updated 2 weeks ago
- Tools and Utils for Experiments (TUX)☆17Jan 28, 2025Updated last year
- ☆18Aug 24, 2024Updated last year
- A fully trainable state space model (SSM)☆16Mar 18, 2025Updated last year
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 3 years ago
- Integration of "Yolov7 StrongSort" with ROS for real time object tracking☆15Dec 4, 2022Updated 3 years ago
- This project is a implementation in PyTorch for ZO-AdaMU optimization: Adapting Perturbation with the Momentum and Uncertainty in Zeroth-…☆15Dec 12, 2023Updated 2 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Tokamax: A GPU and TPU kernel library.☆249Updated this week
- ☆19Oct 3, 2023Updated 2 years ago
- ☆25Dec 16, 2025Updated 7 months ago
- ☆23Jan 23, 2024Updated 2 years ago
- See https://github.com/cuda-mode/triton-index/ instead!☆11May 8, 2024Updated 2 years ago
- Tensorflow 2.x implementation of Gradient Origin Networks☆12Jul 13, 2020Updated 6 years ago
- Implementation of 2-simplicial attention proposed by Clift et al. (2019) and the recent attempt to make practical in Fast and Simplex, Ro…☆49Sep 2, 2025Updated 10 months ago
- ☆15Jun 8, 2026Updated last month
- Pytorch/XLA SPMD Test code in Google TPU☆23Apr 3, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Implementation of PSGD optimizer in JAX☆36Dec 31, 2024Updated last year
- Official implementation of 'A Large-Scale Exploration of mu-Transfer' (CoRR 2024)☆31Jun 5, 2025Updated last year
- implementation of https://arxiv.org/pdf/2312.09299☆21Jul 3, 2024Updated 2 years ago
- ☆115Updated this week
- An embeddable Datalog database that is Datomic-inspired written in Golang☆45Updated this week
- Learn everything you need to know about any academic figure☆16May 10, 2024Updated 2 years ago
- A comprehensive toolkit for streamlining data editing, search, and inspection for large-scale language model training and interpretabilit…☆21Oct 30, 2025Updated 8 months ago
- 4-bit Shampoo for Memory-Efficient Network Training (NeurIPS 2024)☆13Feb 13, 2025Updated last year
- Dice Language Support for VS Code☆10Sep 29, 2020Updated 5 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- Partial implementation of ODE-GAN technique from the paper Training Generative Adversarial Networks by Solving Ordinary Differential Equa…☆16Nov 12, 2020Updated 5 years ago
- Muon in Int8 Precision Made Possible☆20Jun 18, 2026Updated last month
- ☆13Apr 25, 2024Updated 2 years ago
- TPU inference for vLLM, with unified JAX and PyTorch support.☆387Updated this week
- KANs and MLPs☆12Jun 7, 2024Updated 2 years ago
- Pytorch-like dataloaders for JAX.☆104Dec 16, 2025Updated 7 months ago