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
- Generic floating-point types in Python☆20Apr 18, 2026Updated 3 months ago
- A library for unit scaling in PyTorch☆135Jul 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 8 months ago
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 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.
- Code for the paper "Function-Space Learning Rates"☆23Jun 3, 2025Updated last year
- ☆24Jun 18, 2024Updated 2 years ago
- An awesome list that curates the best Flet tools, tutorials, blogs and more.☆10Jan 8, 2023Updated 3 years ago
- ☆18Aug 24, 2024Updated last year
- // 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 2 years ago
- Official implementation of 'A Large-Scale Exploration of mu-Transfer' (CoRR 2024)☆31Jun 5, 2025Updated last year
- Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency☆25Jul 31, 2024Updated last year
- Explorations into the proposed SDFT, Self-Distillation Enables Continual Learning, from Shenfeld et al. of MIT☆32Feb 6, 2026Updated 5 months ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Code for paper "Concrete Subspace Learning based Interference Elimination for Multi-task Model Fusion"☆14Mar 28, 2024Updated 2 years ago
- Official repo of dataset-decomposition paper [NeurIPS 2024]☆21Jan 8, 2025Updated last year
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 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
- itertree python package - full featured tree data structure☆15Sep 8, 2025Updated 10 months ago
- Kinethreads: Soft Full-Body Haptic Exosuit using Low-Cost Motor-Pulley Mechanisms☆17Nov 6, 2025Updated 8 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…☆14Jul 5, 2026Updated 3 weeks ago
- ☆10Oct 22, 2024Updated last year
- 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.
- ☆30Jul 5, 2023Updated 3 years ago
- Implementation of a Light Recurrent Unit in Pytorch☆50Oct 6, 2024Updated last year
- ☆19Feb 15, 2023Updated 3 years ago
- Official Repository for Efficient Linear-Time Attention Transformers.☆18Jun 2, 2024Updated 2 years ago
- A fine-mapping method integrating GWAS summary statistics and functional annotation data☆11Dec 28, 2023Updated 2 years ago
- The Spicy Project☆14May 31, 2016Updated 10 years ago
- Repository for various academic and professional CV versions☆10Jul 14, 2026Updated last week
- Agentic coding harness with persistent memory and a REPL body. Built on Ori Mnemos. Open source must win.☆17May 8, 2026Updated 2 months ago
- The official implementation of ICLR 2025 paper "Polynomial Composition Activations: Unleashing the Dynamics of Large Language Models".☆18Apr 25, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Don't just regulate gradients like in Muon, regulate the weights too☆32Jul 30, 2025Updated 11 months ago
- 🚀 Sliding Window Attention Training for Efficient Large Language Models☆19Jun 7, 2026Updated last month
- Benchmark SGLang on SLURM☆24Apr 20, 2026Updated 3 months ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- Reviving Any-Order Autoregressive Models via Principled Parallel Sampling and Speculative Decoding☆16Nov 16, 2025Updated 8 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Jun 12, 2024Updated 2 years ago
- [ICML 2025] Flat-LoRA: Low-Rank Adaptation over a Flat Loss Landscape☆18May 18, 2025Updated last year