Your favourite classical machine learning algos on the GPU/TPU
☆22Dec 14, 2025Updated 2 months ago
Alternatives and similar repositories for scikit-jax
Users that are interested in scikit-jax are comparing it to the libraries listed below
Sorting:
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- Scalable Opponent Shaping Experiments in JAX☆25Apr 13, 2024Updated last year
- How to write a game of snakes in 10 lines of code☆20Aug 6, 2022Updated 3 years ago
- An Open-Ended Agentic Simulator☆60Aug 11, 2024Updated last year
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆31Feb 23, 2026Updated last week
- ☆34Jul 27, 2025Updated 7 months ago
- Official implementation of "BERTs are Generative In-Context Learners"☆32Mar 14, 2025Updated 11 months ago
- Implementation of PSGD optimizer in JAX☆35Dec 31, 2024Updated last year
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- a collection of colabs useful for molecular biology☆32Feb 27, 2024Updated 2 years ago
- Repo to reproduce the First-Explore paper results☆39Dec 25, 2024Updated last year
- R Shiny application "Vaccine Designer" aiming for the construction of vaccine sequences based on multi epitope design workflow.☆10Jun 25, 2024Updated last year
- Neuroevolution Benchmark in JAX 🦕☆42Nov 5, 2023Updated 2 years ago
- INTeractive learning via REPresentatIon Discovery☆36Jun 2, 2024Updated last year
- Official repo for paper "HiMoE-VLA: Hierarchical Mixture-of-Experts for Generalist Vision-Language-Action Policies"☆23Dec 12, 2025Updated 2 months ago
- Supporting code for Deep Neural Network or Dermatologist?☆10Nov 14, 2019Updated 6 years ago
- ☆13Jan 23, 2023Updated 3 years ago
- ☆16Jul 7, 2025Updated 7 months ago
- Sort & Slice: A Simple and Superior Alternative to Hash-Based Folding for Extended-Connectivity Fingerprints (ECFPs)☆10Feb 13, 2025Updated last year
- Software that runs reinout.vanrees.org☆20Feb 23, 2026Updated last week
- NEFF Calculator and MSA File Converter☆13Sep 16, 2025Updated 5 months ago
- Residual Quantization Autoencoder, used for interpreting LLMs☆14Jan 1, 2025Updated last year
- Fine-tuning GPT-2 to generate research paper abstracts☆12Apr 28, 2021Updated 4 years ago
- A fine-mapping method integrating GWAS summary statistics and functional annotation data☆11Dec 28, 2023Updated 2 years ago
- Distributed data sync using trimerge☆11Mar 26, 2024Updated last year
- Demonstration of the UPGMA hierarchal clustering algorithm in Pandas, Seaborn, and Scipy☆11Sep 29, 2019Updated 6 years ago
- Information geometry and its extension information topology☆11Dec 2, 2017Updated 8 years ago
- Incognito Proxy chrome extension☆10Sep 27, 2023Updated 2 years ago
- ☆78Nov 22, 2025Updated 3 months ago
- A potential solution to the early-stage small molecule hit identification☆14Jan 27, 2026Updated last month
- A proofreading tool using Google's N-gram corpus.☆12Sep 2, 2022Updated 3 years ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆14May 28, 2025Updated 9 months ago
- ☆12Sep 18, 2024Updated last year
- Tutorials on doing AMBER based Metadynamics and Gaussian Accelerated MD (GaMD)☆11Dec 1, 2025Updated 3 months ago
- The entry point for Rust projects to be run on Valida☆10Mar 14, 2025Updated 11 months ago
- AI Assistance for Writing Scientific Alt Text☆14Feb 7, 2024Updated 2 years ago
- Generating Protein Variants with Different Generative Models (HMM, VAE, ESM-2, ProtGPT2)☆11Mar 14, 2024Updated last year
- ☆10Feb 28, 2021Updated 5 years ago