jax

JAX के साथ डिफरेंशिएबल फाइनाइट एलिमेंट मेथड

Tevatron - स्केल, भाषा और मोडाॅलिटी में एकीकृत दस्तावेज़ पुनर्प्राप्ति टूलकिट। SIGIR 2023, SIGIR 2025 में डेमो।

लाइसेंस प्लेट टेक्स्ट रिकग्निशन के लिए हल्के और तेज़ OCR मॉडल

मशीन लर्निंग गाइड। मशीन लर्निंग टूल्स, लाइब्रेरीज़, फ्रेमवर्क्स, लार्ज लैंग्वेज मॉडल्स (LLMs) और ट्रेनिंग मॉडल्स के बारे में विस्तार से जानें।

मेनी-बॉडी क्वांटम सिस्टम के लिए मशीन लर्निंग एल्गोरिदम

फिजिक्स इन्फॉर्म्ड मशीन लर्निंग ट्यूटोरियल्स (Pytorch और Jax)।

JAX और Equinox में गॉसियन प्रोसेस।

JAX में ♟️ वेक्टरयुक्त RL गेम वातावरण

डिफरेंशिएबल फ्लूइड डायनेमिक्स पैकेज

JAX+Equinox में नॉन-लीनियर ऑप्टिमाइज़ेशन (रूट-फाइंडिंग, लीस्ट स्क्वेयर्स, ...)। https://docs.kidger.site/optimistix/

JAX और Equinox में लीनियर सॉल्वर। https://docs.kidger.site/lineax

Finite basis physics-informed neural networks (FBPINNs) का उपयोग करके आंशिक अंतर समीकरणों (PDEs) से संबंधित फॉरवर्ड और इनवर्स समस्याओं को हल करना।

Google Cloud TPU के बारे में वह सब कुछ जो आप जानना चाहते हैं

Felafax गैर-NVIDIA GPU के लिए AI बुनियादी ढांचा तैयार कर रहा है

शून्य से निर्मित LLM डिस्टिलेशन और सर्विंग इंजन: कस्टम Triton/CUDA कर्नेल, FSDP डिस्टिलेशन, पेज्ड-KV कंटीन्यूअस बैचिंग, सट्टा डिकोडिंग, एक Rust गेटवे, एक JAX ओरेकल और व्याख्यात्मकता उपकरण।

Pax बड़े पैमाने के मॉडल को ट्रेन करने के लिए एक Jax-आधारित मशीन लर्निंग फ्रेमवर्क है। Pax उन्नत और पूरी तरह से कॉन्फ़िगर करने योग्य प्रयोग और समानांतरकरण की अनुमति देता है, और इसने उद्योग-अग्रणी मॉडल फ्लॉप उपयोग दर का प्रदर्शन किया है।

Python में टर्बो-चार्ज्ड HPC अनुप्रयोगों के लिए JAX एरे का ज़ीरो-कॉपी MPI संचार :zap:

Orbax JAX यूजर्स के लिए सामान्य चेकपॉइंटिंग और परसिस्टेंस यूटिलिटीज प्रदान करता है

Yahoo Open-NSFW मॉडल का Keras कार्यान्वयन

https://srush.github.io/annotated-s4 का कार्यान्वयन

सिंगल रिजिड बॉडी मॉडल पर आधारित और Python में लिखा गया चौपाया रोबोटों के लिए एक मॉडल प्रेडिक्टिव कंट्रोलर। ग्रेडिएंट-आधारित (acados) या सैंपलिंग-आधारित (jax)

XLB: भौतिकी-आधारित ML के लिए Accelerated Lattice Boltzmann (XLB)