अलोचदार संघटक कृत्रिम तंत्रिका नेटवर्क के लिए एक सामान्यीकृत द्वैत पोटेंशियल: परिमित विकृतियों पर JAX का कार्यान्वयन