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:
- Generic floating-point types in Python☆20Apr 18, 2026Updated 3 months 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 8 months ago
- Code for the paper "Function-Space Learning Rates"☆23Jun 3, 2025Updated last year
- ☆24Jun 18, 2024Updated 2 years ago
- 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.
- An awesome list that curates the best Flet tools, tutorials, blogs and more.☆10Jan 8, 2023Updated 3 years ago
- MESS: Modern Electronic Structure Simulations☆21Sep 24, 2024Updated last year
- ☆18Aug 24, 2024Updated last year
- Pure Java Llama2 inference with optional multi-GPU CUDA implementation☆13Sep 2, 2023Updated 2 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 2 years ago
- Official implementation of 'A Large-Scale Exploration of mu-Transfer' (CoRR 2024)☆31Jun 5, 2025Updated last year
- A collection of niche / personally useful PyTorch optimizers with modified code.☆28Apr 14, 2026Updated 4 months ago
- Official repository of Sparse ISO-FLOP Transformations for Maximizing Training Efficiency☆25Jul 31, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Explorations into the proposed SDFT, Self-Distillation Enables Continual Learning, from Shenfeld et al. of MIT☆32Feb 6, 2026Updated 6 months ago
- Code for paper "Concrete Subspace Learning based Interference Elimination for Multi-task Model Fusion"☆14Mar 28, 2024Updated 2 years ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago
- Official repo of dataset-decomposition paper [NeurIPS 2024]☆21Jan 8, 2025Updated last year
- 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 last year
- ☆25Sep 5, 2025Updated 11 months ago
- Neural multiclass ab initio reconstruction for cryo-EM.☆13Dec 5, 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.
- itertree python package - full featured tree data structure☆15Sep 8, 2025Updated 11 months ago
- Kinethreads: Soft Full-Body Haptic Exosuit using Low-Cost Motor-Pulley Mechanisms☆17Nov 6, 2025Updated 9 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 last month
- ☆10Oct 22, 2024Updated last year
- ☆30Jul 5, 2023Updated 3 years ago
- GITHUB TEMPLATE — Click "Use this template" above or see link below for docs:☆15Jul 13, 2026Updated last month
- Implementation of a Light Recurrent Unit in Pytorch☆50Oct 6, 2024Updated last year
- Code related to the ELM neuron.☆15Feb 27, 2024Updated 2 years ago
- ☆19Feb 15, 2023Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Official Repository for Efficient Linear-Time Attention Transformers.☆17Jun 2, 2024Updated 2 years ago
- A fine-mapping method integrating GWAS summary statistics and functional annotation data☆11Dec 28, 2023Updated 2 years ago
- Repository for various academic and professional CV versions☆10Jul 14, 2026Updated last month
- Don't just regulate gradients like in Muon, regulate the weights too☆32Jul 30, 2025Updated last year
- 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
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆17Jun 18, 2024Updated 2 years ago