Toy implementations of some popular ML optimizers using Python/JAX
☆44Jun 20, 2021Updated 5 years ago
Alternatives and similar repositories for ML-Optimizers-JAX
Users that are interested in ML-Optimizers-JAX are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Source code of "What can linearized neural networks actually say about generalization?☆20Oct 21, 2021Updated 4 years ago
- Local Attention - Flax module for Jax☆22May 26, 2021Updated 5 years ago
- ☆10Dec 17, 2020Updated 5 years ago
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago
- An implementation of the base GPT-3 Model architecture from the paper by OPENAI "Language Models are Few-Shot Learners"☆22Jun 29, 2024Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- Shape Matching Virtual Element Method for direct simulation on CAD models!☆13Feb 12, 2022Updated 4 years ago
- A PyTorch Implementation of the Skipgram Negative Sampling Word2Vec Model as Described in Mikolov et al.☆15Jan 13, 2020Updated 6 years ago
- This is the game Reversi☆12May 21, 2021Updated 5 years ago
- Code for the Paper 'On the Connection Between Adversarial Robustness and Saliency Map Interpretability' by C. Etmann, S. Lunz, P. Maass, …☆16May 9, 2019Updated 7 years ago
- A pytorch implementation of https://arxiv.org/abs/1903.01434☆12Jan 21, 2020Updated 6 years ago
- FastFeedForward Networks☆20Dec 8, 2023Updated 2 years ago
- A JAX implementation of stochastic addition.☆13Aug 15, 2022Updated 4 years ago
- ☆11May 31, 2023Updated 3 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.
- All about the fundamental blocks of TF and JAX!☆280Dec 4, 2021Updated 4 years ago
- pdfplot is a Python library for easily managing your matplotlib figures as PDF files.☆13Jul 8, 2020Updated 6 years ago
- Implementing activation functions from scratch in Tensorflow.☆38Feb 13, 2022Updated 4 years ago
- AI-powered pokemon bot on showdown☆14Oct 18, 2019Updated 6 years ago
- ☆17Aug 21, 2026Updated 2 weeks ago
- ☆10Apr 21, 2023Updated 3 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆51Jul 23, 2026Updated last month
- Solution for G2Net Detecting Continuous Gravitational Waves (2022)☆14Jan 18, 2023Updated 3 years ago
- TaiYiXLCheckpointLoader: An unoffical node support Taiyi-Diffusion-XL(Taiyi-XL) Chinese-English bilingual language model☆10Sep 1, 2024Updated 2 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Mar 23, 2022Updated 4 years ago
- ☆15Oct 29, 2019Updated 6 years ago
- [CVPR'25] Conformal prediction for vision-language models. Enhancing VLMs deployment with reliability gurarantees.☆21Jun 7, 2025Updated last year
- List of awesome JAX resources☆13Dec 8, 2022Updated 3 years ago
- 7th place code at NFL Big Data Bowl☆12Jan 8, 2020Updated 6 years ago
- NoiseGrad (and its extension NoiseGrad++) is a method to enhance explanations of artificial neural networks by adding noise to model weig…☆22May 11, 2023Updated 3 years ago
- S-LIME: Stabilized-LIME for model explanation (KDD 2021). pip install stabilized-lime☆16Mar 7, 2026Updated 6 months ago
- Research code and scripts used in the Silburt et al. (2021) EMNLP 2021 paper 'FANATIC: FAst Noise-Aware TopIc Clustering'☆11Jul 6, 2023Updated 3 years ago
- topic model visualization☆20Oct 12, 2021Updated 4 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- minGPT in JAX☆49Jan 10, 2022Updated 4 years ago
- Python code for HDP(Hierarchical Dirichlet Process) using Direct Assignment☆20Feb 9, 2017Updated 9 years ago
- Speech in Flax/JAX☆14Jul 11, 2022Updated 4 years ago
- Knowledge Graph based Question Answering benchmark.☆10Feb 1, 2020Updated 6 years ago
- A deep learning framework for deubiquitnase-substrate interaction identification☆13Jul 31, 2025Updated last year
- Datasets for Drug Discovery and Development☆10Aug 22, 2020Updated 6 years ago
- ☆774Jan 27, 2024Updated 2 years ago