JAX Scalify: end-to-end scaled arithmetics
☆18Oct 30, 2024Updated last year
Alternatives and similar repositories for jax-scalify
Users that are interested in jax-scalify are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- A library for unit scaling in PyTorch☆136Jul 11, 2025Updated last year
- Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for tr…☆66Nov 18, 2025Updated 10 months ago
- Code for the paper "Function-Space Learning Rates"☆23Jun 3, 2025Updated last year
- ☆24Jun 18, 2024Updated 2 years ago
- 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.
- An awesome list that curates the best Flet tools, tutorials, blogs and more.☆10Jan 8, 2023Updated 3 years ago
- ☆18Aug 24, 2024Updated 2 years ago
- Pure Java Llama2 inference with optional multi-GPU CUDA implementation☆15Sep 2, 2023Updated 3 years ago
- // clone this repo with --depth=1 to save disk size // toolchain compatible with Ubuntu 20.04+ //☆15Apr 28, 2022Updated 4 years ago
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 3 years ago
- Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency☆25Jul 31, 2024Updated 2 years ago
- Official implementation of 'A Large-Scale Exploration of mu-Transfer' (CoRR 2024)☆32Jun 5, 2025Updated last year
- Explorations into the proposed SDFT, Self-Distillation Enables Continual Learning, from Shenfeld et al. of MIT☆33Feb 6, 2026Updated 8 months ago
- Code for paper "Concrete Subspace Learning based Interference Elimination for Multi-task Model Fusion"☆14Mar 28, 2024Updated 2 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Official repo of dataset-decomposition paper [NeurIPS 2024]☆21Sep 11, 2026Updated last month
- Source code for the paper "Do Deep Neural Network Solutions form a Star Domain?"☆11May 26, 2024Updated 2 years ago
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 years ago
- Towards Systematic Measurement for Long Text Quality☆39Sep 5, 2024Updated 2 years ago
- Neural multiclass ab initio reconstruction for cryo-EM.☆13Dec 5, 2024Updated last year
- JAX for Graphcore IPU (experimental)☆21Mar 12, 2024Updated 2 years ago
- Kinethreads: Soft Full-Body Haptic Exosuit using Low-Cost Motor-Pulley Mechanisms☆17Nov 6, 2025Updated 11 months ago
- An AI-powered web application that allows users to virtually try on clothing by overlaying garment images on user images. Built with comp…☆15Jul 5, 2026Updated 3 months ago
- itertree python package - full featured tree data structure☆15Sep 8, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆30Sep 11, 2026Updated last month
- GITHUB TEMPLATE — Click "Use this template" above or see link below for docs:☆15Aug 15, 2026Updated 2 months ago
- Code related to the ELM neuron.☆17Feb 27, 2024Updated 2 years ago
- Implementation of a Light Recurrent Unit in Pytorch☆51Oct 9, 2026Updated last week
- ☆19Feb 15, 2023Updated 3 years ago
- Hyperbolic decision trees☆14Jun 6, 2026Updated 4 months ago
- Official repository of paper [FALQON: Accelerating LoRA Fine-tuning with Low-Bit Floating-Point Arithmetic, NeurIPS 2025]☆22Dec 2, 2025Updated 10 months ago
- Official Repository for Efficient Linear-Time Attention Transformers.☆17Jun 2, 2024Updated 2 years ago
- The Spicy Project☆14May 31, 2016Updated 10 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Repository for various academic and professional CV versions☆10Aug 27, 2026Updated last month
- The official implementation of ICLR 2025 paper "Polynomial Composition Activations: Unleashing the Dynamics of Large Language Models".☆18Apr 25, 2025Updated last year
- Don't just regulate gradients like in Muon, regulate the weights too☆32Jul 30, 2025Updated last year
- Agentic coding harness with persistent memory and a REPL body. Built on Ori Mnemos. Open source must win.☆18May 8, 2026Updated 5 months ago
- Reviving Any-Order Autoregressive Models via Principled Parallel Sampling and Speculative Decoding☆16Nov 16, 2025Updated 11 months ago
- Benchmark SGLang on SLURM☆25Apr 20, 2026Updated 5 months ago
- This simple codebase provides implementation of Spatial Uncertainty Estimation (SUE), which is published in CVPR 2024 as a highlight post…☆30Jun 10, 2024Updated 2 years ago